"""
Vocal Helper — Model Context Protocol (MCP) surface.
Adapter that exposes the FastAPI app defined in :mod:`vocal_helper.api`
as MCP tools so any MCP-aware host (agent runtimes, IDE
integrations, custom shells) can call ``transcribe`` / ``pipeline`` as
first-class tools. Uses :mod:`fastapi_mcp`
(https://github.com/tadata-org/fastapi_mcp) — one line wraps the whole
existing HTTP surface, so we never duplicate the route definitions.
Install the extra to pull in ``fastapi-mcp``::
pip install 'vocal-helper[api,mcp]'
Then run the MCP server::
vocal-helper-mcp # entry point (see pyproject)
# or, equivalently:
python -m vocal_helper.mcp
Usage Example
-------------
>>> # Register the MCP endpoint in your client. It publishes:
>>> # transcribe / pipeline
>>> # …with the same argument names as the FastAPI routes.
Author
------
Warith Harchaoui, Ph.D. — https://linkedin.com/in/warith-harchaoui/
"""
from __future__ import annotations
try:
from fastapi_mcp import FastApiMCP
except ImportError as exc: # pragma: no cover
raise ImportError(
"The MCP surface requires the [mcp] extra. "
"Install with: pip install 'vocal-helper[api,mcp]'"
) from exc
# Reuse the exact same FastAPI app — MCP is a thin wrapper on top.
from vocal_helper.api import app
# ``FastApiMCP`` mounts an MCP endpoint on the existing FastAPI app; we
# store the wrapped instance at module scope so downstream code (tests,
# ASGI runners) can access both the FastAPI app and the MCP handler.
mcp = FastApiMCP(
app,
name="vocal-helper",
description=(
"Vocal Helper MCP tools: transcribe an audio file, or run the full "
"offline pipeline (VAD + diarization + STT + optional Gemma summary)."
),
)
# Attach the MCP endpoint to the FastAPI app. Newer fastapi-mcp releases
# split ``mount()`` into transport-specific ``mount_http()`` (recommended)
# and ``mount_sse()``. Fall back to the legacy ``mount()`` on older
# versions so users can install a range of ``fastapi-mcp`` versions.
if hasattr(mcp, "mount_http"):
mcp.mount_http()
else: # pragma: no cover — legacy fastapi-mcp
mcp.mount()
[docs]
def main() -> None:
"""
Entry point for the ``vocal-helper-mcp`` console script.
Boots the FastAPI app (which now serves both the ``/…`` HTTP routes
and the MCP endpoint) with ``uvicorn`` in single-worker mode. Meant
for local / container usage; behind a real load balancer use
``uvicorn`` / ``gunicorn`` directly.
"""
import os
import uvicorn
host = os.environ.get("VOCAL_HELPER_HOST", "0.0.0.0")
port = int(os.environ.get("VOCAL_HELPER_PORT", "8000"))
# Single worker keeps whisper.cpp + pyannote model instances in a
# shared process (no reload storms across workers).
uvicorn.run(app, host=host, port=port, workers=1)
if __name__ == "__main__": # pragma: no cover
main()