Code fixing

app.py

@@ -9,7 +9,7 @@ import urllib.request
 import torchaudio
 from scipy.spatial.distance import cosine
 from RealtimeSTT import AudioToTextRecorder
-from fastapi import FastAPI
+from fastapi import FastAPI, Request
 from fastrtc import Stream, AsyncStreamHandler, ReplyOnPause, get_cloudflare_turn_credentials_async, get_cloudflare_turn_credentials
 import json
 import io
@@ -618,19 +618,16 @@ async def get_cloudflare_credentials():
     # Check if HF_TOKEN is set in environment
     hf_token = os.environ.get("HF_TOKEN")
     
-    # If not set, use a default Hugging Face token if available
+    # If not set, try to get from huggingface_hub
     if not hf_token:
-        # Log a warning that user should set their own token
-        print("Warning: HF_TOKEN environment variable not set. Please set your own Hugging Face token.")
-        # Try to use the Hugging Face token from the environment
-        from huggingface_hub import HfApi
+        print("Warning: HF_TOKEN environment variable not set. Trying to get token from huggingface_hub.")
         try:
+            from huggingface_hub import HfApi
             api = HfApi()
             hf_token = api.token
-            if not hf_token:
-                print("Error: No Hugging Face token available. TURN relay may not work properly.")
-        except:
-            print("Error: Failed to get Hugging Face token. TURN relay may not work properly.")
+        except Exception as e:
+            print(f"Error getting Hugging Face token: {e}")
+            hf_token = None
     
     # Get Cloudflare TURN credentials using the Hugging Face token
     if hf_token:
@@ -639,7 +636,7 @@ async def get_cloudflare_credentials():
         except Exception as e:
             print(f"Error getting Cloudflare TURN credentials: {e}")
     
-    # Fallback to a default configuration that may not work
+    # Fallback configuration if no token
     return {
         "iceServers": [
             {
@@ -655,16 +652,35 @@ def setup_fastrtc_handler():
     handler = DiarizationHandler(diarization_system)
     
     # Get server-side credentials (longer TTL)
-    server_credentials = get_cloudflare_turn_credentials(ttl=360000)
+    server_credentials = None
+    try:
+        hf_token = os.environ.get("HF_TOKEN")
+        if hf_token:
+            server_credentials = get_cloudflare_turn_credentials(hf_token=hf_token, ttl=360000)
+        else:
+            try:
+                from huggingface_hub import HfApi
+                api = HfApi()
+                hf_token = api.token
+                if hf_token:
+                    server_credentials = get_cloudflare_turn_credentials(hf_token=hf_token, ttl=360000)
+            except:
+                print("Could not get server-side credentials. Using client-side only.")
+    except Exception as e:
+        print(f"Error getting server credentials: {e}")
     
+    # Create the Stream with appropriate configuration
     stream = Stream(
         handler=handler, 
         modality="audio", 
         mode="receive",
-        rtc_configuration=get_cloudflare_credentials,  # Async function for client-side credentials
-        server_rtc_configuration=server_credentials    # Server-side credentials with longer TTL
+        rtc_configuration=get_cloudflare_credentials  # Async function for client-side credentials
     )
     
+    # Set server-side credentials if available
+    if server_credentials:
+        stream.server_rtc_configuration = server_credentials
+    
     return stream
 
 
@@ -689,14 +705,23 @@ def create_interface():
                         document.getElementById('start-fastrtc').addEventListener('click', function() {
                             document.getElementById('fastrtc-status').textContent = 'Connecting...';
                             // FastRTC will initialize the connection
-                            fetch('/start-rtc', { method: 'POST' })
-                                .then(response => response.text())
-                                .then(data => {
+                            fetch('/start-rtc', { 
+                                method: 'POST',
+                                headers: {
+                                    'Content-Type': 'application/json'
+                                }
+                            })
+                            .then(response => response.json())
+                            .then(data => {
+                                if (data.status === 'success') {
                                     document.getElementById('fastrtc-status').textContent = 'Connected! Speak now...';
-                                })
-                                .catch(error => {
-                                    document.getElementById('fastrtc-status').textContent = 'Connection error: ' + error;
-                                });
+                                } else {
+                                    document.getElementById('fastrtc-status').textContent = 'Connection error: ' + data.error;
+                                }
+                            })
+                            .catch(error => {
+                                document.getElementById('fastrtc-status').textContent = 'Connection error: ' + error;
+                            });
                         });
                     </script>
                 </div>
@@ -872,9 +897,14 @@ rtc_stream.mount(app)
 
 # 5) Expose an endpoint to trigger the client-side RTC handshake
 @app.post("/start-rtc")
-async def start_rtc():
-    await rtc_stream.start_client()
-    return {"status": "success"}
+async def start_rtc(request: Request):
+    try:
+        # Initialize client connection
+        await rtc_stream.start_client()
+        return {"status": "success"}
+    except Exception as e:
+        print(f"Error starting RTC client: {e}")
+        return {"status": "error", "error": str(e)}
 
 # 6) Local dev via uvicorn; HF Spaces will auto-detect 'app' and ignore this
 if __name__ == "__main__":