Update app.py

app.py

@@ -832,107 +832,85 @@ class VisualConsensusEngine:
             return f"Error extracting OpenFloor research result: {str(e)}"
     
     def show_research_starting(self, function: str, query: str):
-        """Show research request initiation with enhanced messaging"""
-        session = get_or_create_session_state(self.session_id)
-        current_state = session["roundtable_state"]
-        all_messages = list(current_state.get("messages", []))
-        participants = current_state.get("participants", [])
-        
-        existing_bubbles = list(current_state.get("showBubbles", []))
-        # Ensure both Research Agent AND the calling model stay visible
-        if "Research Agent" not in existing_bubbles:
-            existing_bubbles.append("Research Agent")
-        # Keep the current speaker (the one who requested research) visible
-        current_speaker = current_state.get("currentSpeaker")
-        if current_speaker and current_speaker not in existing_bubbles and current_speaker != "Research Agent":
-            existing_bubbles.append(current_speaker)
-        
-        # Enhanced messages based on function type
-        function_descriptions = {
-            "search_web": "🌐 Web Search - Real-time information",
-            "search_wikipedia": "📚 Wikipedia - Authoritative encyclopedia",
-            "search_academic": "🎓 Academic Research - Peer-reviewed papers",
-            "search_technology_trends": "💻 Technology Trends - GitHub analysis",
-            "search_financial_data": "💰 Financial Data - SEC filings",
-            "multi_source_research": "🔬 Deep Research - Multiple sources"
+        """Invite specific research agent to join conversation"""
+        function_to_agent = {
+            "search_web": "web_search",
+            "search_wikipedia": "wikipedia", 
+            "search_academic": "arxiv",
+            "search_technology_trends": "github",
+            "search_financial_data": "sec_edgar"
         }
         
-        function_desc = function_descriptions.get(function, function.replace('_', ' ').title())
-        
-        estimated_time = self.estimate_research_time(function)
-        message = {
-            "speaker": "Research Agent",
-            "text": f"🔍 **Initiating Research**\n{function_desc}\n📝 Query: \"{query}\"\n⏰ Estimated time: {estimated_time}\n⏳ Connecting to data sources...",
-            "type": "research_starting"
-        }
-        all_messages.append(message)
-        
-        self.update_visual_state({
-            "participants": participants,
-            "messages": all_messages,
-            "currentSpeaker": None,
-            "thinking": [],
-            "showBubbles": existing_bubbles + ["Research Agent"]
-        })
-        time.sleep(0.5)
+        if function in function_to_agent:
+            agent_name = function_to_agent[function]
+            # Use the existing invite method
+            self.invite_research_agent(agent_name, "current_conversation", "AI Expert")
+            
+            # Add the query information
+            research_agent = self.research_agents[agent_name]
+            agent_display_name = research_agent.manifest.identification.conversationalName
+            
+            session = get_or_create_session_state(self.session_id)
+            current_state = session["roundtable_state"]
+            all_messages = list(current_state.get("messages", []))
+            
+            # Add research starting message
+            start_message = {
+                "speaker": agent_display_name,
+                "text": f"🔍 **Starting Research**\n📝 Query: \"{query}\"\n⏳ Connecting to data sources...",
+                "type": "research_starting"
+            }
+            all_messages.append(start_message)
+            
+            existing_bubbles = list(current_state.get("showBubbles", []))
+            self.update_visual_state({
+                "participants": current_state.get("participants", []),
+                "messages": all_messages,
+                "currentSpeaker": agent_display_name,
+                "thinking": [],
+                "showBubbles": existing_bubbles
+            })
+
 
     def show_research_complete(self, function: str, query: str, result_length: int, requesting_model_name: str = None):
-        """Show research ACTUALLY completed with data quality indicators"""
-        session = get_or_create_session_state(self.session_id)
-        current_state = session["roundtable_state"]
-        all_messages = list(current_state.get("messages", []))
-        participants = current_state.get("participants", [])
-        
-        existing_bubbles = list(current_state.get("showBubbles", []))
-        # Ensure both Research Agent AND the calling model stay visible
-        if "Research Agent" not in existing_bubbles:
-            existing_bubbles.append("Research Agent")
-        # Keep the current speaker (the one who requested research) visible
-        current_speaker = current_state.get("currentSpeaker")
-        if current_speaker and current_speaker not in existing_bubbles and current_speaker != "Research Agent":
-            existing_bubbles.append(current_speaker)
-        
-        # Determine data quality based on result length
-        if result_length > 2000:
-            quality_indicator = "📊 High-quality data (comprehensive)"
-            quality_emoji = "🎯"
-        elif result_length > 800:
-            quality_indicator = "📈 Good data (substantial)"
-            quality_emoji = "✅"
-        elif result_length > 200:
-            quality_indicator = "📋 Moderate data (basic)"
-            quality_emoji = "⚠️"
-        else:
-            quality_indicator = "📄 Limited data (minimal)"
-            quality_emoji = "⚡"
-        
-        # Function-specific completion messages
-        function_summaries = {
-            "search_web": "Live web data retrieved",
-            "search_wikipedia": "Encyclopedia articles found",
-            "search_academic": "Academic papers analyzed",
-            "search_technology_trends": "Tech trends mapped",
-            "search_financial_data": "Financial reports accessed",
-            "multi_source_research": "Multi-source synthesis complete"
-        }
-        
-        function_summary = function_summaries.get(function, "Research complete")
-        
-        message = {
-            "speaker": "Research Agent",
-            "text": f"✅ **Research Complete**\n🔬 {function.replace('_', ' ').title()}\n📝 Query: \"{query}\"\n{quality_emoji} {function_summary}\n{quality_indicator}\n📊 {result_length:,} characters analyzed\n🎯 Data ready for expert analysis",
-            "type": "research_complete"
+        """Show research complete and dismiss the specific agent"""
+        function_to_agent = {
+            "search_web": "web_search",
+            "search_wikipedia": "wikipedia", 
+            "search_academic": "arxiv",
+            "search_technology_trends": "github",
+            "search_financial_data": "sec_edgar"
         }
-        all_messages.append(message)
         
-        self.update_visual_state({
-            "participants": participants,
-            "messages": all_messages,
-            "currentSpeaker": requesting_model_name,
-            "thinking": [],
-            "showBubbles": existing_bubbles + ["Research Agent"]
-        })
-        time.sleep(1.5)  # Longer pause to show the detailed completion
+        if function in function_to_agent:
+            agent_name = function_to_agent[function]
+            research_agent = self.research_agents[agent_name]
+            agent_display_name = research_agent.manifest.identification.conversationalName
+            
+            session = get_or_create_session_state(self.session_id)
+            current_state = session["roundtable_state"]
+            all_messages = list(current_state.get("messages", []))
+            
+            # Show completion message
+            complete_message = {
+                "speaker": agent_display_name,
+                "text": f"✅ **Research Complete**\n📊 {result_length:,} characters analyzed\n🎯 Research delivered to {requesting_model_name}",
+                "type": "research_complete"
+            }
+            all_messages.append(complete_message)
+            
+            existing_bubbles = list(current_state.get("showBubbles", []))
+            self.update_visual_state({
+                "participants": current_state.get("participants", []),
+                "messages": all_messages,
+                "currentSpeaker": requesting_model_name,
+                "thinking": [],
+                "showBubbles": existing_bubbles
+            })
+            time.sleep(2)
+            
+            # Use the existing dismiss method
+            self.dismiss_research_agent(agent_name, "current_conversation")
 
     def estimate_research_time(self, function_name: str) -> str:
         """Provide time estimates for different research functions"""
@@ -978,34 +956,51 @@ class VisualConsensusEngine:
         })
         time.sleep(1)
 
-    def update_research_progress(self, progress_text: str):
-        """Update research progress in real-time - ALWAYS REMOVE RESEARCH AGENT FROM THINKING"""
+    def update_research_progress(self, progress_text: str, function_name: str = None):
+        """Update research progress from the specific active research agent"""
+        
+        # Map function to agent to identify which agent is providing progress
+        function_to_agent = {
+            "search_web": "web_search",
+            "search_wikipedia": "wikipedia", 
+            "search_academic": "arxiv",
+            "search_technology_trends": "github",
+            "search_financial_data": "sec_edgar"
+        }
+        
+        # Determine which research agent is active
+        if function_name and function_name in function_to_agent:
+            agent_name = function_to_agent[function_name]
+            research_agent = self.research_agents[agent_name]
+            agent_display_name = research_agent.manifest.identification.conversationalName
+        else:
+            # Fallback to generic research agent if function not specified
+            agent_display_name = "Research Agent"
+        
         session = get_or_create_session_state(self.session_id)
         current_state = session["roundtable_state"]
         all_messages = list(current_state.get("messages", []))
         participants = current_state.get("participants", [])
         
+        # Ensure the specific research agent is visible
         existing_bubbles = list(current_state.get("showBubbles", []))
-        if "Research Agent" not in existing_bubbles:
-            existing_bubbles.append("Research Agent")
+        if agent_display_name not in existing_bubbles:
+            existing_bubbles.append(agent_display_name)
         
+        # Add progress message from the specific agent
         progress_message = {
-            "speaker": "Research Agent",
+            "speaker": agent_display_name,
             "text": f"🔄 {progress_text}",
             "type": "research_progress"
         }
         all_messages.append(progress_message)
         
-        # Get current thinking and ALWAYS remove Research Agent
-        current_thinking = list(current_state.get("thinking", []))
-        if "Research Agent" in current_thinking:
-            current_thinking.remove("Research Agent")
-        
+        # Keep the agent active and visible during progress
         self.update_visual_state({
             "participants": participants,
             "messages": all_messages,
-            "currentSpeaker": None,
-            "thinking": current_thinking,  # Research Agent NEVER in thinking
+            "currentSpeaker": agent_display_name,
+            "thinking": [],
             "showBubbles": existing_bubbles
         })
         time.sleep(0.3)