Update app.py

app.py

@@ -8,7 +8,7 @@ from typing import List, Dict, Any
 
 MCP_SERVER_URL = "https://mcp-1st-birthday-project-conductor-core-mcp.hf.space/execute"
 
-# HARDCODED PROVIDER MAPPING (Directive 1.1, 1.2, 1.3)
+# HARDCODED PROVIDER MAPPING
 PROVIDER_MODEL_MAPPING = {
     "Google Gemini": [],
     "Anthropic": [],
@@ -86,6 +86,7 @@ def format_conversation_log(history: List[Dict[str, Any]]) -> str:
     return "\n".join(log_lines)
 
 def build_master_log(system_log_list, coder_log_list, strategist_log_list, audit_log_list, start_time, final_code_block=None):
+    """TASK 3: Added final_code_block parameter"""
     actor_styles = {
         'SYSTEM': {'icon': '🔧', 'color': '#ef4444'},
         'CODER': {'icon': '💻', 'color': '#3b82f6'},
@@ -168,16 +169,16 @@ def build_master_log(system_log_list, coder_log_list, strategist_log_list, audit
         master_entries.append(html_entry)
         seq_counter += 1
     
-    # Append Final Code Block if available
+    # TASK 3: Append final code block if provided
     if final_code_block:
-        html_entry = f'''
-        <div style="margin-top: 20px; border-top: 2px solid #10b981; padding-top: 10px;">
-            <h3 style="color: #10b981; font-family: monospace;">🚀 GENERATED CODE:</h3>
-            <pre style="background-color: #000; padding: 10px; border-radius: 5px; overflow-x: auto; color: #a5f3fc;"><code>{final_code_block}</code></pre>
+        code_html = f'''
+        <div style="margin-top: 20px; padding: 16px; background-color: #0d1117; border: 2px solid #10b981; border-radius: 8px;">
+            <h3 style="color: #10b981; margin-top: 0;">🎯 GENERATED CODE:</h3>
+            <pre style="background-color: #1e1e1e; padding: 12px; border-radius: 4px; overflow-x: auto; color: #e0e0e0;"><code>{final_code_block}</code></pre>
         </div>
         '''
-        master_entries.append(html_entry)
-
+        master_entries.append(code_html)
+    
     if not master_entries:
         return '<div style="font-family: monospace; color: #6b7280;">No log data captured.</div>'
     
@@ -190,15 +191,16 @@ def build_master_log(system_log_list, coder_log_list, strategist_log_list, audit
     return html_output
 
 def execute_remote_workflow(blueprint, intent, coder_model_label, strategist_model_label, enable_audit):
-    """Streaming workflow execution with real-time log updates"""
+    """TASK 3 & 4: Added enable_audit parameter and final_code tracking"""
     mission_start_time = time.time()
+    final_code = None
     
     payload = {
         "blueprint": blueprint,
         "intent": intent,
         "coder_model_label": coder_model_label,
         "strategist_model_label": strategist_model_label,
-        "enable_audit": enable_audit
+        "enable_audit": enable_audit  # TASK 4: Add audit flag to payload
     }
 
     try:
@@ -212,8 +214,6 @@ def execute_remote_workflow(blueprint, intent, coder_model_label, strategist_mod
             
             response.raise_for_status()
             
-            final_code_content = None
-
             for line in response.iter_lines():
                 if not line:
                     continue
@@ -232,21 +232,21 @@ def execute_remote_workflow(blueprint, intent, coder_model_label, strategist_mod
                 if data.get("status") == "complete":
                     break
                 
+                # TASK 3: Watch for final_code key
+                if "final_code" in data:
+                    final_code = data["final_code"]
+                
                 system_log_list = data.get("system_log", [])
                 coder_log_list = data.get("coder_log", [])
                 strategist_log_list = data.get("strategist_log", [])
                 audit_log_list = data.get("audit_log", [])
                 
-                # Capture final output if present
-                if data.get("final_output"):
-                    final_code_content = data.get("final_output")
-
                 sys_log = "\n".join(system_log_list)
                 cod_log = format_conversation_log(coder_log_list)
                 strat_log = format_conversation_log(strategist_log_list)
                 master_log = build_master_log(system_log_list, coder_log_list, 
                                                strategist_log_list, audit_log_list, 
-                                               mission_start_time, final_code_content)
+                                               mission_start_time, final_code)  # TASK 3: Pass final_code
                 
                 yield sys_log, cod_log, strat_log, master_log
         
@@ -267,7 +267,7 @@ def create_interface():
 Produce a pro level styled and tuned them GUI that has 1 button that when pressed displays a celebration announcement "Happy Birthday MCP!". USE specified colors or themes appropriate for the event. You must incorporate specific colors for design pieces and thematic layouts providing a fun and exciting announcement."""
     DEMO_INTENT = """A simple test button with a clean look. That when pressed creates a simple happy birthday MCP response on the screen. The gui should be gradio and contain colors that are used in professional builds and match the Gradio, MCP or other color theme or as close as possible."""
 
-    # Get initial defaults
+    # TASK 1: Use existing defaults from config
     coder_default = CLIENT_CONFIG['coder_default']
     strategist_default = CLIENT_CONFIG['strategist_default']
     
@@ -281,36 +281,20 @@ Produce a pro level styled and tuned them GUI that has 1 button that when presse
     if strategist_default:
         initial_strategist_provider = classify_model_by_provider(strategist_default)
 
-    # TASK 1: Set Default Models on Launch
-    # We override the config default if specific models are requested
-    default_model_label = "gemini-2.0-flash-lite" # Assumes this key exists in config
-    
-    # Helper to safely get a default value
-    def get_safe_default(provider, preferred_label):
-        models = CLIENT_CONFIG['coder_by_provider'].get(provider, [])
-        if preferred_label in models:
-            return preferred_label
-        return models[0] if models else None
-
-    initial_coder_model_val = get_safe_default(initial_coder_provider, default_model_label)
-    initial_strategist_model_val = get_safe_default(initial_strategist_provider, default_model_label)
-
     def load_demo_data():
-        # TASK 2: "Load Demo" Button Model Switch
-        # Switch to a high-end model for the demo
-        demo_model_label = "gemini-2.5-pro" # Assumes this key exists or will be handled
-        # Fallback if exact label not found, try to find a 'pro' model
-        models = CLIENT_CONFIG['coder_by_provider'].get("Google Gemini", [])
-        target_model = demo_model_label
-        if demo_model_label not in models:
-             # Try to find a pro model
-             pro_models = [m for m in models if 'pro' in m.lower()]
-             if pro_models:
-                 target_model = pro_models[0]
-             elif models:
-                 target_model = models[0]
-
-        return DEMO_BLUEPRINT, DEMO_INTENT, gr.update(value=target_model)
+        """TASK 2: Switch Coder to gemini 2.5 pro when demo loaded"""
+        # Find the gemini 2.5 pro model in config
+        demo_coder_model = None
+        for model_label in CLIENT_CONFIG['coder_models']:
+            if '2.5' in model_label and 'pro' in model_label.lower():
+                demo_coder_model = model_label
+                break
+        
+        if demo_coder_model:
+            return DEMO_BLUEPRINT, DEMO_INTENT, gr.update(value=demo_coder_model)
+        else:
+            # Fallback: return without model change if 2.5 pro not found
+            return DEMO_BLUEPRINT, DEMO_INTENT, gr.update()
 
     def start_execution_sequence():
         return gr.update(visible=False), gr.update(visible=True), gr.update(visible=False)
@@ -334,15 +318,14 @@ Produce a pro level styled and tuned them GUI that has 1 button that when presse
             return gr.update(choices=[], value=None)
         return gr.update(choices=models, value=models[0])
     
-    # RESET MISSION function (Directive 2.1)
+    # RESET MISSION function
     def reset_mission():
         """Total system reset: clear all fields and reset to defaults"""
-        # Determine initial models for reset
         coder_models_for_reset = CLIENT_CONFIG['coder_by_provider'].get(initial_coder_provider, [])
         strategist_models_for_reset = CLIENT_CONFIG['strategist_by_provider'].get(initial_strategist_provider, [])
         
-        coder_reset_value = initial_coder_model_val
-        strategist_reset_value = initial_strategist_model_val
+        coder_reset_value = coder_models_for_reset[0] if coder_models_for_reset else None
+        strategist_reset_value = strategist_models_for_reset[0] if strategist_models_for_reset else None
         
         return (
             "",  # blueprint_input
@@ -355,14 +338,14 @@ Produce a pro level styled and tuned them GUI that has 1 button that when presse
             gr.update(choices=coder_models_for_reset, value=coder_reset_value),  # coder_model
             gr.update(value=initial_strategist_provider),  # strategist_provider
             gr.update(choices=strategist_models_for_reset, value=strategist_reset_value),  # strategist_model
-            False # audit_checkbox
+            gr.update(value=False)  # TASK 4: Reset audit checkbox
         )
 
     with gr.Blocks(title="Project: Conductor V4.0 Client") as interface:
         gr.Markdown("# Project: Conductor V4.0 (Client Mode)")
         gr.Markdown(f"### The Command Center: Connected to MCP Server at `{MCP_SERVER_URL}`")
         
-        # PRIMARY INTERFACE ROW - Three Columns (Directive: Command Console Layout)
+        # PRIMARY INTERFACE ROW - Three Columns
         with gr.Row():
             # COLUMN 1 - CODER (Left Flank)
             with gr.Column():
@@ -375,7 +358,7 @@ Produce a pro level styled and tuned them GUI that has 1 button that when presse
                 )
                 coder_model = gr.Dropdown(
                     choices=CLIENT_CONFIG['coder_by_provider'].get(initial_coder_provider, []),
-                    value=initial_coder_model_val,
+                    value=coder_default if coder_default else None,  # TASK 1: Use existing default
                     label="Model",
                     interactive=True
                 )
@@ -424,7 +407,7 @@ Produce a pro level styled and tuned them GUI that has 1 button that when presse
                 )
                 strategist_model = gr.Dropdown(
                     choices=CLIENT_CONFIG['strategist_by_provider'].get(initial_strategist_provider, []),
-                    value=initial_strategist_model_val,
+                    value=strategist_default if strategist_default else None,  # TASK 1: Use existing default
                     label="Model",
                     interactive=True
                 )
@@ -439,11 +422,8 @@ Produce a pro level styled and tuned them GUI that has 1 button that when presse
             demo_btn = gr.Button("LOAD DEMO MODE", variant="secondary")
             execute_btn = gr.Button("EXECUTE VIA MCP", variant="primary")
             reset_btn = gr.Button("RESET MISSION", variant="secondary")
+            audit_checkbox = gr.Checkbox(label="AUDIT", value=False)  # TASK 4: Add audit checkbox
         
-        # TASK 4: Audit Checkbox
-        with gr.Row():
-            audit_checkbox = gr.Checkbox(label="ENABLE AUDIT LOOP (Strategist Review)", value=False)
-
         # LOG ROWS
         with gr.Row():
             system_terminal = gr.Textbox(label="SYSTEM LOG", interactive=False, lines=15, max_lines=15)
@@ -471,14 +451,14 @@ Produce a pro level styled and tuned them GUI that has 1 button that when presse
             outputs=[strategist_model]
         )
         
-        # Demo button
+        # Demo button - TASK 2: Now updates coder_model
         demo_btn.click(
             fn=load_demo_data,
             inputs=None,
             outputs=[blueprint_input, intent_input, coder_model]
         )
         
-        # Execute button
+        # Execute button - TASK 4: Now passes audit_checkbox
         execute_btn.click(
             fn=start_execution_sequence,
             inputs=None,
@@ -500,7 +480,7 @@ Produce a pro level styled and tuned them GUI that has 1 button that when presse
             outputs=[default_image, working_video, celebrate_video]
         )
         
-        # RESET MISSION button (Directive 2.1)
+        # RESET MISSION button - TASK 4: Now includes audit_checkbox in outputs
         reset_btn.click(
             fn=reset_mission,
             inputs=[],
@@ -520,11 +500,11 @@ Produce a pro level styled and tuned them GUI that has 1 button that when presse
         )
         
         gr.Markdown("---")
-        gr.Markdown("**Note:** This client requires the Conductor MCP Server V4.1 with streaming and audit support. Logs now update in real-time with color-coded visual differentiation.")
+        gr.Markdown("**Note:** This client requires the Conductor MCP Server V4.0 with streaming support. Logs now update in real-time with color-coded visual differentiation. **AUDIT mode will loop code back to Strategist for review.**")
 
     return interface
 
 if __name__ == "__main__":
     app = create_interface()
     app.queue()
-    app.launch()
+    app.launch()