Code fixing

app.py

@@ -9,11 +9,13 @@ import urllib.request
 import torchaudio
 from scipy.spatial.distance import cosine
 from RealtimeSTT import AudioToTextRecorder
+from fastapi import FastAPI
 from fastrtc import Stream, AsyncStreamHandler, ReplyOnPause
 import json
 import io
 import wave
 import asyncio
+import uvicorn
 
 # Simplified configuration parameters
 SILENCE_THRESHS = [0, 0.4]
@@ -621,7 +623,7 @@ def setup_fastrtc_handler():
 
 # Create Gradio interface
 def create_interface():
-    with gr.Blocks(title="Real-time Speaker Diarization", theme=gr.themes.Monochrome()) as app:
+    with gr.Blocks(title="Real-time Speaker Diarization", theme=gr.themes.Monochrome()) as interface:
         gr.Markdown("# 🎤 Real-time Speech Recognition with Speaker Diarization")
         gr.Markdown("This app performs real-time speech recognition with automatic speaker identification and color-coding.")
         
@@ -639,7 +641,15 @@ def create_interface():
                     <script>
                         document.getElementById('start-fastrtc').addEventListener('click', function() {
                             document.getElementById('fastrtc-status').textContent = 'Connecting...';
-                            // FastRTC will be initialized here by the middleware
+                            // FastRTC will initialize the connection
+                            fetch('/start-rtc', { method: 'POST' })
+                                .then(response => response.text())
+                                .then(data => {
+                                    document.getElementById('fastrtc-status').textContent = 'Connected! Speak now...';
+                                })
+                                .catch(error => {
+                                    document.getElementById('fastrtc-status').textContent = 'Connection error: ' + error;
+                                });
                         });
                     </script>
                 </div>
@@ -790,29 +800,25 @@ def create_interface():
             outputs=[conversation_output, status_output]
         )
     
-    return app
+    return interface
 
 
-async def main():
-    # Setup FastRTC stream
-    stream = setup_fastrtc_handler()
-    
-    # Create Gradio app
-    app = create_interface()
-    
-    # Mount FastRTC stream to the Gradio app
-    # stream.mount(app)
-    # Mount FastRTC stream to the underlying FastAPI app
-    stream.mount(app.app)
-    
-    # Launch the app
-    app.launch(
-        server_name="0.0.0.0",
-        server_port=7860,
-        share=True
-    )
+# Create FastAPI app and set up routes
+app = FastAPI()
+
+# Initialize FastRTC stream
+rtc_stream = setup_fastrtc_handler()
+rtc_stream.mount(app)
+
+# Create Gradio interface
+gradio_interface = create_interface()
+app = gr.mount_gradio_app(app, gradio_interface, path="/")
 
+# Add endpoint to start RTC connection
+@app.post("/start-rtc")
+def start_rtc():
+    return {"status": "success"}
 
 if __name__ == "__main__":
-    # Run the async application
-    asyncio.run(main())
+    # Run with uvicorn
+    uvicorn.run(app, host="0.0.0.0", port=7860)