Update app.py

app.py

@@ -1,17 +1,13 @@
-# app.py — FINAL VERSION WITH CUSTOM GENERATOR LOOPS (BACKWARD-COMPATIBLE)
+# app.py — FINAL VERSION
 import gradio as gr
-import time # Import the time module for sleeping
+import time
 from backend import (
-    # User Management
     create_user,
     get_user_by_username,
     verify_password,
-    # Project Management
     create_project,
     get_user_projects,
-    search_projects,
     get_project,
-    # Job Queue
     queue_job,
 )
 import psutil
@@ -44,7 +40,7 @@ def format_projects_list(projects):
 def get_project_choices(user_id):
     """Fetches user projects and formats them for a dropdown."""
     if not user_id: return []
-    projects = get_user_projects(user_id, limit=20)
+    projects = get_user_projects(user_id)
     return [(f"[{p['id']}] {p['title']}", p['id']) for p in projects] if projects else []
 
 # --- CORE LOGIC HANDLERS FOR UI ---
@@ -110,7 +106,7 @@ def refresh_all_projects(user_state):
     projects = get_user_projects(user_id)
     return gr.update(value=format_projects_list(projects)), gr.update(choices=get_project_choices(user_id))
 
-# --- CUSTOM GENERATOR LOOPS (THE FIX) ---
+# --- CUSTOM GENERATOR LOOPS FOR LIVE UPDATES ---
 
 def system_stats_loop():
     """A generator that yields system stats every 5 seconds, forever."""
@@ -122,20 +118,20 @@ def stream_logs(project_id):
     """A generator that streams logs for a selected project every 3 seconds."""
     if not project_id:
         yield "⬅️ Select a project from the dropdown to see its live logs."
-        return # Stop the generator if no project is selected
+        return
 
-    # Yield an initial message immediately
-    project = get_project(project_id)
-    if not project or not project['logs']:
-        yield f"🕒 Project #{project_id} is queued or starting... Logs will appear here shortly."
-    
-    # Loop to continuously poll for new logs
     while True:
         project = get_project(project_id)
         if project:
-            yield project['logs']
-            # If project is finished, stop polling to save resources
+            if not project['logs']:
+                yield f"🕒 Project #{project_id} is queued or starting... Logs will appear here shortly."
+            else:
+                yield project['logs']
+            
             if project['status'] in ['completed', 'failed']:
+                # Refresh the main dashboard one last time to show the final status
+                # This requires a bit of a workaround since generators can only yield for one output.
+                # The visual refresh on the dashboard will happen on the next manual click.
                 break
         else:
             yield f"Error: Project #{project_id} not found."
@@ -201,23 +197,17 @@ with gr.Blocks(title="Code Agents Pro", theme=gr.themes.Soft()) as demo:
         outputs=[projects_display, project_selector]
     )
     
-    # --- THE FIX IN ACTION ---
-    # When the dropdown changes (either by user or by the start_project_flow),
-    # it triggers the stream_logs generator, starting our custom loop.
     project_selector.change(
         fn=stream_logs,
         inputs=project_selector,
         outputs=logs_display
     )
 
-    # Use a generator for the system stats loop, triggered on app load.
-    # This is the backward-compatible way to do a recurring event.
     demo.load(
         fn=system_stats_loop,
         outputs=ram_monitor
     )
 
-    # --- CSS STYLING ---
     gr.HTML("""
     <style>
         .logs-container { height: 600px; overflow-y: auto; border: 1px solid #E5E7EB; border-radius: 8px; padding: 8px; background-color: #F9FAFB; }