Process flow visualizer + key skills [for validation only) V3

app.py

@@ -768,10 +768,10 @@ async def process_message_async(message: str, history: Optional[List], session_i
         
         return error_history, "", reasoning_data, {}, {}, session_id, ""
 
-def process_message(message: str, history: Optional[List], session_id: Optional[str] = None) -> Tuple[List, str, dict, dict, dict, str, str]:
+def process_message(message: str, history: Optional[List], session_id: Optional[str] = None) -> Tuple[List, str, dict, dict, dict, str, str, str, gr.update, str, dict, dict, dict, dict, dict]:
     """
     Synchronous wrapper for async processing
-    Returns (history, empty_string, reasoning_data, performance_data, context_data, session_id, skills_html)
+    Returns (history, empty_string, reasoning_data, performance_data, context_data, session_id, skills_html, skills_content, skills_visible_update, flow_display, flow_stats, performance_metrics, intent_details, synthesis_details, safety_details)
     """
     import asyncio
     
@@ -784,7 +784,29 @@ def process_message(message: str, history: Optional[List], session_id: Optional[
         loop = asyncio.new_event_loop()
         asyncio.set_event_loop(loop)
         result = loop.run_until_complete(process_message_async(message, history, session_id))
-        return result
+        
+        # Extract skills_html from result and determine visibility
+        skills_html = result[6]
+        skills_content, skills_visible = _update_skills_display(skills_html)
+        
+        # Return all required outputs (15 total)
+        return (
+            result[0],  # history
+            result[1],  # empty_string
+            result[2],  # reasoning_data
+            result[3],  # performance_data
+            result[4],  # context_data
+            result[5],  # session_id
+            result[6],  # skills_html
+            skills_content,  # skills_content
+            gr.update(visible=skills_visible),  # skills_visible_update
+            "",  # flow_display
+            {},  # flow_stats
+            {},  # performance_metrics
+            {},  # intent_details
+            {},  # synthesis_details
+            {}   # safety_details
+        )
     except Exception as e:
         logger.error(f"Error in process_message: {e}", exc_info=True)
         error_history = list(history) if history else []
@@ -813,7 +835,24 @@ def process_message(message: str, history: Optional[List], session_id: Optional[
             "confidence_calibration": {"overall_confidence": 0.2, "sync_error": True}
         }
         
-        return error_history, "", reasoning_data, {}, {}, session_id, ""
+        # Return all required outputs for error case
+        return (
+            error_history,  # history
+            "",  # empty_string
+            reasoning_data,  # reasoning_data
+            {},  # performance_data
+            {},  # context_data
+            session_id,  # session_id
+            "",  # skills_html
+            "",  # skills_content
+            gr.update(visible=False),  # skills_visible_update
+            "",  # flow_display
+            {},  # flow_stats
+            {},  # performance_metrics
+            {},  # intent_details
+            {},  # synthesis_details
+            {}   # safety_details
+        )
 
 # Decorate the chat handler with GPU if available
 if SPACES_GPU_AVAILABLE and GPU is not None:
@@ -824,10 +863,8 @@ if SPACES_GPU_AVAILABLE and GPU is not None:
         if not session_id:
             session_id = str(uuid.uuid4())[:8]
         result = process_message(message, history, session_id)
-        # Extract skills_html from result and determine visibility
-        skills_html = result[6]
-        skills_content, skills_visible = _update_skills_display(skills_html)
-        return result[0], result[1], result[2], result[3], result[4], result[5], skills_content, gr.update(visible=skills_visible)
+        # Return all 15 values directly
+        return result
     chat_handler_fn = gpu_chat_handler
 else:
     def chat_handler_wrapper(message, history, session_id=None):
@@ -904,7 +941,7 @@ else:
         if process_flow_available and flow_updates:
             return (
                 result[0], result[1], result[2], result[3], result[4], result[5], 
-                skills_content, gr.update(visible=skills_visible),
+                result[6], skills_content, gr.update(visible=skills_visible),
                 flow_updates.get("flow_display", ""),
                 flow_updates.get("flow_stats", {}),
                 flow_updates.get("performance_metrics", {}),
@@ -913,7 +950,7 @@ else:
                 flow_updates.get("safety_details", {})
             )
         else:
-            return result[0], result[1], result[2], result[3], result[4], result[5], skills_content, gr.update(visible=skills_visible)
+            return result
     chat_handler_fn = chat_handler_wrapper
 
 # Initialize orchestrator on module load

main.py

@@ -18,6 +18,7 @@ try:
     from src.agents.intent_agent import create_intent_agent
     from src.agents.synthesis_agent import create_synthesis_agent
     from src.agents.safety_agent import create_safety_agent
+    from src.agents.skills_identification_agent import create_skills_identification_agent
     from src.config import settings
     from src.llm_router import LLMRouter
     from src.orchestrator_engine import MVPOrchestrator
@@ -34,6 +35,7 @@ except ImportError as e:
     create_intent_agent = lambda x: MockComponent()
     create_synthesis_agent = lambda x: MockComponent()
     create_safety_agent = lambda x: MockComponent()
+    create_skills_identification_agent = lambda x: MockComponent()
     settings = type('Settings', (), {'hf_token': os.getenv('HF_TOKEN', '')})()
     LLMRouter = type('MockRouter', (), {'__init__': lambda self, x: None})
     MVPOrchestrator = type('MockOrchestrator', (), {'__init__': lambda self, x, y, z: None})
@@ -55,7 +57,8 @@ def initialize_components():
         agents = {
             'intent_recognition': create_intent_agent(llm_router),
             'response_synthesis': create_synthesis_agent(llm_router),
-            'safety_check': create_safety_agent(llm_router)
+            'safety_check': create_safety_agent(llm_router),
+            'skills_identification': create_skills_identification_agent(llm_router)
         }
         components['agents'] = agents