Upload folder using huggingface_hub

app.py

@@ -1,10 +1,10 @@
+import json
 from pathlib import Path
 
 import gradio as gr
-from gradio.utils import get_space
 import numpy as np
-import json
 from dotenv import load_dotenv
+from fastapi import FastAPI
 from fastapi.responses import HTMLResponse, StreamingResponse
 from fastrtc import (
     AdditionalOutputs,
@@ -13,6 +13,7 @@ from fastrtc import (
     audio_to_bytes,
     get_twilio_turn_credentials,
 )
+from gradio.utils import get_space
 from groq import AsyncClient
 
 cur_dir = Path(__file__).parent
@@ -44,8 +45,12 @@ stream = Stream(
     concurrency_limit=20 if get_space() else None,
 )
 
+app = FastAPI()
 
-@stream.get("/transcript")
+stream.mount(app)
+
+
+@app.get("/transcript")
 def _(webrtc_id: str):
     async def output_stream():
         async for output in stream.output_stream(webrtc_id):
@@ -55,7 +60,7 @@ def _(webrtc_id: str):
     return StreamingResponse(output_stream(), media_type="text/event-stream")
 
 
-@stream.get("/")
+@app.get("/")
 def index():
     rtc_config = get_twilio_turn_credentials() if get_space() else None
     html_content = (cur_dir / "index.html").read_text()
@@ -64,6 +69,13 @@ def index():
 
 
 if __name__ == "__main__":
-    import uvicorn
+    import os
+
+    if (mode := os.getenv("MODE")) == "UI":
+        stream.ui.launch(server_port=7860)
+    elif mode == "PHONE":
+        stream.fastphone(host="0.0.0.0", port=7860)
+    else:
+        import uvicorn
 
-    uvicorn.run(stream, host="0.0.0.0", port=7860)
+        uvicorn.run(app, host="0.0.0.0", port=7860)