Upload folder using huggingface_hub

app.py

@@ -15,7 +15,7 @@ from fastrtc import (
     Stream,
     WebRTCError,
     get_twilio_turn_credentials,
-    stt,
+    get_stt_model,
 )
 from gradio.utils import get_space
 from pydantic import BaseModel
@@ -29,6 +29,7 @@ client = openai.OpenAI(
     api_key=os.environ.get("SAMBANOVA_API_KEY"),
     base_url="https://api.sambanova.ai/v1",
 )
+stt_model = get_stt_model()
 
 
 def response(
@@ -39,7 +40,7 @@ def response(
     gradio_chatbot = gradio_chatbot or []
     conversation_state = conversation_state or []
 
-    text = stt(audio)
+    text = stt_model.stt(audio)
     sample_rate, array = audio
     gradio_chatbot.append(
         {"role": "user", "content": gr.Audio((sample_rate, array.squeeze()))}

index.html

@@ -183,8 +183,6 @@
             top: 20px;
             left: 50%;
             transform: translateX(-50%);
-            background-color: #f44336;
-            color: white;
             padding: 16px 24px;
             border-radius: 4px;
             font-size: 14px;
@@ -192,6 +190,16 @@
             display: none;
             box-shadow: 0 2px 5px rgba(0, 0, 0, 0.2);
         }
+
+        .toast.error {
+            background-color: #f44336;
+            color: white;
+        }
+
+        .toast.warning {
+            background-color: #ffd700;
+            color: black;
+        }
     </style>
 </head>
 
@@ -279,6 +287,7 @@
         function showError(message) {
             const toast = document.getElementById('error-toast');
             toast.textContent = message;
+            toast.className = 'toast error';
             toast.style.display = 'block';
 
             // Hide toast after 5 seconds
@@ -319,6 +328,18 @@
             const config = __RTC_CONFIGURATION__;
             peerConnection = new RTCPeerConnection(config);
 
+            const timeoutId = setTimeout(() => {
+                const toast = document.getElementById('error-toast');
+                toast.textContent = "Connection is taking longer than usual. Are you on a VPN?";
+                toast.className = 'toast warning';
+                toast.style.display = 'block';
+
+                // Hide warning after 5 seconds
+                setTimeout(() => {
+                    toast.style.display = 'none';
+                }, 5000);
+            }, 5000);
+
             try {
                 const stream = await navigator.mediaDevices.getUserMedia({
                     audio: true
@@ -352,6 +373,11 @@
 
                 peerConnection.addEventListener('connectionstatechange', () => {
                     console.log('connectionstatechange', peerConnection.connectionState);
+                    if (peerConnection.connectionState === 'connected') {
+                        clearTimeout(timeoutId);
+                        const toast = document.getElementById('error-toast');
+                        toast.style.display = 'none';
+                    }
                     updateButtonState();
                 });
 
@@ -387,6 +413,7 @@
                     addMessage(eventJson.message.role, eventJson.audio ?? eventJson.message.content);
                 });
             } catch (err) {
+                clearTimeout(timeoutId);
                 console.error('Error setting up WebRTC:', err);
                 showError('Failed to establish connection. Please try again.');
                 stop();