Update app.py

app.py

@@ -678,6 +678,10 @@ print(f"Search client initialized: {search_client is not None}, API key present:
 # Store web search settings by connection
 web_search_settings = {}
 
+def update_chatbot(chatbot: list[dict], response: ResponseAudioTranscriptDoneEvent):
+    chatbot.append({"role": "assistant", "content": response.transcript})
+    return chatbot
+
 
 class OpenAIHandler(AsyncStreamHandler):
     def __init__(self, web_search_enabled: bool = False, webrtc_id: str = None) -> None:
@@ -698,14 +702,21 @@ class OpenAIHandler(AsyncStreamHandler):
         print(f"Handler created with web_search_enabled={web_search_enabled}, webrtc_id={webrtc_id}")
 
     def copy(self):
-        # Check for stored settings when copying
-        if self.webrtc_id and self.webrtc_id in web_search_settings:
-            web_search_enabled = web_search_settings[self.webrtc_id]
-        else:
-            web_search_enabled = self.web_search_enabled
+        # Get the most recent settings
+        if web_search_settings:
+            # Get the most recent webrtc_id
+            recent_ids = sorted(web_search_settings.keys(), 
+                              key=lambda k: web_search_settings[k].get('timestamp', 0), 
+                              reverse=True)
+            if recent_ids:
+                recent_id = recent_ids[0]
+                settings = web_search_settings[recent_id]
+                web_search_enabled = settings.get('enabled', False)
+                print(f"Handler.copy() using recent settings - webrtc_id={recent_id}, web_search_enabled={web_search_enabled}")
+                return OpenAIHandler(web_search_enabled=web_search_enabled, webrtc_id=recent_id)
         
-        print(f"Handler.copy() called - creating new handler with web_search_enabled={web_search_enabled}")
-        return OpenAIHandler(web_search_enabled=web_search_enabled, webrtc_id=self.webrtc_id)
+        print(f"Handler.copy() called - creating new handler with default settings")
+        return OpenAIHandler(web_search_enabled=False)
 
     async def search_web(self, query: str) -> str:
         """Perform web search and return formatted results"""
@@ -730,6 +741,18 @@ class OpenAIHandler(AsyncStreamHandler):
 
     async def start_up(self):
         """Connect to realtime API with function calling enabled"""
+        # First check if we have the most recent settings
+        if web_search_settings:
+            recent_ids = sorted(web_search_settings.keys(), 
+                              key=lambda k: web_search_settings[k].get('timestamp', 0), 
+                              reverse=True)
+            if recent_ids:
+                recent_id = recent_ids[0]
+                settings = web_search_settings[recent_id]
+                self.web_search_enabled = settings.get('enabled', False)
+                self.webrtc_id = recent_id
+                print(f"start_up: Updated settings from storage - webrtc_id={self.webrtc_id}, web_search_enabled={self.web_search_enabled}")
+        
         print(f"Starting up handler with web_search_enabled={self.web_search_enabled}")
         self.client = openai.AsyncOpenAI()
         
@@ -871,23 +894,15 @@ class OpenAIHandler(AsyncStreamHandler):
             self.connection = None
 
 
-# Factory function to create handler with settings
-def create_handler():
-    """Factory function to create handler - will be customized per connection"""
-    return OpenAIHandler(web_search_enabled=False)
-
-
-def update_chatbot(chatbot: list[dict], response: ResponseAudioTranscriptDoneEvent):
-    chatbot.append({"role": "assistant", "content": response.transcript})
-    return chatbot
-
+# Create initial handler instance
+handler = OpenAIHandler(web_search_enabled=False)
 
 # Create components
 chatbot = gr.Chatbot(type="messages")
 
-# Create initial stream with factory function
+# Create stream with handler instance
 stream = Stream(
-    create_handler,  # Pass factory function instead of instance
+    handler,  # Pass instance, not factory
     mode="send-receive",
     modality="audio",
     additional_inputs=[chatbot],
@@ -900,23 +915,6 @@ stream = Stream(
 
 app = FastAPI()
 
-# Custom handler factory that reads settings
-def custom_handler_factory():
-    """Custom factory that checks for web search settings"""
-    # Try to get the most recent settings
-    if web_search_settings:
-        # Get the most recent webrtc_id
-        recent_id = max(web_search_settings.keys(), key=lambda k: web_search_settings[k].get('timestamp', 0))
-        settings = web_search_settings[recent_id]
-        return OpenAIHandler(
-            web_search_enabled=settings.get('enabled', False),
-            webrtc_id=recent_id
-        )
-    return OpenAIHandler(web_search_enabled=False)
-
-# Replace the handler factory
-stream.handler = custom_handler_factory
-
 # Mount stream
 stream.mount(app)