Update app.py

app.py

@@ -1,6 +1,4 @@
 import gradio as gr
-import torch
-import psutil
 from backend import (
     create_user,
     get_user_by_username,
@@ -11,7 +9,8 @@ from backend import (
     queue_job,
     get_project
 )
-
+import psutil
+import torch
 user_state = {"user_id": None, "username": ""}
 
 def page_login(username, password):
@@ -45,7 +44,7 @@ def page_projects():
 def start_new_project(prompt):
     if not user_state["user_id"]:
         return "❌ Please login first.", "", "", ""
-    
+
     projects = get_user_projects(user_state["user_id"])
     if len(projects) >= 3:
         return "⚠️ You can only have 3 projects. Delete one to create another.", "", "", ""
@@ -53,7 +52,12 @@ def start_new_project(prompt):
     project_id = create_project(user_state["user_id"], "New Project", prompt)
     queue_job(project_id, user_state["user_id"], prompt)
     
-    return f"✅ Project #{project_id} started! You can leave this page — we'll notify you when it's ready.", "", "", ""
+    return (
+        f"✅ Project #{project_id} started! You can leave this page — we'll notify you when it's ready.",
+        "",
+        "",
+        gr.update(value=f"✅ Project #{project_id} started! Now go to ‘Agent Conversation’ → click ‘Load My Projects’ to see logs.")
+    )
 
 def search_projects_ui(query):
     if not user_state["user_id"]:
@@ -66,8 +70,14 @@ def search_projects_ui(query):
         for p in results
     ])
 
+def update_dropdown():
+    if not user_state["user_id"]:
+        return []
+    projects = get_user_projects(user_state["user_id"], limit=10)
+    return [(f"[{p[0]}] {p[1]}", p[0]) for p in projects]
+
 def show_agent_logs(project_id):
-    if not project_id:
+    if not project_id or project_id == "":
         return "No project selected."
     project = get_project(project_id)
     if not project:
@@ -112,22 +122,40 @@ with gr.Blocks(title="Code Agents Pro") as demo:
     # ---- PAGE 3: AGENT CONVERSATION LOGS ----
     with gr.Group(visible=False) as page_logs_ui:
         gr.Markdown("### 🧠 Agent Conversation Logs")
-        project_selector = gr.Dropdown(label="Select Project", choices=[])
+        project_selector = gr.Dropdown(
+            label="Select Project",
+            choices=[],
+            interactive=True,
+            allow_custom_value=False
+        )
+        refresh_dropdown_btn = gr.Button("🔄 Load My Projects")
         view_logs_btn = gr.Button("👁️ View Full Conversation")
-        logs_display = gr.Markdown(label="Agent Messages", value="", height=600)
-        refresh_logs_btn = gr.Button("🔄 Refresh Logs")
+        logs_display = gr.Markdown(
+            label="Agent Messages",
+            value="",
+            height=600,
+            elem_classes=["monospace"]
+        )
+
+        # When user clicks "Load My Projects", populate dropdown
+        refresh_dropdown_btn.click(
+            fn=update_dropdown,
+            inputs=None,
+            outputs=project_selector
+        )
 
-        # Update dropdown on page load
-        def update_dropdown():
-            if not user_state["user_id"]:
-                return []
-            projects = get_user_projects(user_state["user_id"], limit=10)
-            return [(f"[{p[0]}] {p[1]}", p[0]) for p in projects]
+        # When user selects a project, show logs
+        project_selector.change(
+            fn=show_agent_logs,
+            inputs=project_selector,
+            outputs=logs_display
+        )
 
-        demo.load(update_dropdown, outputs=project_selector)
-        project_selector.change(fn=show_agent_logs, inputs=project_selector, outputs=logs_display)
-        view_logs_btn.click(fn=show_agent_logs, inputs=project_selector, outputs=logs_display)
-        refresh_logs_btn.click(fn=show_agent_logs, inputs=project_selector, outputs=logs_display)
+        view_logs_btn.click(
+            fn=show_agent_logs,
+            inputs=project_selector,
+            outputs=logs_display
+        )
 
     # ------------------------------ EVENTS ------------------------------
 
@@ -157,7 +185,7 @@ with gr.Blocks(title="Code Agents Pro") as demo:
         outputs=[projects_display, search_results]
     )
 
-    # Switch to logs tab when project starts
+    # Switch to logs tab after starting project
     start_btn.click(
         lambda: (gr.update(visible=True), gr.update(visible=False)),
         outputs=[page_logs_ui, page_main_ui]
@@ -170,4 +198,20 @@ with gr.Blocks(title="Code Agents Pro") as demo:
         outputs=[page_logs_ui, page_main_ui]
     )
 
+    # Add monospace CSS
+    gr.HTML("""
+    <style>
+        .monospace {
+            font-family: 'Courier New', Courier, monospace;
+            font-size: 14px;
+            white-space: pre-wrap;
+            word-wrap: break-word;
+            background-color: #f4f4f4;
+            padding: 10px;
+            border-radius: 5px;
+            margin-top: 10px;
+        }
+    </style>
+    """)
+
 demo.queue().launch(server_name="0.0.0.0", server_port=7860)