2eb2c5a413
SmallWebRTCConnection doesn't have a `close()`. There's a `_close()` but I assume that's private due to its naming. The closest function that uses `_close()` is `disconnect()`. I assume then, that the intended resource freeing function call should be to `disconnect()`.
100 lines
2.8 KiB
Python
100 lines
2.8 KiB
Python
#
|
||
# Copyright (c) 2024–2025, Daily
|
||
#
|
||
# SPDX-License-Identifier: BSD 2-Clause License
|
||
#
|
||
|
||
import argparse
|
||
import asyncio
|
||
import sys
|
||
from contextlib import asynccontextmanager
|
||
from typing import Dict
|
||
|
||
import uvicorn
|
||
from bot import run_bot
|
||
from dotenv import load_dotenv
|
||
from fastapi import BackgroundTasks, FastAPI
|
||
from fastapi.responses import RedirectResponse
|
||
from loguru import logger
|
||
from pipecat_ai_small_webrtc_prebuilt.frontend import SmallWebRTCPrebuiltUI
|
||
|
||
from pipecat.transports.network.webrtc_connection import IceServer, SmallWebRTCConnection
|
||
|
||
# Load environment variables
|
||
load_dotenv(override=True)
|
||
|
||
app = FastAPI()
|
||
|
||
# Store connections by pc_id
|
||
pcs_map: Dict[str, SmallWebRTCConnection] = {}
|
||
|
||
ice_servers = [
|
||
IceServer(
|
||
urls="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):
|
||
pc_id = request.get("pc_id")
|
||
|
||
if pc_id and pc_id in pcs_map:
|
||
pipecat_connection = pcs_map[pc_id]
|
||
logger.info(f"Reusing existing connection for pc_id: {pc_id}")
|
||
await pipecat_connection.renegotiate(
|
||
sdp=request["sdp"], type=request["type"], restart_pc=request.get("restart_pc", False)
|
||
)
|
||
else:
|
||
pipecat_connection = SmallWebRTCConnection(ice_servers)
|
||
await pipecat_connection.initialize(sdp=request["sdp"], type=request["type"])
|
||
|
||
@pipecat_connection.event_handler("closed")
|
||
async def handle_disconnected(webrtc_connection: SmallWebRTCConnection):
|
||
logger.info(f"Discarding peer connection for pc_id: {webrtc_connection.pc_id}")
|
||
pcs_map.pop(webrtc_connection.pc_id, None)
|
||
|
||
background_tasks.add_task(run_bot, pipecat_connection)
|
||
|
||
answer = pipecat_connection.get_answer()
|
||
# Updating the peer connection inside the map
|
||
pcs_map[answer["pc_id"]] = pipecat_connection
|
||
|
||
return answer
|
||
|
||
|
||
@asynccontextmanager
|
||
async def lifespan(app: FastAPI):
|
||
yield # Run app
|
||
coros = [pc.disconnect() for pc in pcs_map.values()]
|
||
await asyncio.gather(*coros)
|
||
pcs_map.clear()
|
||
|
||
|
||
if __name__ == "__main__":
|
||
parser = argparse.ArgumentParser(description="WebRTC demo")
|
||
parser.add_argument(
|
||
"--host", default="localhost", help="Host for HTTP server (default: localhost)"
|
||
)
|
||
parser.add_argument(
|
||
"--port", type=int, default=7860, help="Port for HTTP server (default: 7860)"
|
||
)
|
||
parser.add_argument("--verbose", "-v", action="count")
|
||
args = parser.parse_args()
|
||
|
||
logger.remove(0)
|
||
if args.verbose:
|
||
logger.add(sys.stderr, level="TRACE")
|
||
else:
|
||
logger.add(sys.stderr, level="DEBUG")
|
||
|
||
uvicorn.run(app, host=args.host, port=args.port)
|