Update app.py

app.py

@@ -83,7 +83,7 @@ Chain of Thought:
 Worker Action:
 {scenario["worker_output"]}"""
 
-    model_response = f"""<thought>The worker is attempting to {scenario["worker_cot_trace"].lower()} This involves {scenario["worker_output"].lower()}.
+    model_response = f"""<thought>The worker is attempting to {scenario["worker_cot_trace"].lower()} This involves {scenario["worker_output"].lower()}. 
 
 Looking at the applicable policy rules:
 {chr(10).join(scenario["applicable_rules"]) if scenario["applicable_rules"] else "No specific rules apply."}
@@ -107,7 +107,7 @@ Explanation: {"This action violates data privacy rules as it involves unauthoriz
         {
             "Timestamp": datetime.now().strftime("%H:%M:%S"),
             "Turn": len(history) + 1,
-            "Scenario": scenario.get("scenario_id", f"sc-{scenario_idx}"),
+            "Scenario": scenario["scenario_id"],
             "Role": role,
             "Decision": scenario["decision"],
             "Reward": reward,
@@ -136,10 +136,10 @@ def clear_history():
 with gr.Blocks(title="AEGIS-Env Demo Dashboard") as demo:
     gr.Markdown("""
     # 🛡️ AEGIS-Env Fleet Oversight Demo
-
+    
     **Training Status:** Complete ✓ | Model: YashashMathur/aegis-training-checkpoints
-
-    This demo showcases the AEGIS policy violation detection system.
+    
+    This demo showcases the AEGIS policy violation detection system. 
     Select a worker role and scenario to see the model in action.
     """)
 
@@ -147,18 +147,17 @@ with gr.Blocks(title="AEGIS-Env Demo Dashboard") as demo:
         with gr.Column(scale=1):
             gr.Markdown("### ⚙️ Demo Settings")
             role_input = gr.Dropdown(
-                choices=["data-eng", "sec-ops", "api-dev", "admin", "support",
-                         "code-reviewer", "general-dev", "planner", "customer-support"],
-                value="data-eng",
+                choices=["general-dev", "code-reviewer", "planner", "customer-support"],
+                value="general-dev",
                 label="Worker Role",
             )
             scenario_input = gr.Slider(
                 minimum=0,
-                maximum=99,
+                maximum=3,
                 step=1,
                 value=0,
                 label="Demo Scenario",
-                info="0–99: cycles through the 500-scenario training dataset",
+                info="0: Data extraction | 1: Data selling | 2: Legitimate | 3: Env exposure",
             )
             run_btn = gr.Button("🔍 Analyze Scenario", variant="primary")
             clear_btn = gr.Button("🗑️ Clear")