Update ui/components.py

ui/components.py

@@ -1,16 +1,18 @@
 """
-Minimalist, math-first Gradio UI for Agentic Reliability Framework (ARF)
-Only Gradio components; Streamlit removed
+Gradio-only UI components for ARF
+Ensures full compatibility with app.py
 """
 
 import gradio as gr
 from typing import Dict, List, Any
+from demo.scenarios import INCIDENT_SCENARIOS
 
 # -----------------------------
 # Header & Status
 # -----------------------------
-def create_header() -> gr.HTML:
-    return gr.HTML("<h2>🚀 Agentic Reliability Framework v3.3.6 (OSS Edition)</h2>")
+def create_header(version="3.3.6", mock_mode=False) -> gr.HTML:
+    mock_text = " · MOCK MODE" if mock_mode else ""
+    return gr.HTML(f"<h2>🚀 Agentic Reliability Framework v{version} (OSS Edition){mock_text}</h2>")
 
 def create_status_bar() -> gr.HTML:
     return gr.HTML(
@@ -20,60 +22,91 @@ def create_status_bar() -> gr.HTML:
 # -----------------------------
 # Tab 1: Incident Demo
 # -----------------------------
-def create_tab1_incident_demo() -> tuple:
-    """
-    Returns 12 Gradio components (matching previous unpacking in app.py)
-    """
-    scenario_dropdown = gr.Dropdown(choices=["Cache Miss Storm", "DB Latency Spike"], label="Incident Scenario")
+def create_tab1_incident_demo(scenarios=INCIDENT_SCENARIOS, default_scenario="Cache Miss Storm") -> tuple:
+    scenario_dropdown = gr.Dropdown(choices=list(scenarios.keys()), value=default_scenario, label="Incident Scenario")
     scenario_description = gr.Textbox(value="Select a scenario to begin analysis.", label="Description")
     metrics_display = gr.Textbox(value="Live Metrics: TBD", label="Metrics")
     impact_display = gr.Textbox(value="Estimated Business Impact: TBD", label="Impact")
     
+    timeline_output = gr.Textbox(value="Incident timeline TBD.", label="Timeline")
+    oss_btn = gr.Button("Run OSS Analysis")
+    enterprise_btn = gr.Button("Execute Enterprise Healing")
+    approval_toggle = gr.Checkbox(label="Require Human Approval?", value=False)
+    demo_btn = gr.Button("Run Demo")
+    
     approval_display = gr.Textbox(value="Approval Status: N/A", label="Approval")
     oss_results_display = gr.Textbox(value="OSS Results: N/A", label="OSS")
     enterprise_results_display = gr.Textbox(value="Enterprise Results: N/A", label="Enterprise")
     
-    run_oss_btn = gr.Button("Run OSS Analysis")
-    execute_enterprise_btn = gr.Button("Execute Enterprise Healing")
-    require_human_btn = gr.Button("Require Human Approval")
-    run_demo_btn = gr.Button("Run Demo")
-    
     return (
         scenario_dropdown, scenario_description, metrics_display, impact_display,
-        approval_display, oss_results_display, enterprise_results_display,
-        run_oss_btn, execute_enterprise_btn, require_human_btn, run_demo_btn, gr.Textbox(value="", visible=False)
+        timeline_output, oss_btn, enterprise_btn, approval_toggle, demo_btn,
+        approval_display, oss_results_display, enterprise_results_display
     )
 
 # -----------------------------
 # Tab 2: Business ROI
 # -----------------------------
-def create_tab2_business_roi() -> tuple:
-    roi_display = gr.Textbox(value="ROI Calculator Output: TBD", label="ROI")
-    return (roi_display,)
+def create_tab2_business_roi(scenarios=INCIDENT_SCENARIOS) -> tuple:
+    dashboard_output = gr.Textbox(value="Dashboard TBD", label="Dashboard")
+    roi_scenario_dropdown = gr.Dropdown(choices=list(scenarios.keys()), value="Cache Miss Storm", label="Scenario")
+    monthly_slider = gr.Slider(minimum=1, maximum=50, value=15, step=1, label="Monthly Incidents")
+    team_slider = gr.Slider(minimum=1, maximum=50, value=5, step=1, label="Team Size")
+    calculate_btn = gr.Button("Calculate ROI")
+    roi_output = gr.Textbox(value="ROI Output TBD", label="ROI")
+    roi_chart = gr.Plot()
+    
+    return (dashboard_output, roi_scenario_dropdown, monthly_slider, team_slider,
+            calculate_btn, roi_output, roi_chart)
 
 # -----------------------------
 # Tab 3: Enterprise Features
 # -----------------------------
 def create_tab3_enterprise_features() -> tuple:
-    features_display = gr.Textbox(
-        value="- Self-Healing Agentic Core\n- Enhanced Monitoring\n- Auto-Scaling",
-        label="Enterprise Features"
-    )
-    return (features_display,)
+    license_display = gr.Textbox(value="License info TBD", label="License")
+    validate_btn = gr.Button("Validate License")
+    trial_btn = gr.Button("Start Trial")
+    upgrade_btn = gr.Button("Upgrade")
+    
+    mcp_mode = gr.Dropdown(choices=["advisory", "approval", "autonomous"], value="advisory", label="MCP Mode")
+    mcp_mode_info = gr.Textbox(value="MCP Mode Info TBD", label="Mode Info")
+    features_table = gr.Dataframe(headers=["Feature", "Status"], value=[["Self-Healing Core", "Active"]])
+    integrations_table = gr.Dataframe(headers=["Integration", "Status"], value=[["Monitoring", "Connected"]])
+    
+    return (license_display, validate_btn, trial_btn, upgrade_btn,
+            mcp_mode, mcp_mode_info, features_table, integrations_table)
 
 # -----------------------------
 # Tab 4: Audit Trail
 # -----------------------------
 def create_tab4_audit_trail() -> tuple:
-    audit_display = gr.Textbox(value="Audit logs will appear here.", label="Audit Trail")
-    return (audit_display,)
+    refresh_btn = gr.Button("Refresh")
+    clear_btn = gr.Button("Clear History")
+    export_btn = gr.Button("Export")
+    
+    execution_table = gr.Dataframe(headers=["Time", "Scenario", "Mode", "Status", "Savings", "Details"])
+    incident_table = gr.Dataframe(headers=["Time", "Component", "Scenario", "Severity", "Status"])
+    export_text = gr.Textbox(value="Export JSON will appear here", label="Export")
+    
+    return (refresh_btn, clear_btn, export_btn, execution_table, incident_table, export_text)
 
 # -----------------------------
 # Tab 5: Learning Engine
 # -----------------------------
 def create_tab5_learning_engine() -> tuple:
-    learning_display = gr.Textbox(value="Learning engine stats TBD.", label="Learning Engine")
-    return (learning_display,)
+    learning_graph = gr.Plot()
+    graph_type = gr.Dropdown(choices=["Graph A", "Graph B"], value="Graph A", label="Graph Type")
+    show_labels = gr.Checkbox(label="Show Labels", value=True)
+    search_query = gr.Textbox(label="Search Patterns")
+    search_btn = gr.Button("Search")
+    clear_btn_search = gr.Button("Clear Search")
+    search_results = gr.Textbox(value="Search Results TBD", label="Results")
+    stats_display = gr.Textbox(value="Stats TBD", label="Stats")
+    patterns_display = gr.Textbox(value="Patterns TBD", label="Patterns")
+    performance_display = gr.Textbox(value="Performance TBD", label="Performance")
+    
+    return (learning_graph, graph_type, show_labels, search_query, search_btn,
+            clear_btn_search, search_results, stats_display, patterns_display, performance_display)
 
 # -----------------------------
 # Footer