Update app.py

app.py

@@ -7,9 +7,6 @@ from voice_agent import CrewVoiceAgent
 app = FastAPI()
 agent = CrewVoiceAgent()
 
-# --------------------------
-# FRONTEND
-# --------------------------
 @app.get("/")
 async def home():
     return HTMLResponse("""
@@ -22,10 +19,10 @@ async def home():
 
     <script>
     let ws;
-    let recorder;
+    let rec;
 
     async function start() {
-        ws = new WebSocket("ws://" + location.host + "/ws");   // HF Spaces requires ws://
+        ws = new WebSocket("ws://" + location.host + "/ws");  // HF requires ws://
 
         ws.onmessage = (event) => {
             const audio = new Audio("data:audio/wav;base64," + event.data);
@@ -33,45 +30,42 @@ async def home():
         };
 
         const stream = await navigator.mediaDevices.getUserMedia({ audio: true });
-        recorder = new MediaRecorder(stream, { mimeType: "audio/webm" });
+        rec = new MediaRecorder(stream, { mimeType: "audio/webm" });
 
-        recorder.ondataavailable = async (e) => {
-            const arrayBuf = await e.data.arrayBuffer();
-            ws.send(arrayBuf);
+        rec.ondataavailable = async (e) => {
+            ws.send(await e.data.arrayBuffer());
         };
 
-        recorder.start(300); // send audio chunks
-        document.getElementById("status").textContent = "Listening...";
+        rec.start(500);
+        document.getElementById("status").innerText = "Listening…";
     }
 
     function stop() {
-        recorder.stop();
+        rec.stop();
         ws.close();
-        document.getElementById("status").textContent = "Stopped";
+        document.getElementById("status").innerText = "Stopped";
     }
     </script>
     </body>
     </html>
     """)
 
-# --------------------------
-# WEBSOCKET
-# --------------------------
 @app.websocket("/ws")
-async def websocket_endpoint(ws: WebSocket):
+async def websocket(ws: WebSocket):
     await ws.accept()
-
     while True:
         try:
-            audio_bytes = await ws.receive_bytes()     # Receive raw binary audio
-            audio_reply = await agent.handle_audio(audio_bytes)
+            data = await ws.receive_bytes()
+            audio = await agent.handle_audio(data)
+
+            if not audio:
+                continue
 
-            b64_audio = base64.b64encode(audio_reply).decode()
-            await ws.send_text(b64_audio)
+            b64 = base64.b64encode(audio).decode()
+            await ws.send_text(b64)
 
         except Exception:
             break
 
-
 if __name__ == "__main__":
     uvicorn.run(app, host="0.0.0.0", port=7860)