fastrtfc

ui.py

@@ -1,6 +1,6 @@
 import gradio as gr
 from fastapi import FastAPI, WebSocket, WebSocketDisconnect
-from fastapi.responses import JSONResponse
+from fastapi.responses import JSONResponse, RedirectResponse
 import asyncio
 import json
 import logging
@@ -11,8 +11,13 @@ import httpx
 import websockets
 from fastrtc import RTCComponent
 
+# Configure logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
 class Config:
     def __init__(self):
+        # URLs should not include http/https prefix as we add it contextually
         self.hf_space_url = os.getenv("HF_SPACE_URL", "androidguy-speaker-diarization.hf.space")
         self.render_url = os.getenv("RENDER_URL", "render-signal-audio.onrender.com")
         self.default_threshold = float(os.getenv("DEFAULT_THRESHOLD", "0.7"))
@@ -20,7 +25,6 @@ class Config:
         self.max_speakers_limit = int(os.getenv("MAX_SPEAKERS_LIMIT", "8"))
 
 config = Config()
-logger = logging.getLogger(__name__)
 
 class ConnectionManager:
     """Manage WebSocket connections"""
@@ -494,9 +498,54 @@ async def process_audio_chunk(audio_data: bytes) -> dict:
 fastapi_app = create_fastapi_app()
 gradio_app = create_gradio_app()
 
-# Mount Gradio app to FastAPI
+# Root redirect
+@fastapi_app.get("/")
+async def root():
+    return RedirectResponse(url="/ui")
+
+# Mount Gradio app to FastAPI at /ui
 fastapi_app.mount("/ui", gradio_app.app)
 
+# Add diagnostic endpoints to check connections
+@fastapi_app.get("/check-backend")
+async def check_backend():
+    """Check connection to the Render backend"""
+    try:
+        # Check if we can connect to the WebSocket endpoint on Render
+        websocket_url = f"wss://{config.render_url}/stream"
+        logger.info(f"Checking connection to Render backend at {websocket_url}")
+        
+        # Don't actually connect, just return status
+        return {
+            "status": "configured",
+            "render_backend_url": websocket_url,
+            "hf_space_url": f"wss://{config.hf_space_url}/ws_inference",
+            "rtc_component_config": {
+                "url": f"wss://{config.render_url}/stream",
+                "modality": "audio",
+                "mode": "send-receive"
+            }
+        }
+    except Exception as e:
+        logger.error(f"Error checking backend: {e}")
+        return {
+            "status": "error",
+            "error": str(e)
+        }
+
+# Log configuration on startup
+@fastapi_app.on_event("startup")
+async def log_configuration():
+    logger.info(f"Starting UI with configuration:")
+    logger.info(f"- HF Space URL: {config.hf_space_url}")
+    logger.info(f"- Render URL: {config.render_url}")
+    logger.info(f"- WebRTC URL: wss://{config.render_url}/stream")
+    logger.info(f"- WebSocket URL: wss://{config.hf_space_url}/ws_inference")
+    logger.info("Note: Audio will be streamed through the Render backend using WebRTC")
+
 if __name__ == "__main__":
     import uvicorn
-    uvicorn.run(fastapi_app, host="0.0.0.0", port=7860)
+    # Use the correct port for Hugging Face Spaces (7860)
+    port = int(os.environ.get("PORT", 7860))
+    logger.info(f"Starting server on port {port}")
+    uvicorn.run(fastapi_app, host="0.0.0.0", port=port)