Update app.py

app.py

@@ -8,8 +8,9 @@ from character_learner import CharacterLearner
 from audio_handler import AudioHandler
 
 class ConversationalAgent:
-    def __init__(self):
-        self.llm_handler = LLMHandler()
+    def __init__(self, model_name: str | None = None):
+        # Allow dynamic model override from UI; fall back to env / default.
+        self.llm_handler = LLMHandler(model_override=model_name)
         self.memory_manager = MemoryManager()
         self.tool_executor = ToolExecutor()
         self.character_learner = CharacterLearner(self.memory_manager)
@@ -123,12 +124,13 @@ class ConversationalAgent:
         stats.append(f"\n**Total Conversations:** {self.memory_manager.get_conversation_count(self.user_id)}")
         return "\n".join(stats)
 
-# --- REPLACE THIS ENTIRE FUNCTION ---
 def create_interface():
-    """Create and configure the Gradio interface."""
-    agent = ConversationalAgent()
-    
+    """Create and configure the Gradio interface with LLM selection."""
+    # Top-level stateful agent; will be re-created when model changes.
     with gr.Blocks(title="Personal AI Assistant", theme=gr.themes.Soft()) as demo:
+        # Global state: selected model and agent instance
+        model_state = gr.State(os.getenv("PREFERRED_MODEL", "google/gemini-2.0-flash-exp"))
+        agent_state = gr.State(ConversationalAgent(model_state.value))
         gr.Markdown(
             """
             # 🤖 Personal AI Assistant
@@ -182,71 +184,110 @@ def create_interface():
                 autoplay=True,
                 type="numpy"
             )
-            
-            with gr.Row():
+
+            # Settings / model selection + memory stats
+            with gr.Tab("Settings"):
+                gr.Markdown("### 🔧 LLM Settings")
+                with gr.Row():
+                    llm_model = gr.Dropdown(
+                        label="Select LLM model (via OpenRouter)",
+                        choices=[
+                            "google/gemini-2.0-flash-exp",
+                            "anthropic/claude-3.5-sonnet",
+                            "anthropic/claude-3.5-haiku",
+                            "openai/gpt-4.1-mini",
+                        ],
+                        value=os.getenv("PREFERRED_MODEL", "google/gemini-2.0-flash-exp"),
+                    )
+                apply_model_btn = gr.Button("Apply Model")
+
+            with gr.Tab("Memory Stats"):
                 gr.Markdown("### 📊 Memory Stats")
                 stats_display = gr.Markdown("Click 'Refresh Stats' to view")
                 refresh_btn = gr.Button("Refresh Stats 🔄")
-        
-        # --- Define the update_stats function here ---
-        def update_stats():
-            return agent.get_memory_stats()
-        
-        # --- Define all event handlers ---
-        def respond(message, history):
-            return agent.process_message(message, history, use_voice=False)
-        
-        def respond_with_voice(message, history):
-            return agent.process_message(message, history, use_voice=True)
-        
-        def process_audio(audio, history):
-            return agent.process_voice_input(audio, history)
-        
+
+        # --- Helper to (re)build agent when model changes ---
+        def build_agent(model_name):
+            try:
+                return ConversationalAgent(model_name)
+            except Exception as e:
+                # Surface configuration errors (e.g., missing OPENROUTER_API_KEY)
+                print(f"Error initializing ConversationalAgent with model '{model_name}': {e}")
+                # Fallback to default without crashing UI
+                return ConversationalAgent()
+
+        # --- Stats updater uses current agent instance ---
+        def update_stats(agent_obj):
+            return agent_obj.get_memory_stats()
+
+        # --- Core chat handlers using agent_state ---
+        def respond(message, history, agent_obj):
+            return agent_obj.process_message(message, history, use_voice=False)
+
+        def respond_with_voice(message, history, agent_obj):
+            return agent_obj.process_message(message, history, use_voice=True)
+
+        def process_audio(audio, history, agent_obj):
+            return agent_obj.process_voice_input(audio, history)
+
         def clear_history():
             return [], ""
         
-        # --- Correctly wire up events ---
+        # --- Wire chat events to use current agent_state ---
         msg_input.submit(
             respond,
-            inputs=[msg_input, chatbot],
-            outputs=[chatbot, msg_input, audio_output]
+            inputs=[msg_input, chatbot, agent_state],
+            outputs=[chatbot, msg_input, audio_output],
         )
-        
+
         submit_btn.click(
             respond,
-            inputs=[msg_input, chatbot],
-            outputs=[chatbot, msg_input, audio_output]
+            inputs=[msg_input, chatbot, agent_state],
+            outputs=[chatbot, msg_input, audio_output],
         )
-        
+
         voice_btn.click(
             respond_with_voice,
-            inputs=[msg_input, chatbot],
-            outputs=[chatbot, msg_input, audio_output]
+            inputs=[msg_input, chatbot, agent_state],
+            outputs=[chatbot, msg_input, audio_output],
         )
-        
+
         audio_input.change(
             process_audio,
-            inputs=[audio_input, chatbot],
-            outputs=[chatbot, msg_input]
+            inputs=[audio_input, chatbot, agent_state],
+            outputs=[chatbot, msg_input],
         )
-        
+
         clear_btn.click(
             clear_history,
-            outputs=[chatbot, msg_input]
+            outputs=[chatbot, msg_input],
         )
-        
-        # Correctly wire up the stats button
+
+        # Model apply: update model_state + agent_state
+        def apply_model(selected_model, _old_agent):
+            new_agent = build_agent(selected_model)
+            return selected_model, new_agent
+
+        apply_model_btn.click(
+            apply_model,
+            inputs=[llm_model, agent_state],
+            outputs=[model_state, agent_state],
+        )
+
+        # Stats button uses current agent_state
         refresh_btn.click(
-            update_stats, # Use the function
-            outputs=[stats_display] # To the correct output component
+            update_stats,
+            inputs=[agent_state],
+            outputs=[stats_display],
         )
-        
+
         # Load stats on startup
         demo.load(
-            update_stats, # Use the function
-            outputs=[stats_display] # To the correct output component
+            update_stats,
+            inputs=[agent_state],
+            outputs=[stats_display],
         )
-    
+
     return demo
 
 # ... (rest of your file is unchanged)