Update app.py

app.py

@@ -16,7 +16,7 @@ Features:
 - Enhanced error handling and recovery
 - 300s timeout for real-time behavior
 """
-
+import gradio as gr
 import asyncio
 import os
 import time
@@ -436,6 +436,39 @@ def get_connection_status():
     else:
         return f"🔴 Disconnected | Status: {app_state['last_status']}"
 
+
+
+
+
+# --- Async helpers for streaming handlers ---
+
+async def audio_stream_handler(audio):
+    if app_state.get("handler") and app_state.get("connected"):
+        return await app_state["handler"].receive(audio)
+    return None
+
+async def video_stream_handler(frame):
+    if app_state.get("handler") and app_state.get("connected"):
+        return await app_state["handler"].video_receive(frame)
+    return None
+
+async def ai_audio_output_handler():
+    if app_state.get("handler") and app_state.get("connected"):
+        return await app_state["handler"].emit()
+    return None
+
+# --- Backend logic for mic test and screen sharing ---
+
+def backend_mic_test():
+    if app_state.get("handler") and app_state.get("connected"):
+        return "🎙️ Microphone is active and streaming to backend."
+    return "⚠️ Please connect first to test microphone."
+
+def backend_screen_share():
+    if app_state.get("handler") and app_state.get("connected"):
+        return "🖥️ Screen sharing is active and streaming to backend."
+    return "⚠️ Please connect first to share your screen."
+
 def create_interface():
     """PREMIUM: Enhanced interface with complete real-time integration"""
     # Initialize premium stream
@@ -487,20 +520,6 @@ def create_interface():
             mic_test_btn = gr.Button("🎙️ Test Microphone", variant="secondary")
             screen_share_btn = gr.Button("🖥️ Share Screen", variant="secondary")
 
-        # --- Backend logic for mic test and screen sharing ---
-        def backend_mic_test():
-            # Simulate a backend mic test (could be extended to record/playback)
-            if app_state.get("handler") and app_state.get("connected"):
-                return "🎙️ Microphone is active and streaming to backend."
-            return "⚠️ Please connect first to test microphone."
-
-        def backend_screen_share():
-            # Simulate backend screen sharing trigger
-            if app_state.get("handler") and app_state.get("connected"):
-                # In a real implementation, you might set a flag or trigger a backend event
-                return "🖥️ Screen sharing is active and streaming to backend."
-            return "⚠️ Please connect first to share your screen."
-
         # PREMIUM: Real-time streaming interface
         gr.Markdown("### 📡 Premium Real-Time Stream")
 
@@ -520,9 +539,9 @@ def create_interface():
                 interactive=True
             )
 
-            # PREMIUM: Connect streaming handlers
+            # PREMIUM: Connect streaming handlers with async functions
             audio_stream.stream(
-                fn=lambda audio: app_state["handler"].receive(audio) if app_state["handler"] and app_state["connected"] else None,
+                fn=audio_stream_handler,
                 inputs=[audio_stream],
                 outputs=[],
                 time_limit=300,  # Real-time optimized
@@ -530,7 +549,7 @@ def create_interface():
             )
 
             video_stream.stream(
-                fn=lambda frame: app_state["handler"].video_receive(frame) if app_state["handler"] and app_state["connected"] else None,
+                fn=video_stream_handler,
                 inputs=[video_stream],
                 outputs=[],
                 time_limit=300,  # Real-time optimized
@@ -552,9 +571,9 @@ def create_interface():
                 streaming=True
             )
 
-            # Connect AI response handlers
+            # Connect AI response handlers with async function
             ai_audio_output.stream(
-                fn=lambda: app_state["handler"].emit() if app_state["handler"] and app_state["connected"] else None,
+                fn=ai_audio_output_handler,
                 inputs=[],
                 outputs=[ai_audio_output],
                 time_limit=300