Update app.py

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 import time
 import os
 from backend import (
-    init_db, # Added this import
+    init_db,
     create_user,
     get_user_by_username,
     verify_password,
@@ -89,7 +89,7 @@ def login_and_setup(username, password, user_state):
             gr.update(value=welcome_message),
             gr.update(value=format_projects_list(projects)),
             gr.update(choices=get_project_choices(user_id), value=None),
-            gr.update(value=format_files_list(projects))  # For files tab
+            gr.update(value=format_files_list(projects))
         )
     return user_state, gr.update(), gr.update()
 
@@ -113,7 +113,7 @@ 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 when new project starts
+        gr.update(value=format_files_list(projects))
     )
 
 def refresh_all_projects(user_state):
@@ -129,6 +129,13 @@ def refresh_all_projects(user_state):
         gr.update(value=format_files_list(projects))
     )
 
+# NEW: Refresh files tab button
+def refresh_files(user_state):
+    user_id = user_state.get('user_id')
+    if not user_id: return gr.update()
+    projects = get_user_projects(user_id)
+    return gr.update(value=format_files_list(projects))
+
 # --- CUSTOM GENERATOR LOOPS FOR LIVE UPDATES ---
 
 def system_stats_loop():
@@ -143,7 +150,6 @@ def stream_logs(project_id):
         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:
@@ -151,7 +157,6 @@ def stream_logs(project_id):
         if project:
             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['zip_path']
@@ -159,15 +164,13 @@ def stream_logs(project_id):
                     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
+                break
             
             elif project['status'] == 'failed':
                 yield logs, gr.update(visible=False)
-                break # Stop the loop
+                break
             
-            # Otherwise, just stream the current logs
             yield logs, gr.update(visible=False)
-
         else:
             yield f"Error: Project #{project_id} not found.", gr.update(visible=False)
             break
@@ -210,10 +213,8 @@ 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"])
                 
-                # 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")
@@ -233,7 +234,7 @@ with gr.Blocks(title="Code Agents Pro", theme=gr.themes.Soft()) as demo:
             welcome_msg, 
             projects_display, 
             project_selector,
-            files_display  # New files tab content
+            files_display
         ]
     )
     
@@ -246,7 +247,7 @@ with gr.Blocks(title="Code Agents Pro", theme=gr.themes.Soft()) as demo:
             projects_display, 
             project_selector, 
             tabs,
-            files_display  # Update files tab when new project starts
+            files_display
         ]
     )
 
@@ -256,13 +257,12 @@ with gr.Blocks(title="Code Agents Pro", theme=gr.themes.Soft()) as demo:
         outputs=[
             projects_display, 
             project_selector,
-            files_display  # Also refresh files tab
+            files_display
         ]
     )
     
-    # 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(),
+        fn=refresh_files,
         inputs=[user_state],
         outputs=[files_display]
     )
@@ -278,7 +278,6 @@ with gr.Blocks(title="Code Agents Pro", theme=gr.themes.Soft()) as demo:
         outputs=ram_monitor
     )
     
-    # Initialize the database when the app loads
     demo.load(init_db, inputs=None, outputs=None, show_progress=False)
 
     gr.HTML("""