Function infer

script/echomimic_api.py:119–158

接收 image + audio，返回视频字节（video/mp4）。忙时返回 503。

(
    image: UploadFile = File(...),
    audio: UploadFile = File(...),
)

@app.post("/infer")
async def infer(
    image: UploadFile = File(...),
    audio: UploadFile = File(...),
):
    """接收 image + audio，返回视频字节（video/mp4）。忙时返回 503。"""
    global _total_requests, _in_flight
    lock = _get_lock()
    if not lock.acquire(blocking=False):
        raise HTTPException(status_code=503, detail="Instance busy, try another or retry later")

    _in_flight = 1
    try:
        _total_requests += 1
        with tempfile.TemporaryDirectory(prefix="echomimic_api_") as tmpdir:
            tmp = Path(tmpdir)
            img_path = tmp / (image.filename or "image.png")
            aud_path = tmp / (audio.filename or "audio.wav")
            with open(img_path, "wb") as f:
                f.write(await image.read())
            with open(aud_path, "wb") as f:
                f.write(await audio.read())

            loop = asyncio.get_event_loop()
            out_mp4 = await loop.run_in_executor(
                None,
                lambda: _run_echomimic_subprocess(str(img_path), str(aud_path), str(tmp)),
            )
            if not out_mp4 or not out_mp4.exists():
                raise HTTPException(status_code=500, detail="EchoMimic did not produce output mp4")
            data = out_mp4.read_bytes()
        return Response(content=data, media_type="video/mp4")
    except HTTPException:
        raise
    except Exception as e:
        import traceback
        tb = traceback.format_exc()
        raise HTTPException(status_code=500, detail=f"EchoMimic inference failed: {e}\n{tb}")
    finally:
        _in_flight = 0
        lock.release()


@app.get("/health")