diff --git a/examples/p2p-webrtc/video-transform/server/requirements.txt b/examples/p2p-webrtc/video-transform/server/requirements.txt index 24579bd41..f638127b5 100644 --- a/examples/p2p-webrtc/video-transform/server/requirements.txt +++ b/examples/p2p-webrtc/video-transform/server/requirements.txt @@ -3,4 +3,5 @@ fastapi[all] uvicorn aiortc opencv-python -pipecat-ai[google,silero] \ No newline at end of file +pipecat-ai[google,silero] +pipecat-ai-small-webrtc-prebuilt \ No newline at end of file diff --git a/examples/p2p-webrtc/video-transform/server/server.py b/examples/p2p-webrtc/video-transform/server/server.py index 79ee75ea3..295fe4f44 100644 --- a/examples/p2p-webrtc/video-transform/server/server.py +++ b/examples/p2p-webrtc/video-transform/server/server.py @@ -8,6 +8,8 @@ import uvicorn from bot import run_bot from dotenv import load_dotenv from fastapi import BackgroundTasks, FastAPI +from fastapi.responses import RedirectResponse +from pipecat_ai_small_webrtc_prebuilt.frontend import SmallWebRTCPrebuiltUI from pipecat.transports.network.webrtc_connection import SmallWebRTCConnection @@ -23,6 +25,14 @@ pcs_map: Dict[str, SmallWebRTCConnection] = {} ice_servers = ["stun:stun.l.google.com:19302"] +# Mount the frontend at / +app.mount("/prebuilt", SmallWebRTCPrebuiltUI) + + +@app.get("/", include_in_schema=False) +async def root_redirect(): + return RedirectResponse(url="/prebuilt/") + @app.post("/api/offer") async def offer(request: dict, background_tasks: BackgroundTasks):