Code fixing

app.py

@@ -10,7 +10,7 @@ import torchaudio
 from scipy.spatial.distance import cosine
 from RealtimeSTT import AudioToTextRecorder
 from fastapi import FastAPI
-from fastrtc import Stream, AsyncStreamHandler, ReplyOnPause
+from fastrtc import Stream, AsyncStreamHandler, ReplyOnPause, get_cloudflare_turn_credentials_async, get_cloudflare_turn_credentials
 import json
 import io
 import wave
@@ -613,11 +613,58 @@ def get_status():
     return diarization_system.get_status_info()
 
 
-# Setup FastRTC stream handler
+# Get Cloudflare TURN credentials for FastRTC
+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 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
+        try:
+            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.")
+    
+    # Get Cloudflare TURN credentials using the Hugging Face token
+    if hf_token:
+        try:
+            return await get_cloudflare_turn_credentials_async(hf_token=hf_token)
+        except Exception as e:
+            print(f"Error getting Cloudflare TURN credentials: {e}")
+    
+    # Fallback to a default configuration that may not work
+    return {
+        "iceServers": [
+            {
+                "urls": "stun:stun.l.google.com:19302"
+            }
+        ]
+    }
+
+
+# Setup FastRTC stream handler with TURN server configuration
 def setup_fastrtc_handler():
-    """Set up FastRTC audio stream handler"""
+    """Set up FastRTC audio stream handler with TURN server configuration"""
     handler = DiarizationHandler(diarization_system)
-    stream = Stream(handler=handler, modality="audio", mode="receive")
+    
+    # Get server-side credentials (longer TTL)
+    server_credentials = get_cloudflare_turn_credentials(ttl=360000)
+    
+    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
+    )
+    
     return stream
 
 
@@ -725,6 +772,13 @@ def create_interface():
                 This app uses FastRTC for low-latency audio streaming. 
                 For optimal performance, use a modern browser and allow microphone access when prompted.
                 """)
+                
+                # Hugging Face Token Information
+                gr.Markdown("""
+                ## 🔑 Hugging Face Token
+                This app uses Cloudflare TURN server via Hugging Face integration.
+                If audio connection fails, set your HF_TOKEN environment variable in the Space settings.
+                """)
         
         # Auto-refresh conversation and status
         def refresh_display():