Update app.py

app.py

@@ -3,6 +3,7 @@ import requests
 import json
 import os
 import time
+import tempfile
 from configparser import ConfigParser
 from typing import List, Dict, Any
 
@@ -47,7 +48,7 @@ def load_client_config():
         except Exception as e:
             print(f"Config parsing warning: {e}")
         
-        # DIRECTIVE #1 FIX: Read defaults from config.ini [API] section
+        # Read defaults from config.ini [API] section
         if 'API' in parser:
             config['coder_default'] = parser['API'].get('coder_model_default_label', '').strip()
             config['strategist_default'] = parser['API'].get('strategist_model_default_label', '').strip()
@@ -162,7 +163,7 @@ def build_master_log(system_log_list, coder_log_list, strategist_log_list, audit
         master_entries.append(html_entry)
         seq_counter += 1
     
-    # DIRECTIVE #2 FIX: Enhanced tool failure logging
+    # Enhanced tool failure logging
     for diagnostic in audit_log_list:
         elapsed = time.time() - start_time
         minutes = int(elapsed // 60)
@@ -272,7 +273,6 @@ def execute_remote_workflow(blueprint, intent, coder_model_label, strategist_mod
                                                strategist_log_list, audit_log_list, 
                                                mission_start_time, final_code)
                 
-                # DIRECTIVE #3: Return final_code as separate output
                 yield sys_log, cod_log, strat_log, master_log, final_code
         
     except requests.exceptions.RequestException as e:
@@ -292,7 +292,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."""
 
-    # DIRECTIVE #1 FIX: Use config defaults explicitly
+    # Use config defaults explicitly
     coder_default = CLIENT_CONFIG['coder_default']
     strategist_default = CLIENT_CONFIG['strategist_default']
     
@@ -310,7 +310,7 @@ Produce a pro level styled and tuned them GUI that has 1 button that when presse
     initial_coder_models = CLIENT_CONFIG['coder_by_provider'].get(initial_coder_provider, [])
     initial_strategist_models = CLIENT_CONFIG['strategist_by_provider'].get(initial_strategist_provider, [])
     
-    # DIRECTIVE #1 FIX: Ensure default value is set or fallback to first available
+    # Ensure default value is set or fallback to first available
     coder_initial_value = coder_default if coder_default and coder_default in initial_coder_models else (initial_coder_models[0] if initial_coder_models else None)
     strategist_initial_value = strategist_default if strategist_default and strategist_default in initial_strategist_models else (initial_strategist_models[0] if initial_strategist_models else None)
 
@@ -330,8 +330,20 @@ Produce a pro level styled and tuned them GUI that has 1 button that when presse
     def start_execution_sequence():
         return gr.update(visible=False), gr.update(visible=True), gr.update(visible=False)
 
+    # ISSUE #1 FIX: Button visibility control in end sequence
     def end_execution_sequence(sys_log, cod_log, strat_log, master_log, final_code):
-        return sys_log, cod_log, strat_log, master_log, final_code, gr.update(visible=False), gr.update(visible=True)
+        # Button becomes visible only when final_code exists
+        button_visibility = final_code is not None and final_code.strip() != ""
+        return (
+            sys_log, 
+            cod_log, 
+            strat_log, 
+            master_log, 
+            final_code, 
+            gr.update(visible=False),  # working_video
+            gr.update(visible=True),   # celebrate_video
+            gr.update(visible=button_visibility)  # download_btn
+        )
 
     def reset_to_default():
         return gr.update(visible=True), gr.update(visible=False), gr.update(visible=False)
@@ -349,17 +361,24 @@ 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])
     
-    # DIRECTIVE #3: Download code function
+    # ISSUE #2 FIX: Download code function with proper file handling
     def download_code(final_code_content):
         """Generate downloadable Python file from final code"""
-        if not final_code_content:
+        if not final_code_content or final_code_content.strip() == "":
             return None
         
         # Create temporary file with timestamp
         timestamp = time.strftime("%Y%m%d_%H%M%S")
         filename = f"generated_code_{timestamp}.py"
         
-        return final_code_content
+        # Write to temporary file and return filepath
+        temp_dir = tempfile.gettempdir()
+        filepath = os.path.join(temp_dir, filename)
+        
+        with open(filepath, 'w', encoding='utf-8') as f:
+            f.write(final_code_content)
+        
+        return filepath
     
     # RESET MISSION function
     def reset_mission():
@@ -382,7 +401,8 @@ Produce a pro level styled and tuned them GUI that has 1 button that when presse
             gr.update(value=initial_strategist_provider),  # strategist_provider
             gr.update(choices=strategist_models_for_reset, value=strategist_reset_value),  # strategist_model
             gr.update(value=False),  # audit_checkbox
-            None  # final_code_state
+            None,  # final_code_state
+            gr.update(visible=False)  # ISSUE #1 FIX: Hide download button on reset
         )
 
     with gr.Blocks(title="Project: Conductor V4.0 Client") as interface:
@@ -405,7 +425,7 @@ Produce a pro level styled and tuned them GUI that has 1 button that when presse
                 )
                 coder_model = gr.Dropdown(
                     choices=initial_coder_models,
-                    value=coder_initial_value,  # DIRECTIVE #1 FIX
+                    value=coder_initial_value,
                     label="Model",
                     interactive=True
                 )
@@ -454,7 +474,7 @@ Produce a pro level styled and tuned them GUI that has 1 button that when presse
                 )
                 strategist_model = gr.Dropdown(
                     choices=initial_strategist_models,
-                    value=strategist_initial_value,  # DIRECTIVE #1 FIX
+                    value=strategist_initial_value,
                     label="Model",
                     interactive=True
                 )
@@ -483,9 +503,13 @@ Produce a pro level styled and tuned them GUI that has 1 button that when presse
                 label="MASTER LOG WINDOW"
             )
         
-        # DIRECTIVE #3: Download button row
+        # ISSUE #1 FIX: Download button initialized as invisible
         with gr.Row():
-            download_btn = gr.DownloadButton(label="DOWNLOAD CODE (.py)", variant="primary", visible=True)
+            download_btn = gr.DownloadButton(
+                label="DOWNLOAD CODE (.py)", 
+                variant="primary", 
+                visible=False
+            )
         
         # EVENT HANDLERS
         
@@ -521,10 +545,10 @@ Produce a pro level styled and tuned them GUI that has 1 button that when presse
         ).then(
             fn=end_execution_sequence,
             inputs=[system_terminal, coder_log, strategist_log, master_log_output, final_code_state],
-            outputs=[system_terminal, coder_log, strategist_log, master_log_output, final_code_state, working_video, celebrate_video]
+            outputs=[system_terminal, coder_log, strategist_log, master_log_output, final_code_state, working_video, celebrate_video, download_btn]
         )
         
-        # DIRECTIVE #3: Download button handler
+        # ISSUE #2 FIX: Download button handler with proper file return
         download_btn.click(
             fn=download_code,
             inputs=[final_code_state],
@@ -538,7 +562,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
+        # RESET MISSION button with download button visibility control
         reset_btn.click(
             fn=reset_mission,
             inputs=[],
@@ -554,7 +578,8 @@ Produce a pro level styled and tuned them GUI that has 1 button that when presse
                 strategist_provider,
                 strategist_model,
                 audit_checkbox,
-                final_code_state
+                final_code_state,
+                download_btn
             ]
         )