Update app.py

app.py

@@ -1,7 +1,7 @@
-# app.py — FINAL VERSION (with Download)
+# app.py — FINAL VERSION (with Download and Files Tab)
 import gradio as gr
 import time
-import os # ### NEW ### - Needed for checking if the zip file exists
+import os
 from backend import (
     create_user,
     get_user_by_username,
@@ -44,6 +44,21 @@ def get_project_choices(user_id):
     projects = get_user_projects(user_id)
     return [(f"[{p['id']}] {p['title']}", p['id']) for p in projects] if projects else []
 
+def format_files_list(projects):
+    """Formats completed projects into downloadable file links."""
+    completed_projects = [p for p in projects if p['status'] == 'completed']
+    if not completed_projects:
+        return "No completed projects yet. Your finished work will appear here!"
+    
+    file_lines = []
+    for p in completed_projects:
+        zip_path = p.get('zip_path')
+        if zip_path and os.path.exists(zip_path):
+            file_lines.append(f"📥 **[{p['id']}] {p['title']}** - [Download ZIP]({zip_path})")
+        else:
+            file_lines.append(f"⚠️ **[{p['id']}] {p['title']}** - ZIP file missing")
+    return "\n".join(file_lines)
+
 # --- CORE LOGIC HANDLERS FOR UI ---
 
 def login_and_setup(username, password, user_state):
@@ -73,7 +88,8 @@ def login_and_setup(username, password, user_state):
             gr.update(visible=True),
             gr.update(value=welcome_message),
             gr.update(value=format_projects_list(projects)),
-            gr.update(choices=get_project_choices(user_id), value=None)
+            gr.update(choices=get_project_choices(user_id), value=None),
+            gr.update(value=format_files_list(projects))  # For files tab
         )
     return user_state, gr.update(), gr.update()
 
@@ -82,11 +98,11 @@ def start_project_flow(prompt, user_state):
     """The complete, seamless flow for starting a new project."""
     user_id = user_state.get('user_id')
     if not user_id:
-        return "Error: Not logged in.", "", "", None, gr.update()
+        return "Error: Not logged in.", "", "", None, gr.update(), gr.update()
 
     project_id = create_project(user_id, "New AI Project", prompt)
     if not project_id:
-        return "Error: Could not create project.", "", "", None, gr.update()
+        return "Error: Could not create project.", "", "", None, gr.update(), gr.update()
         
     queue_job(project_id, user_id, prompt)
     
@@ -97,15 +113,21 @@ def start_project_flow(prompt, user_state):
         gr.update(value=format_projects_list(projects)),
         gr.update(choices=get_project_choices(user_id), value=project_id),
         gr.Tabs(selected="logs_tab"),
+        gr.update(value=format_files_list(projects))  # Update files tab
     )
 
 def refresh_all_projects(user_state):
     """Manually refresh project lists on both tabs."""
     user_id = user_state.get('user_id')
-    if not user_id: return gr.update(), gr.update()
+    if not user_id: 
+        return gr.update(), gr.update(), gr.update()
     
     projects = get_user_projects(user_id)
-    return gr.update(value=format_projects_list(projects)), gr.update(choices=get_project_choices(user_id))
+    return (
+        gr.update(value=format_projects_list(projects)),
+        gr.update(choices=get_project_choices(user_id)),
+        gr.update(value=format_files_list(projects))
+    )
 
 # --- CUSTOM GENERATOR LOOPS FOR LIVE UPDATES ---
 
@@ -115,7 +137,6 @@ def system_stats_loop():
         yield get_system_stats()
         time.sleep(5)
 
-### MODIFIED ###
 def stream_logs(project_id):
     """A generator that streams logs and provides a download link when complete."""
     if not project_id:
@@ -189,9 +210,15 @@ with gr.Blocks(title="Code Agents Pro", theme=gr.themes.Soft()) as demo:
                 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
+                # File download component, initially hidden
                 download_zip_ui = gr.File(label="Download Project ZIP", visible=False, interactive=False)
-        
+            
+            # NEW FILES TAB
+            with gr.Tab("📁 Files", id="files_tab"):
+                gr.Markdown("### 📦 Download Completed Projects")
+                refresh_files_btn = gr.Button("🔄 Refresh Files")
+                files_display = gr.Markdown("Login to see your completed projects.")
+
         ram_monitor = gr.Markdown()
 
     # ------------------------------ EVENT WIRING ------------------------------
@@ -199,26 +226,51 @@ with gr.Blocks(title="Code Agents Pro", theme=gr.themes.Soft()) as demo:
     login_btn.click(
         fn=login_and_setup,
         inputs=[username_input, password_input, user_state],
-        outputs=[user_state, login_ui, main_ui_container, welcome_msg, projects_display, project_selector]
+        outputs=[
+            user_state, 
+            login_ui, 
+            main_ui_container, 
+            welcome_msg, 
+            projects_display, 
+            project_selector,
+            files_display  # New files tab content
+        ]
     )
     
     start_btn.click(
         fn=start_project_flow,
         inputs=[project_prompt, user_state],
-        outputs=[status_msg, project_prompt, projects_display, project_selector, tabs]
+        outputs=[
+            status_msg, 
+            project_prompt, 
+            projects_display, 
+            project_selector, 
+            tabs,
+            files_display  # Update files tab when new project starts
+        ]
     )
 
     refresh_all_btn.click(
         fn=refresh_all_projects,
         inputs=[user_state],
-        outputs=[projects_display, project_selector]
+        outputs=[
+            projects_display, 
+            project_selector,
+            files_display  # Also refresh files tab
+        ]
+    )
+    
+    # NEW: Refresh files tab button
+    refresh_files_btn.click(
+        fn=lambda user_state: gr.update(value=format_files_list(get_user_projects(user_state.get('user_id', 0)))) if user_state.get('user_id') else gr.update(),
+        inputs=[user_state],
+        outputs=[files_display]
     )
     
-    ### MODIFIED ###
     project_selector.change(
         fn=stream_logs,
         inputs=project_selector,
-        outputs=[logs_display, download_zip_ui] # Now updates both logs and the download component
+        outputs=[logs_display, download_zip_ui]
     )
 
     demo.load(