Update app.py

app.py

@@ -1,55 +1,118 @@
 import gradio as gr
 from orchestrator import run_workflow
 
-def launch_app(problem, agent_order, bias_mode, human_override, human_text):
-    outputs = run_workflow(problem, agent_order, bias_mode)
 
-    final = outputs["Final Recommendation"]
+def run_system(problem, order, bias, override, human_text):
 
-    if human_override:
-        final = f"HUMAN OVERRIDE APPLIED:\n\n{human_text}"
+    results = run_workflow(problem, order, bias)
+
+    final = results.get("Final", "")
+
+    if override and human_text.strip():
+        final = f"🧑‍⚖️ HUMAN OVERRIDE\n\n{human_text}"
 
     return (
-        outputs.get("Market", ""),
-        outputs.get("Finance", ""),
-        outputs.get("Risk", ""),
-        outputs.get("Ethics", ""),
+        results.get("Market", ""),
+        results.get("Finance", ""),
+        results.get("Risk", ""),
+        results.get("Ethics", ""),
         final
     )
 
-with gr.Blocks(title="AI Strategy Lab") as demo:
-    gr.Markdown("# 🧠 AI Strategy Lab")
-    gr.Markdown("**Multi-Agent AI | Workflow Order | Human-in-the-Loop**")
 
-    problem = gr.Textbox(label="Problem Statement", lines=4)
+# Material-style theme
+theme = gr.themes.Soft(
+    primary_hue="blue",
+    secondary_hue="cyan",
+    radius_size="lg",
+    font=["Inter", "sans-serif"]
+)
+
 
-    agent_order = gr.CheckboxGroup(
-        ["Market", "Finance", "Risk", "Ethics"],
-        value=["Market", "Finance", "Risk", "Ethics"],
-        label="Agent Execution Order (ORDER MATTERS!)"
+with gr.Blocks(theme=theme) as demo:
+
+    gr.Markdown(
+        """
+        # 🧠 AI Strategy Lab
+        ### Multi-Agent Decision System
+
+        Demonstrates:
+        • Agent workflows  
+        • Order effects  
+        • Bias  
+        • Human-in-the-loop  
+        """
     )
 
-    bias_mode = gr.Checkbox(label="Inject Market Optimism Bias")
-    human_override = gr.Checkbox(label="Enable Human Override")
-    human_text = gr.Textbox(label="Human Decision", lines=3)
-
-    run = gr.Button("🚀 Run AI Strategy Lab")
-
-    with gr.Tab("Market Agent"):
-        market_out = gr.Textbox(lines=10)
-    with gr.Tab("Finance Agent"):
-        finance_out = gr.Textbox(lines=10)
-    with gr.Tab("Risk Agent"):
-        risk_out = gr.Textbox(lines=10)
-    with gr.Tab("Ethics Agent"):
-        ethics_out = gr.Textbox(lines=10)
-    with gr.Tab("Final Decision"):
-        final_out = gr.Textbox(lines=12)
-
-    run.click(
-        launch_app,
-        inputs=[problem, agent_order, bias_mode, human_override, human_text],
-        outputs=[market_out, finance_out, risk_out, ethics_out, final_out]
+    with gr.Row():
+
+        with gr.Column(scale=2):
+
+            problem = gr.Textbox(
+                label="📌 Problem Statement",
+                lines=4,
+                placeholder="Should we launch Product X in Market Y?"
+            )
+
+            order = gr.Textbox(
+                label="🔁 Agent Order (comma-separated)",
+                value="Market, Finance, Risk, Ethics",
+                info="Example: Risk, Finance, Market, Ethics"
+            )
+
+            bias = gr.Checkbox(
+                label="Inject Market Bias"
+            )
+
+            override = gr.Checkbox(
+                label="Enable Human Override"
+            )
+
+            human_text = gr.Textbox(
+                label="Human Decision",
+                lines=3
+            )
+
+            run_btn = gr.Button(
+                "🚀 Run Analysis",
+                variant="primary"
+            )
+
+        with gr.Column(scale=3):
+
+            with gr.Tab("📊 Market"):
+                market_out = gr.Textbox(lines=10)
+
+            with gr.Tab("💰 Finance"):
+                finance_out = gr.Textbox(lines=10)
+
+            with gr.Tab("⚠️ Risk"):
+                risk_out = gr.Textbox(lines=10)
+
+            with gr.Tab("🌱 Ethics"):
+                ethics_out = gr.Textbox(lines=10)
+
+            with gr.Tab("✅ Final Decision"):
+                final_out = gr.Textbox(lines=12)
+
+
+    run_btn.click(
+        run_system,
+        inputs=[
+            problem,
+            order,
+            bias,
+            override,
+            human_text
+        ],
+        outputs=[
+            market_out,
+            finance_out,
+            risk_out,
+            ethics_out,
+            final_out
+        ]
     )
 
+
 demo.launch()