Update app.py

app.py

@@ -85,7 +85,7 @@ def format_conversation_log(history: List[Dict[str, Any]]) -> str:
         log_lines.append("")
     return "\n".join(log_lines)
 
-def build_master_log(system_log_list, coder_log_list, strategist_log_list, audit_log_list, start_time):
+def build_master_log(system_log_list, coder_log_list, strategist_log_list, audit_log_list, start_time, final_code_block=None):
     actor_styles = {
         'SYSTEM': {'icon': '🔧', 'color': '#ef4444'},
         'CODER': {'icon': '💻', 'color': '#3b82f6'},
@@ -168,6 +168,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
+    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>
+        </div>
+        '''
+        master_entries.append(html_entry)
+
     if not master_entries:
         return '<div style="font-family: monospace; color: #6b7280;">No log data captured.</div>'
     
@@ -179,7 +189,7 @@ 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):
+def execute_remote_workflow(blueprint, intent, coder_model_label, strategist_model_label, enable_audit):
     """Streaming workflow execution with real-time log updates"""
     mission_start_time = time.time()
     
@@ -187,7 +197,8 @@ def execute_remote_workflow(blueprint, intent, coder_model_label, strategist_mod
         "blueprint": blueprint,
         "intent": intent,
         "coder_model_label": coder_model_label,
-        "strategist_model_label": strategist_model_label
+        "strategist_model_label": strategist_model_label,
+        "enable_audit": enable_audit
     }
 
     try:
@@ -201,6 +212,8 @@ 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
@@ -224,12 +237,16 @@ def execute_remote_workflow(blueprint, intent, coder_model_label, strategist_mod
                 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)
+                                               mission_start_time, final_code_content)
                 
                 yield sys_log, cod_log, strat_log, master_log
         
@@ -264,8 +281,36 @@ 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():
-        return DEMO_BLUEPRINT, DEMO_INTENT
+        # 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)
 
     def start_execution_sequence():
         return gr.update(visible=False), gr.update(visible=True), gr.update(visible=False)
@@ -296,8 +341,8 @@ Produce a pro level styled and tuned them GUI that has 1 button that when presse
         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 = 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
+        coder_reset_value = initial_coder_model_val
+        strategist_reset_value = initial_strategist_model_val
         
         return (
             "",  # blueprint_input
@@ -309,7 +354,8 @@ Produce a pro level styled and tuned them GUI that has 1 button that when presse
             gr.update(value=initial_coder_provider),  # coder_provider
             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
+            gr.update(choices=strategist_models_for_reset, value=strategist_reset_value),  # strategist_model
+            False # audit_checkbox
         )
 
     with gr.Blocks(title="Project: Conductor V4.0 Client") as interface:
@@ -329,7 +375,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=coder_default if coder_default else None,
+                    value=initial_coder_model_val,
                     label="Model",
                     interactive=True
                 )
@@ -378,7 +424,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=strategist_default if strategist_default else None,
+                    value=initial_strategist_model_val,
                     label="Model",
                     interactive=True
                 )
@@ -394,6 +440,10 @@ Produce a pro level styled and tuned them GUI that has 1 button that when presse
             execute_btn = gr.Button("EXECUTE VIA MCP", variant="primary")
             reset_btn = gr.Button("RESET MISSION", variant="secondary")
         
+        # 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)
@@ -425,7 +475,7 @@ Produce a pro level styled and tuned them GUI that has 1 button that when presse
         demo_btn.click(
             fn=load_demo_data,
             inputs=None,
-            outputs=[blueprint_input, intent_input]
+            outputs=[blueprint_input, intent_input, coder_model]
         )
         
         # Execute button
@@ -435,7 +485,7 @@ Produce a pro level styled and tuned them GUI that has 1 button that when presse
             outputs=[default_image, working_video, celebrate_video]
         ).then(
             fn=execute_remote_workflow,
-            inputs=[blueprint_input, intent_input, coder_model, strategist_model],
+            inputs=[blueprint_input, intent_input, coder_model, strategist_model, audit_checkbox],
             outputs=[system_terminal, coder_log, strategist_log, master_log_output]
         ).then(
             fn=end_execution_sequence,
@@ -464,12 +514,13 @@ Produce a pro level styled and tuned them GUI that has 1 button that when presse
                 coder_provider,
                 coder_model,
                 strategist_provider,
-                strategist_model
+                strategist_model,
+                audit_checkbox
             ]
         )
         
         gr.Markdown("---")
-        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.")
+        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.")
 
     return interface