Update demo/orchestrator.py

demo/orchestrator.py

@@ -1,4 +1,4 @@
-# demo/orchestrator.py - FIXED VERSION
+# demo/orchestrator.py - COMPLETE FIXED VERSION
 from __future__ import annotations
 
 import logging
@@ -17,8 +17,9 @@ try:
         calculate_pattern_confidence
     )
     MOCK_ARF_AVAILABLE = True
-except ImportError:
-    logger.warning("Mock ARF functions not available")
+    logger.info("Mock ARF functions available")
+except ImportError as e:
+    logger.warning(f"Mock ARF functions not available: {e}")
     MOCK_ARF_AVAILABLE = False
 
 
@@ -29,6 +30,7 @@ class DemoOrchestrator:
 
     def __init__(self, enable_streamlit: bool = False):
         self.enable_streamlit = enable_streamlit
+        logger.info("DemoOrchestrator initialized")
     
     async def analyze_incident(self, scenario_name: str, scenario_data: Dict[str, Any]) -> Dict[str, Any]:
         """
@@ -38,6 +40,7 @@ class DemoOrchestrator:
         logger.info(f"Analyzing incident: {scenario_name}")
         
         if not MOCK_ARF_AVAILABLE:
+            logger.error("Mock ARF functions not available")
             return {
                 "status": "error",
                 "message": "Mock ARF functions not available",
@@ -46,19 +49,22 @@ class DemoOrchestrator:
         
         try:
             # Step 1: Detection Agent
+            logger.debug("Running detection agent...")
             detection_result = simulate_arf_analysis(scenario_data)
             
             # Step 2: Recall Agent  
+            logger.debug("Running recall agent...")
             similar_incidents = run_rag_similarity_search(scenario_data)
             
             # Step 3: Decision Agent
+            logger.debug("Running decision agent...")
             confidence = calculate_pattern_confidence(scenario_data, similar_incidents)
             healing_intent = create_mock_healing_intent(scenario_data, similar_incidents, confidence)
             
             # Simulate processing time
             await asyncio.sleep(0.5)
             
-            return {
+            result = {
                 "status": "success",
                 "scenario": scenario_name,
                 "detection": detection_result,
@@ -68,8 +74,11 @@ class DemoOrchestrator:
                 "processing_time_ms": 450
             }
             
+            logger.info(f"Analysis complete for {scenario_name}")
+            return result
+            
         except Exception as e:
-            logger.error(f"Error analyzing incident: {e}")
+            logger.error(f"Error analyzing incident: {e}", exc_info=True)
             return {
                 "status": "error",
                 "message": str(e),