Update ui/components.py

ui/components.py

@@ -1,5 +1,5 @@
 """
-UI components for the 5-tab demo - FIXED VERSION
+UI components for the 5-tab demo - COMPLETE FIXED VERSION
 """
 
 import gradio as gr
@@ -7,11 +7,11 @@ from typing import Dict, List, Any, Optional, Tuple
 import plotly.graph_objects as go
 
 
-def create_header(oss_version: str, oss_available: bool) -> gr.HTML:  # CHANGED from gr.Markdown to gr.HTML
+def create_header(oss_version: str, oss_available: bool) -> gr.HTML:
     """Create the demo header - FIXED VERSION"""
     status_badge = "✅ Connected" if oss_available else "⚠️ Mock Mode"
     
-    return gr.HTML(f"""  # CHANGED from gr.Markdown to gr.HTML
+    return gr.HTML(f"""
     <div style="text-align: center; padding: 30px 20px 20px 20px; background: linear-gradient(135deg, #f8fafc 0%, #ffffff 100%); border-radius: 0 0 20px 20px; margin-bottom: 30px; border-bottom: 3px solid #4ECDC4;">
         <h1 style="margin-bottom: 10px;">🚀 Agentic Reliability Framework</h1>
         <h2 style="color: #4a5568; font-weight: 600; margin-bottom: 20px;">Investor Demo v3.8.0</h2>
@@ -152,7 +152,7 @@ def create_tab1_incident_demo(scenarios: Dict, default_scenario: str = "Cache Mi
             approval_display, oss_results_display, enterprise_results_display)
 
 
-def create_tab2_business_roi() -> Tuple:
+def create_tab2_business_roi(scenarios: Dict) -> Tuple:
     """Create Tab 2: Business Impact & ROI - FIXED VERSION"""
     with gr.Column():
         gr.Markdown("### 📊 Executive Dashboard")
@@ -161,10 +161,10 @@ def create_tab2_business_roi() -> Tuple:
         gr.Markdown("### 🧮 ROI Calculator")
         with gr.Row():
             with gr.Column(scale=1):
-                # Scenario selector
+                # Scenario selector - FIXED: Initialize with scenarios
                 roi_scenario_dropdown = gr.Dropdown(
-                    choices=[],
-                    value="",
+                    choices=list(scenarios.keys()),
+                    value="Cache Miss Storm",
                     label="Select scenario for ROI calculation:",
                     interactive=True
                 )
@@ -200,7 +200,7 @@ def create_tab2_business_roi() -> Tuple:
 
 
 def create_tab3_enterprise_features() -> Tuple:
-    """Create Tab 3: Enterprise Features"""
+    """Create Tab 3: Enterprise Features - UPDATED"""
     with gr.Row():
         # Left Column
         with gr.Column(scale=1):
@@ -210,8 +210,9 @@ def create_tab3_enterprise_features() -> Tuple:
                 value={
                     "status": "Active",
                     "tier": "Enterprise",
-                    "expires": "2024-12-31",
-                    "features": ["autonomous_healing", "compliance", "audit_trail"]
+                    "expires": "2026-12-31",
+                    "features": ["autonomous_healing", "compliance", "audit_trail", 
+                               "predictive_analytics", "multi_cloud", "role_based_access"]
                 },
                 label="Current License"
             )
@@ -253,6 +254,10 @@ def create_tab3_enterprise_features() -> Tuple:
                     ["Multi-Cloud Support", "❌", "✅"],
                     ["Audit Trail", "Basic", "Comprehensive"],
                     ["Role-Based Access", "❌", "✅"],
+                    ["Custom Dashboards", "❌", "✅"],
+                    ["Enterprise Support", "Community", "24/7 SLA"],
+                    ["Custom Integrations", "❌", "✅"],
+                    ["Advanced Analytics", "❌", "✅"]
                 ],
                 label="",
                 interactive=False
@@ -261,14 +266,23 @@ def create_tab3_enterprise_features() -> Tuple:
             gr.Markdown("### 🔗 Integrations")
             
             integrations_table = gr.Dataframe(
-                headers=["Platform", "Status"],
+                headers=["Platform", "Status", "Type"],
                 value=[
-                    ["AWS", "✅ Connected"],
-                    ["Azure", "✅ Connected"],
-                    ["GCP", "✅ Connected"],
-                    ["Datadog", "✅ Connected"],
-                    ["PagerDuty", "✅ Connected"],
-                    ["ServiceNow", "✅ Connected"],
+                    ["AWS", "✅ Connected", "Cloud"],
+                    ["Azure", "✅ Connected", "Cloud"],
+                    ["GCP", "✅ Connected", "Cloud"],
+                    ["Datadog", "✅ Connected", "Monitoring"],
+                    ["PagerDuty", "✅ Connected", "Alerting"],
+                    ["ServiceNow", "✅ Connected", "ITSM"],
+                    ["Slack", "✅ Connected", "Collaboration"],
+                    ["Teams", "✅ Connected", "Collaboration"],
+                    ["GitHub", "✅ Connected", "DevOps"],
+                    ["GitLab", "✅ Connected", "DevOps"],
+                    ["Jira", "✅ Connected", "Project Management"],
+                    ["Splunk", "✅ Connected", "Monitoring"],
+                    ["New Relic", "✅ Connected", "APM"],
+                    ["Prometheus", "✅ Connected", "Metrics"],
+                    ["Elasticsearch", "✅ Connected", "Logging"]
                 ],
                 label="",
                 interactive=False
@@ -279,7 +293,31 @@ def create_tab3_enterprise_features() -> Tuple:
 
 
 def create_tab4_audit_trail() -> Tuple:
-    """Create Tab 4: Audit Trail & History"""
+    """Create Tab 4: Audit Trail & History - WITH DEMO DATA"""
+    # Demo data
+    demo_executions = [
+        ["14:30", "Cache Miss Storm", "Autonomous", "✅ Success", "$7,225", "Auto-execution"],
+        ["14:15", "Database Connection Pool", "Approval", "✅ Success", "$3,570", "Approved by admin"],
+        ["13:45", "Memory Leak", "Advisory", "⚠️ Analysis", "$0", "OSS analysis only"],
+        ["13:20", "Cache Miss Storm", "Autonomous", "✅ Success", "$7,225", "Pattern match"],
+        ["12:50", "API Rate Limit", "Approval", "✅ Success", "$3,230", "Scheduled fix"],
+        ["12:15", "Network Partition", "Autonomous", "✅ Success", "$10,200", "Emergency response"],
+        ["11:40", "Storage I/O", "Advisory", "⚠️ Analysis", "$0", "Performance review"]
+    ]
+    
+    demo_incidents = [
+        ["14:30", "redis_cache", "Cache Miss Storm", "CRITICAL", "Resolved"],
+        ["14:15", "postgresql", "Database Connection Pool", "HIGH", "Resolved"],
+        ["13:45", "java_service", "Memory Leak", "HIGH", "Analyzed"],
+        ["13:20", "redis_cache", "Cache Miss Storm", "CRITICAL", "Resolved"],
+        ["12:50", "api_gateway", "API Rate Limit", "MEDIUM", "Resolved"],
+        ["12:15", "database", "Network Partition", "CRITICAL", "Resolved"],
+        ["11:40", "storage", "Storage I/O", "HIGH", "Analyzed"],
+        ["11:10", "redis_cache", "Cache Performance", "LOW", "Monitoring"],
+        ["10:45", "load_balancer", "Traffic Spike", "MEDIUM", "Auto-scaled"],
+        ["10:20", "api_gateway", "Rate Limit", "MEDIUM", "Resolved"]
+    ]
+    
     with gr.Row():
         # Left Column
         with gr.Column(scale=1):
@@ -292,7 +330,7 @@ def create_tab4_audit_trail() -> Tuple:
             
             execution_table = gr.Dataframe(
                 headers=["Time", "Scenario", "Mode", "Status", "Savings", "Details"],
-                value=[],
+                value=demo_executions,
                 label="",
                 interactive=False
             )
@@ -303,7 +341,7 @@ def create_tab4_audit_trail() -> Tuple:
             
             incident_table = gr.Dataframe(
                 headers=["Time", "Component", "Scenario", "Severity", "Status"],
-                value=[],
+                value=demo_incidents,
                 label="",
                 interactive=False
             )
@@ -320,13 +358,39 @@ def create_tab4_audit_trail() -> Tuple:
 
 
 def create_tab5_learning_engine() -> Tuple:
-    """Create Tab 5: Learning Engine"""
+    """Create Tab 5: Learning Engine - WITH DEMO DATA"""
+    # Demo data
+    demo_search_results = [
+        ["Cache Miss Storm", "92%", "Scale Redis + Circuit Breaker", "✅ Auto-healed"],
+        ["Database Connection", "85%", "Increase pool + Monitoring", "✅ Approved"],
+        ["Memory Leak Pattern", "78%", "Heap analysis + Restart", "⚠️ Advisory"],
+        ["API Rate Limit", "72%", "Backoff + Queue", "✅ Auto-healed"],
+        ["Network Partition", "65%", "Quorum + Consensus", "✅ Emergency"]
+    ]
+    
+    # Create a simple demo graph
+    fig = go.Figure(data=go.Scatter(
+        x=[1, 2, 3, 4, 5],
+        y=[2, 5, 3, 8, 7],
+        mode='markers+text',
+        marker=dict(size=[20, 30, 25, 40, 35], color=['#FF6B6B', '#4ECDC4', '#45B7D1', '#FFE66D', '#9B59B6']),
+        text=['Cache', 'DB', 'Memory', 'API', 'Network'],
+        textposition="top center"
+    ))
+    
+    fig.update_layout(
+        title="Incident Pattern Relationships",
+        height=400,
+        paper_bgcolor="rgba(0,0,0,0)",
+        plot_bgcolor="rgba(0,0,0,0)"
+    )
+    
     with gr.Row():
         # Left Column
         with gr.Column(scale=2):
             gr.Markdown("### 🧠 Incident Memory Graph")
             
-            learning_graph = gr.Plot(label="", show_label=False)
+            learning_graph = gr.Plot(value=fig, label="", show_label=False)
             
             with gr.Row():
                 graph_type = gr.Radio(
@@ -352,7 +416,7 @@ def create_tab5_learning_engine() -> Tuple:
             
             search_results = gr.Dataframe(
                 headers=["Incident", "Similarity", "Resolution", "Actions"],
-                value=[],
+                value=demo_search_results,
                 label="",
                 interactive=False
             )
@@ -363,10 +427,12 @@ def create_tab5_learning_engine() -> Tuple:
             
             stats_display = gr.JSON(
                 value={
-                    "total_incidents": 0,
-                    "patterns_detected": 0,
-                    "similarity_searches": 0,
-                    "confidence_threshold": 0.85
+                    "total_incidents": 42,
+                    "patterns_detected": 6,
+                    "similarity_searches": 128,
+                    "confidence_threshold": 0.85,
+                    "successful_predictions": 38,
+                    "accuracy_rate": "90.5%"
                 },
                 label="Statistics"
             )
@@ -374,7 +440,14 @@ def create_tab5_learning_engine() -> Tuple:
             gr.Markdown("### 🎯 Pattern Detection")
             
             patterns_display = gr.JSON(
-                value={},
+                value={
+                    "cache_miss_storm": {"occurrences": 12, "confidence": 0.92, "auto_heal": True},
+                    "db_connection_exhaustion": {"occurrences": 8, "confidence": 0.88, "auto_heal": True},
+                    "memory_leak_java": {"occurrences": 5, "confidence": 0.85, "auto_heal": False},
+                    "api_rate_limit": {"occurrences": 10, "confidence": 0.91, "auto_heal": True},
+                    "network_partition": {"occurrences": 3, "confidence": 0.79, "auto_heal": True},
+                    "storage_io_saturation": {"occurrences": 4, "confidence": 0.86, "auto_heal": False}
+                },
                 label="Detected Patterns"
             )
             
@@ -382,9 +455,12 @@ def create_tab5_learning_engine() -> Tuple:
             
             performance_display = gr.JSON(
                 value={
-                    "avg_resolution_time": "0 min",
-                    "success_rate": "0%",
-                    "auto_heal_rate": "0%"
+                    "avg_resolution_time": "8.2 min",
+                    "success_rate": "95.2%",
+                    "auto_heal_rate": "78.6%",
+                    "mttr_reduction": "85%",
+                    "cost_savings": "$1.2M",
+                    "roi_multiplier": "5.2×"
                 },
                 label="Performance Metrics"
             )
@@ -393,9 +469,9 @@ def create_tab5_learning_engine() -> Tuple:
             clear_btn, search_results, stats_display, patterns_display, performance_display)
 
 
-def create_footer() -> gr.HTML:  # CHANGED from gr.Markdown to gr.HTML
+def create_footer() -> gr.HTML:
     """Create the demo footer - UPDATED FOR 2026"""
-    return gr.HTML("""  # CHANGED from gr.Markdown to gr.HTML
+    return gr.HTML("""
     <div style="margin-top: 40px; padding: 30px; background: linear-gradient(135deg, #1a365d 0%, #2d3748 100%); border-radius: 20px; color: white;">
         <div style="border-top: 1px solid #4a5568; padding-top: 20px; text-align: center; color: #a0aec0; font-size: 0.9rem;">
             <p style="margin: 0;">© 2026 Agentic Reliability Framework. Demo v3.8.0 Enterprise Edition.</p>