Update app.py

app.py

@@ -79,7 +79,7 @@ class ExtensionManager:
     
     def __init__(self):
         self.extensions: Dict[str, BaseExtension] = {}
-        self.extension_interactions: Dict[str, List[str]] = {}  # Track which extensions work together
+        self.extension_interactions: Dict[str, List[str]] = {}
         self.load_extensions()
     
     def load_extensions(self):
@@ -117,21 +117,18 @@ class ExtensionManager:
                 traceback.print_exc()
         
         log(f"📊 Total extensions loaded: {len(self.extensions)}")
-        
-        # Detect extension relationships
         self._detect_extension_relationships()
     
     def _detect_extension_relationships(self):
         """Detect which extensions can work together based on capabilities"""
         log("🔗 Detecting extension relationships...")
         
-        # Common patterns of extensions that work well together
         relationships = {
-            'yfinance': ['visualization', 'deep_research'],  # Finance + viz/research
-            'deep_research': ['visualization', 'yfinance'],  # Research + viz/finance
-            'youtube': ['deep_research', 'visualization'],    # YouTube + research/viz
-            'timer': ['deep_research'],                       # Timer + research
-            'visualization': ['yfinance', 'deep_research', 'youtube']  # Viz works with data sources
+            'yfinance': ['visualization', 'deep_research'],
+            'deep_research': ['visualization', 'yfinance'],
+            'youtube': ['deep_research', 'visualization'],
+            'timer': ['deep_research'],
+            'visualization': ['yfinance', 'deep_research', 'youtube']
         }
         
         for ext_name, compatible in relationships.items():
@@ -162,20 +159,15 @@ class ExtensionManager:
         if enabled_exts:
             prompt += "\n\n# ENABLED EXTENSIONS\nYou currently have these extensions enabled:\n\n"
             for ext in enabled_exts:
-                # Initialize state for user if needed
                 ext.initialize_state(user_id)
-                
-                # Get extension context
                 prompt += f"## {ext.display_name}\n{ext.get_system_context()}\n\n"
                 
-                # Add current state summary if extension provides it
                 state = ext.get_state(user_id)
                 if state and hasattr(ext, 'get_state_summary'):
                     state_summary = ext.get_state_summary(user_id)
                     if state_summary:
                         prompt += f"**Current State:** {state_summary}\n\n"
             
-            # Add multi-extension reasoning guidance
             if len(enabled_exts) > 1:
                 prompt += self._build_multi_extension_guidance(enabled_list)
         
@@ -186,7 +178,6 @@ class ExtensionManager:
         guidance = "\n\n# MULTI-EXTENSION REASONING\n"
         guidance += "You have multiple extensions enabled. You can combine them intelligently:\n\n"
         
-        # Detect enabled combinations
         combinations = []
         
         if 'yfinance' in enabled_list and 'visualization' in enabled_list:
@@ -228,7 +219,6 @@ class ExtensionManager:
         
         suggestions = []
         
-        # If finance extension returned data with arrays, suggest visualization
         if current_extension == 'yfinance' and isinstance(result, dict):
             if 'dates' in result and 'close_prices' in result and 'visualization' in compatible:
                 suggestions.append({
@@ -245,7 +235,6 @@ class ExtensionManager:
                     'data_available': True
                 })
         
-        # If research completed, suggest visualization of findings
         if current_extension == 'deep_research' and isinstance(result, dict):
             if result.get('success') and 'visualization' in compatible:
                 suggestions.append({
@@ -255,7 +244,6 @@ class ExtensionManager:
                     'data_available': True
                 })
         
-        # If YouTube analysis completed, suggest research on topic
         if current_extension == 'youtube' and isinstance(result, dict):
             if result.get('success') and 'deep_research' in compatible:
                 suggestions.append({
@@ -291,12 +279,10 @@ class ExtensionManager:
             
             log(f"🔧 Executing: {function_name}({json.dumps(args, indent=2)[:100]}...)")
             
-            # Find which extension owns this function
             handled = False
             for ext_name in enabled_list:
                 ext = self.get_extension(ext_name)
                 if ext:
-                    # Check if this function is in the extension's tools
                     for tool in ext.get_tools():
                         if hasattr(tool, 'function_declarations'):
                             for func_decl in tool.function_declarations:
@@ -343,13 +329,12 @@ class ImprovedOrchestrator:
         self.user_id = user_id
         self.enabled_extensions = enabled_extensions
         self.search_chat = client.chats.create(model="gemini-2.5-flash")
-        self.execution_history = []  # Track what's been done
+        self.execution_history = []
         
     def analyze_query(self, query: str, file_parts: List = None) -> Dict[str, Any]:
         """Analyze query to determine optimal execution strategy"""
         log("🧠 Analyzing query to determine strategy...")
         
-        # Check for explicit extension requests first
         query_lower = query.lower()
         explicit_extensions = []
         
@@ -409,14 +394,12 @@ Respond in JSON format:
             
             if response.candidates and response.candidates[0].content and response.candidates[0].content.parts:
                 text = response.candidates[0].content.parts[0].text
-                # Extract JSON from markdown code blocks if present
                 json_match = re.search(r'```(?:json)?\s*(\{.*?\})\s*```', text, re.DOTALL)
                 if json_match:
                     text = json_match.group(1)
                 
                 analysis = json.loads(text)
                 
-                # Force explicit extensions into the analysis
                 if explicit_extensions:
                     analysis['explicit_extensions'] = explicit_extensions
                     analysis['multi_step'] = True
@@ -427,7 +410,6 @@ Respond in JSON format:
         except Exception as e:
             log(f"⚠️ Query analysis failed: {e}, using heuristics")
         
-        # Fallback to heuristics with explicit extensions
         heuristic = self._heuristic_analysis(query)
         if explicit_extensions:
             heuristic['explicit_extensions'] = explicit_extensions
@@ -438,19 +420,15 @@ Respond in JSON format:
         """Fallback heuristic analysis"""
         query_lower = query.lower()
         
-        # Check for search indicators
         search_keywords = ['search', 'find', 'look up', 'what is', 'latest', 'current', 'news']
         needs_search = "yes" if any(kw in query_lower for kw in search_keywords) else "maybe"
         
-        # Check for multi-step indicators
         multi_step_keywords = ['then', 'after that', 'and then', 'research and', 'analyze and']
         multi_step = any(kw in query_lower for kw in multi_step_keywords)
         
-        # Check for context references
         context_keywords = ['the', 'my', 'that', 'previous', 'earlier']
         uses_context = any(kw in query_lower for kw in context_keywords)
         
-        # Determine complexity
         word_count = len(query.split())
         if word_count < 10 and not multi_step:
             complexity = "simple"
@@ -469,33 +447,27 @@ Respond in JSON format:
         }
     
     def execute_with_planning(self, query: str, file_parts: List = None, 
-                            reasoning_budget: int = -1) -> Tuple[str, List, List]:
+                            reasoning_budget: int = -1) -> Tuple[str, List, List, str, Optional[str]]:
         """Execute query with planning and adaptive strategy"""
         
-        # Step 1: Analyze query
         analysis = self.analyze_query(query, file_parts)
         
-        # Step 2: Execute based on analysis
         search_results = ""
         search_citations = None
         
-        # Handle search if needed
         if analysis['needs_search'] in ['yes', 'maybe']:
             log("🔍 Executing search phase...")
             search_results, search_citations = self.call_search_agent(query, file_parts)
         
-        # Step 3: Execute with tools
         log("🛠️ Executing tool phase...")
         tool_results = []
         generated_images = []
         thoughts = ""
         
-        # Build context-aware prompt
         system_prompt = self.extension_manager.build_system_prompt(
             self.user_id, self.enabled_extensions
         )
         
-        # Enhanced instructions for multi-step tasks
         if analysis['multi_step']:
             system_prompt += """
 
@@ -515,7 +487,6 @@ CONTEXT AWARENESS:
 - Reference specific items by their IDs or names from the state
 """
         
-        # Add explicit extension requirements
         if analysis.get('explicit_extensions'):
             explicit_exts = analysis['explicit_extensions']
             system_prompt += f"""
@@ -535,7 +506,6 @@ YOU MUST use these extensions in your response:
             
             system_prompt += "\n\nDo NOT skip these explicitly requested extensions!\n"
 
-        # Multi-turn tool execution
         max_rounds = 5 if analysis['complexity'] == 'complex' else 3
         current_round = 0
         
@@ -556,11 +526,9 @@ YOU MUST use these extensions in your response:
             if not function_calls:
                 log(f"✅ No more tools needed after round {current_round}")
                 if text_response:
-                    # Agent provided final answer
                     return text_response, tool_results, generated_images, thoughts, search_citations
                 break
             
-            # Execute function calls
             results = self.extension_manager.handle_function_calls(
                 self.user_id, self.enabled_extensions, function_calls
             )
@@ -568,17 +536,17 @@ YOU MUST use these extensions in your response:
             for (tool_name, result) in results:
                 tool_results.append((tool_name, result))
                 
-                # Extract generated images IMMEDIATELY
+                # CRITICAL: Extract generated images IMMEDIATELY
                 if isinstance(result, dict) and 'image_base64' in result:
-                    generated_images.append({
+                    img_data = {
                         'base64': result['image_base64'],
                         'title': result.get('message', 'Generated visualization'),
                         'filepath': result.get('filepath', '')
-                    })
+                    }
+                    generated_images.append(img_data)
                     log(f"📊 Captured visualization: {result.get('message', 'Chart')}")
+                    log(f"   🖼️ Image data length: {len(result['image_base64'])} chars")
                 
-                # Check if we should suggest follow-up extension
-                # Find which extension this tool belongs to
                 for ext_name in self.enabled_extensions:
                     ext = self.extension_manager.get_extension(ext_name)
                     if ext:
@@ -586,7 +554,6 @@ YOU MUST use these extensions in your response:
                             if hasattr(tool, 'function_declarations'):
                                 for func_decl in tool.function_declarations:
                                     if func_decl.name == tool_name:
-                                        # Found the extension, check for suggestions
                                         suggestion = self.extension_manager.suggest_next_extension(
                                             ext_name, result, self.enabled_extensions
                                         )
@@ -594,24 +561,20 @@ YOU MUST use these extensions in your response:
                                             log(f"💡 Suggestion: Use {suggestion['extension']}.{suggestion['tool']} - {suggestion['reason']}")
                                         break
             
-            # Prepare for next round
             if current_round < max_rounds:
                 results_summary = self._format_results_for_context(results)
                 
-                # Build enhanced context with extension-aware guidance
                 next_prompt = f"""Previous actions completed:
 {results_summary}
 
 Original query: {query}
 
 """
-                # Add suggestions for next steps based on results
                 suggestions = self._generate_next_step_suggestions(results)
                 if suggestions:
                     next_prompt += f"\n**⚠️ REQUIRED NEXT STEPS (DO NOT SKIP):**\n{suggestions}\n\n"
                     next_prompt += "You MUST complete these steps before finishing. The user explicitly requested these actions.\n\n"
                 
-                # Check if explicit extensions still need to be satisfied
                 if analysis.get('explicit_extensions'):
                     unsatisfied = self._check_unsatisfied_extensions(
                         analysis['explicit_extensions'], 
@@ -633,22 +596,21 @@ Original query: {query}
                 next_prompt += "Continue with required steps. If all user requirements are satisfied, provide final answer."
                 
                 prompt = next_prompt
-                
-                # Don't pass file_parts again in subsequent rounds
                 file_parts = None
         
-        # Check if we have a final text response from tool agent
         if text_response and not function_calls:
-            # Agent provided final answer without needing synthesis
             log(f"✅ Tool agent provided final answer")
             return text_response, tool_results, generated_images, thoughts, search_citations
         
-        # Step 4: Synthesize
+        log(f"🎨 Before synthesis: {len(generated_images)} images in list")
+        
         final_answer, synth_images = self.synthesize_response(
             query, search_results, tool_results, search_citations, file_parts
         )
         
+        log(f"🎨 After synthesis: {len(synth_images)} new images")
         generated_images.extend(synth_images)
+        log(f"🎨 Total images to return: {len(generated_images)}")
         
         return final_answer, tool_results, generated_images, thoughts, search_citations
     
@@ -657,9 +619,8 @@ Original query: {query}
         formatted = []
         for tool_name, result in results:
             if isinstance(result, dict):
-                # Clean up result for context
                 clean_result = dict(result)
-                clean_result.pop('image_base64', None)  # Remove large data
+                clean_result.pop('image_base64', None)
                 formatted.append(f"- {tool_name}: {json.dumps(clean_result, indent=2)[:300]}")
             else:
                 formatted.append(f"- {tool_name}: {str(result)[:200]}")
@@ -673,36 +634,28 @@ Original query: {query}
             if not isinstance(result, dict):
                 continue
             
-            # Stock history → visualization
             if tool_name == 'get_stock_history' and 'visualization' in self.enabled_extensions:
                 if result.get('success') and 'dates' in result and 'close_prices' in result:
                     suggestions.append("📈 MUST create a line chart with the stock history data (dates and prices are ready)")
             
-            # Stock comparison → visualization  
             if tool_name == 'compare_stocks' and 'visualization' in self.enabled_extensions:
                 if result.get('success') and 'comparison' in result:
                     suggestions.append("📊 MUST create a bar chart comparing the stocks (comparison data is ready)")
             
-            # Deep research completed → get stock history if visualization requested
             if tool_name == 'conduct_deep_research' and 'yfinance' in self.enabled_extensions:
                 if result.get('success'):
-                    # Extract company/stock mentioned in research
                     topic = result.get('topic', '')
-                    # Common stock keywords
                     if any(word in topic.lower() for word in ['stock', 'nvidia', 'nvda', 'company']):
                         ticker = 'NVDA' if 'nvidia' in topic.lower() or 'nvda' in topic.lower() else None
                         if ticker and 'visualization' in self.enabled_extensions:
                             suggestions.append(f"📈 MUST get {ticker} stock history for past 4 years (user requested graph of growth)")
             
-            # Deep research → visualization
             if tool_name == 'conduct_deep_research' and 'visualization' in self.enabled_extensions:
                 if result.get('success') and result.get('num_searches', 0) > 0:
-                    # Check if there's numerical data to visualize
                     report = result.get('report', '')
                     if 'pros' in report.lower() and 'cons' in report.lower():
                         suggestions.append("📊 Create a bar chart showing pros vs cons count")
             
-            # YouTube → research
             if tool_name == 'analyze_youtube_video' and 'deep_research' in self.enabled_extensions:
                 if result.get('success'):
                     suggestions.append("🔬 Conduct deep research on topics mentioned in the video")
@@ -714,23 +667,20 @@ Original query: {query}
         """Check which explicitly requested extensions haven't been used yet"""
         used_extensions = set()
         
-        # Map tool calls to extensions
         extension_tool_map = {
-            'deep_research': ['conduct_deep_research'],  # Only actual research counts
+            'deep_research': ['conduct_deep_research'],
             'yfinance': ['get_stock_price', 'get_stock_info', 'get_stock_history', 'compare_stocks'],
             'visualization': ['create_line_chart', 'create_bar_chart', 'create_scatter_plot', 'create_pie_chart'],
             'youtube': ['analyze_youtube_video', 'get_video_chapters'],
             'timer': ['set_timer', 'list_timers', 'check_timer', 'cancel_timer']
         }
         
-        # Check which extensions have been used
         for tool_name, _ in tool_results:
             for ext_name, tool_list in extension_tool_map.items():
                 if tool_name in tool_list:
                     used_extensions.add(ext_name)
                     break
         
-        # Return extensions that were requested but not used
         unsatisfied = []
         for ext in explicit_extensions:
             if ext not in used_extensions:
@@ -845,7 +795,7 @@ Original query: {query}
         generated_images = []
         deep_research_report = None
         deep_research_sources = None
-        has_deep_research = False  # Track if deep research was done
+        has_deep_research = False
         
         if tool_results:
             synthesis_prompt += "[Tool Execution Results]\n"
@@ -856,10 +806,10 @@ Original query: {query}
                         'title': result.get('message', 'Generated visualization'),
                         'filepath': result.get('filepath', '')
                     })
+                    log(f"📊 Synthesis: Captured image from {tool_name}")
                     result_clean = dict(result)
                     result_clean.pop('image_base64', None)
                     synthesis_prompt += f"- {tool_name}: {result_clean.get('message', 'Chart created')}\n"
-                # Capture deep research report
                 elif tool_name == 'conduct_deep_research' and isinstance(result, dict):
                     if result.get('success') and result.get('report'):
                         has_deep_research = True
@@ -871,7 +821,6 @@ Original query: {query}
                     result_str = str(result)[:500]
                     synthesis_prompt += f"- {tool_name}: {result_str}\n"
         
-        # MODIFIED: Different instructions based on whether deep research was done
         if has_deep_research:
             synthesis_prompt += "\n\nProvide a brief summary of what was accomplished (2-3 sentences). DO NOT include the full research report - it will be displayed separately below your summary."
         else:
@@ -880,7 +829,7 @@ Original query: {query}
         config = types.GenerateContentConfig(
             system_instruction="You are a synthesis specialist. Combine information from multiple sources into coherent, helpful responses. Be concise when deep research reports are available since they will be displayed separately.",
             temperature=0.7,
-            max_output_tokens=2048  # Reduced since we're not including full report
+            max_output_tokens=2048
         )
         
         try:
@@ -901,12 +850,10 @@ Original query: {query}
                     if getattr(part, "text", None):
                         result_text += part.text
             
-            # MODIFIED: Only append research report if it exists
             if has_deep_research and deep_research_report:
                 result_text += "\n\n---\n\n## 📋 Complete Research Report\n\n"
                 result_text += deep_research_report
                 
-                # Add sources at the end
                 if deep_research_sources:
                     result_text += "\n\n### 📚 Research Sources\n\n"
                     for idx, source in enumerate(deep_research_sources[:30], 1):
@@ -918,6 +865,7 @@ Original query: {query}
                     if len(deep_research_sources) > 30:
                         result_text += f"\n*...and {len(deep_research_sources) - 30} more sources*\n"
             
+            log(f"🎨 Synthesis returning {len(generated_images)} images")
             return result_text, generated_images
         except Exception as e:
             log(f"⚠️ Synthesis error: {e}")
@@ -968,7 +916,6 @@ def insert_citations_from_grounding(candidates):
         return None
 
 
-# Global instances
 EXTENSION_MANAGER = ExtensionManager()
 CHAT_SESSIONS: Dict[str, Dict[str, Any]] = {}
 
@@ -1018,7 +965,6 @@ def chat_with_gemini(api_key, chat_history_msgs, multimodal_input, show_thoughts
     if chat_history_msgs is None:
         chat_history_msgs = []
     
-    # Process uploaded files
     file_parts = []
     if uploaded_files:
         log(f"📎 Processing {len(uploaded_files)} uploaded file(s)...")
@@ -1034,7 +980,6 @@ def chat_with_gemini(api_key, chat_history_msgs, multimodal_input, show_thoughts
 
     assistant_base_index = len(chat_history_msgs)
     
-    # Setup thinking display
     if show_thoughts:
         thought_index = assistant_base_index
         chat_history_msgs.append({"role": "assistant", "content": "<em>💭 Thinking...</em>"})
@@ -1048,7 +993,6 @@ def chat_with_gemini(api_key, chat_history_msgs, multimodal_input, show_thoughts
     yield chat_history_msgs
 
     try:
-        # Check for proactive messages from extensions first
         proactive_msgs = EXTENSION_MANAGER.check_proactive_messages(api_key, enabled_extensions)
         if proactive_msgs:
             for msg in proactive_msgs:
@@ -1056,7 +1000,6 @@ def chat_with_gemini(api_key, chat_history_msgs, multimodal_input, show_thoughts
                 answer_index += 1
             yield chat_history_msgs
         
-        # Use improved orchestrator
         if enabled_extensions:
             log("🎭 Using improved orchestrator")
             orchestrator = ImprovedOrchestrator(
@@ -1065,11 +1008,14 @@ def chat_with_gemini(api_key, chat_history_msgs, multimodal_input, show_thoughts
             
             budget = reasoning_budget(reasoning_level)
             
-            # Execute with planning
             final_answer, tool_results, generated_images, thoughts, search_citations = \
                 orchestrator.execute_with_planning(user_text, file_parts, budget)
             
-            # Show thoughts
+            # CRITICAL DEBUG: Log what we got back
+            log(f"🎨 Orchestrator returned {len(generated_images)} images")
+            for idx, img in enumerate(generated_images, 1):
+                log(f"   Image {idx}: {img.get('title', 'Untitled')} - base64 length: {len(img.get('base64', ''))}")
+            
             if thoughts and show_thoughts:
                 chat_history_msgs[thought_index]["content"] = (
                     f"<details open>"  
@@ -1081,7 +1027,6 @@ def chat_with_gemini(api_key, chat_history_msgs, multimodal_input, show_thoughts
                 )
                 yield chat_history_msgs
             
-            # Build final content
             final_content = (
                 f"<div><strong>🤖 Response</strong>"
                 f"<div style='white-space:pre-wrap;background:inherit;color:inherit;"
@@ -1089,20 +1034,28 @@ def chat_with_gemini(api_key, chat_history_msgs, multimodal_input, show_thoughts
                 f"{final_answer.strip()}</div></div>"
             )
             
-            # Add generated images
+            # CRITICAL: Add generated images with debug logging
+            log(f"🎨 Attempting to add {len(generated_images)} images to final content")
             if generated_images:
-                for img_data in generated_images:
+                for idx, img_data in enumerate(generated_images, 1):
+                    log(f"  📊 Adding image {idx}: {img_data.get('title', 'Untitled')}")
                     final_content += f"\n\n<div style='margin-top:16px;'>"
-                    final_content += f"<strong>📊 {img_data['title']}</strong><br/>"
-                    final_content += f"<img src='data:image/png;base64,{img_data['base64']}' style='max-width:100%;border-radius:8px;box-shadow:0 2px 8px rgba(0,0,0,0.1);'/>"
-                    if img_data['filepath']:
+                    final_content += f"<strong>📊 {img_data.get('title', 'Chart')}</strong><br/>"
+                    
+                    base64_data = img_data.get('base64', '')
+                    if base64_data:
+                        log(f"     ✅ Base64 data present: {len(base64_data)} chars")
+                        final_content += f"<img src='data:image/png;base64,{base64_data}' style='max-width:100%;border-radius:8px;box-shadow:0 2px 8px rgba(0,0,0,0.1);'/>"
+                    else:
+                        log(f"     ❌ WARNING: No base64 data for image {idx}!")
+                    
+                    if img_data.get('filepath'):
                         final_content += f"<br/><small style='color:#666;'>Saved to: {img_data['filepath']}</small>"
                     final_content += "</div>"
+                log(f"  ✅ All {len(generated_images)} images added to HTML")
+            else:
+                log(f"  ⚠️ WARNING: generated_images list is EMPTY!")
             
-            # REMOVED: The duplicate research report block that was here (lines 1000-1070)
-            # Research report is now handled ONLY in synthesize_response() method
-            
-            # Add citations
             if search_citations:
                 final_content += "\n\n" + search_citations
             
@@ -1110,7 +1063,6 @@ def chat_with_gemini(api_key, chat_history_msgs, multimodal_input, show_thoughts
             yield chat_history_msgs
             
         else:
-            # No extensions - simple streaming with search
             log("📺 Using simple streaming mode")
             
             parts = []
@@ -1178,7 +1130,6 @@ def chat_with_gemini(api_key, chat_history_msgs, multimodal_input, show_thoughts
                             )
                             yield chat_history_msgs
             
-            # Add citations
             if last_chunk:
                 citations = insert_citations_from_grounding(last_chunk.candidates)
                 if citations:
@@ -1260,7 +1211,6 @@ with gr.Blocks(
                 info="Display reasoning process before answers.",
             )
             
-            # Build extension checkboxes
             extension_checkboxes = build_extension_ui()
 
         with gr.Column(scale=4):
@@ -1284,14 +1234,12 @@ with gr.Blocks(
                     autofocus=True
                 )
     
-    # Hidden state to track enabled extensions
     enabled_extensions_state = gr.State([])
 
     def clear_box():
         return {"text": "", "files": []}
 
     def handle_chat(api_key_input, chat_history_msgs, multimodal_dict, thinking_flag, reasoning_lvl, *extension_states):
-        # Convert extension checkbox states to list of enabled extension names
         enabled = []
         for (ext_name, _), is_enabled in zip(extension_checkboxes, extension_states):
             if is_enabled:
@@ -1309,7 +1257,6 @@ with gr.Blocks(
         if not api_key_input or not enabled_exts:
             return chat_history
         
-        # Check all extensions for proactive messages
         proactive_msgs = EXTENSION_MANAGER.check_proactive_messages(api_key_input, enabled_exts)
         
         if proactive_msgs:
@@ -1322,10 +1269,8 @@ with gr.Blocks(
         
         return chat_history
 
-    # Get just the checkbox components for inputs
     checkbox_components = [cb for _, cb in extension_checkboxes]
     
-    # Main chat submission
     multimodal_msg.submit(
         fn=handle_chat,
         inputs=[api_key, chatbot, multimodal_msg, show_thoughts, reasoning_level] + checkbox_components,
@@ -1333,7 +1278,6 @@ with gr.Blocks(
         queue=True,
     ).then(fn=clear_box, outputs=[multimodal_msg])
     
-    # Background timer check - runs every 10 seconds
     timer_check = gr.Timer(value=10, active=True)
     
     def update_enabled_state(*extension_states):
@@ -1343,7 +1287,6 @@ with gr.Blocks(
                 enabled.append(ext_name)
         return enabled
     
-    # Update enabled extensions state whenever checkboxes change
     for _, cb in extension_checkboxes:
         cb.change(
             fn=update_enabled_state,
@@ -1351,7 +1294,6 @@ with gr.Blocks(
             outputs=[enabled_extensions_state]
         )
     
-    # Timer polling for proactive messages
     timer_check.tick(
         fn=check_timers,
         inputs=[api_key, chatbot, enabled_extensions_state],