Update app.py

app.py

@@ -1,6 +1,7 @@
 """
 🚀 ARF Ultimate Investor Demo v3.8.0 - ENTERPRISE EDITION
 MODULAR VERSION - Properly integrated with all components
+FIXED VERSION: Import and method call fixes applied
 """
 
 import logging
@@ -29,7 +30,7 @@ logger = logging.getLogger(__name__)
 sys.path.insert(0, str(Path(__file__).parent))
 
 # ===========================================
-# IMPORT MODULAR COMPONENTS
+# IMPORT MODULAR COMPONENTS - FIXED IMPORTS
 # ===========================================
 try:
     # Import scenarios
@@ -38,8 +39,8 @@ try:
     # Import orchestrator
     from demo.orchestrator import DemoOrchestrator
     
-    # Import ROI calculator
-    from core.calculators import ROI_Calculator
+    # Import ROI calculator - FIXED: Use EnhancedROICalculator instead of ROI_Calculator
+    from core.calculators import EnhancedROICalculator
     
     # Import visualizations
     from core.visualizations import EnhancedVisualizationEngine
@@ -104,6 +105,21 @@ class AuditTrailManager:
             for e in self.incidents[:15]
         ]
 
+# ===========================================
+# SCENARIO IMPACT MAPPING
+# ===========================================
+def get_scenario_impact(scenario_name: str) -> float:
+    """Get average impact for a given scenario"""
+    impact_map = {
+        "Cache Miss Storm": 8500,
+        "Database Connection Pool Exhaustion": 4200,
+        "Kubernetes Memory Leak": 5500,
+        "API Rate Limit Storm": 3800,
+        "Network Partition": 12000,
+        "Storage I/O Saturation": 6800
+    }
+    return impact_map.get(scenario_name, 5000)
+
 # ===========================================
 # CREATE DEMO INTERFACE - MODULAR VERSION
 # ===========================================
@@ -112,9 +128,9 @@ def create_demo_interface():
     
     import gradio as gr
     
-    # Initialize components
+    # Initialize components - FIXED: Use EnhancedROICalculator
     viz_engine = EnhancedVisualizationEngine()
-    roi_calculator = ROI_Calculator()
+    roi_calculator = EnhancedROICalculator()
     audit_manager = AuditTrailManager()
     orchestrator = DemoOrchestrator()
     
@@ -283,15 +299,20 @@ def create_demo_interface():
             outputs=[approval_display, enterprise_results_display, execution_table]
         )
         
-        # Calculate ROI
+        # Calculate ROI - FIXED: Use correct method from EnhancedROICalculator
         def calculate_roi(scenario_name, monthly_incidents, team_size):
-            # Use ROI calculator
-            roi_result = roi_calculator.calculate_scenario_roi(
-                scenario_name, monthly_incidents, team_size
+            # Get scenario-specific impact
+            avg_impact = get_scenario_impact(scenario_name)
+            
+            # Use the correct method from EnhancedROICalculator
+            roi_result = roi_calculator.calculate_comprehensive_roi(
+                monthly_incidents=int(monthly_incidents),
+                avg_impact=avg_impact,
+                team_size=int(team_size)
             )
             
-            # Create chart
-            chart = roi_calculator.create_comparison_chart(scenario_name)
+            # Create chart using visualization engine
+            chart = viz_engine.create_executive_dashboard(roi_result)
             
             return roi_result, chart