Update app.py

app.py

@@ -1,6 +1,7 @@
-# app.py — FINAL VERSION
+# app.py — FINAL VERSION (with Download)
 import gradio as gr
 import time
+import os # ### NEW ### - Needed for checking if the zip file exists
 from backend import (
     create_user,
     get_user_by_username,
@@ -114,24 +115,40 @@ def system_stats_loop():
         yield get_system_stats()
         time.sleep(5)
 
+### MODIFIED ###
 def stream_logs(project_id):
-    """A generator that streams logs for a selected project every 3 seconds."""
+    """A generator that streams logs and provides a download link when complete."""
     if not project_id:
-        yield "⬅️ Select a project from the dropdown to see its live logs."
+        yield "⬅️ Select a project from the dropdown to see its live logs.", gr.update(visible=False)
         return
 
+    # On first run for a new selection, ensure the download button is hidden
+    yield "Fetching logs...", gr.update(visible=False)
+
     while True:
         project = get_project(project_id)
         if project:
-            if not project['logs']:
-                yield f"🕒 Project #{project_id} is queued or starting... Logs will appear here shortly."
-            else:
-                yield project['logs']
+            logs = project['logs'] if project['logs'] else f"🕒 Project #{project_id} is queued or starting... Logs will appear here shortly."
+            
+            # Check for terminal states
+            if project['status'] == 'completed':
+                final_log = logs + "\n\n✅ **Project Completed!** Your file is ready for download."
+                zip_path = project.get('zip_path')
+                if zip_path and os.path.exists(zip_path):
+                    yield final_log, gr.update(value=zip_path, visible=True)
+                else:
+                    yield logs + "\n\n⚠️ **Warning:** Project completed, but the ZIP file was not found.", gr.update(visible=False)
+                break # Stop the loop
+            
+            elif project['status'] == 'failed':
+                yield logs, gr.update(visible=False)
+                break # Stop the loop
             
-            if project['status'] in ['completed', 'failed']:
-                break
+            # Otherwise, just stream the current logs
+            yield logs, gr.update(visible=False)
+
         else:
-            yield f"Error: Project #{project_id} not found."
+            yield f"Error: Project #{project_id} not found.", gr.update(visible=False)
             break
         time.sleep(3)
 
@@ -171,6 +188,9 @@ with gr.Blocks(title="Code Agents Pro", theme=gr.themes.Soft()) as demo:
                 project_selector = gr.Dropdown(label="Select Project to View", choices=[], interactive=True)
                 with gr.Group(elem_classes=["logs-container"]):
                     logs_display = gr.Markdown("Logs will appear here in real-time.", elem_classes=["monospace"])
+                
+                # ### NEW ### - File download component, initially hidden
+                download_zip_ui = gr.File(label="Download Project ZIP", visible=False, interactive=False)
         
         ram_monitor = gr.Markdown()
 
@@ -194,10 +214,11 @@ with gr.Blocks(title="Code Agents Pro", theme=gr.themes.Soft()) as demo:
         outputs=[projects_display, project_selector]
     )
     
+    ### MODIFIED ###
     project_selector.change(
         fn=stream_logs,
         inputs=project_selector,
-        outputs=logs_display
+        outputs=[logs_display, download_zip_ui] # Now updates both logs and the download component
     )
 
     demo.load(
@@ -207,7 +228,7 @@ with gr.Blocks(title="Code Agents Pro", theme=gr.themes.Soft()) as demo:
 
     gr.HTML("""
     <style>
-        .logs-container { height: 600px; overflow-y: auto; border: 1px solid #E5E7EB; border-radius: 8px; padding: 8px; background-color: #F9FAFB; }
+        .logs-container { height: 550px; overflow-y: auto; border: 1px solid #E5E7EB; border-radius: 8px; padding: 8px; background-color: #F9FAFB; }
         .monospace { font-family: 'SF Mono', 'Consolas', 'Courier New', monospace; font-size: 14px; white-space: pre-wrap; word-wrap: break-word; }
     </style>
     """)