Update app.py

app.py

@@ -6,15 +6,12 @@ from backend import (
     create_project,
     get_user_projects,
     search_projects,
-    queue_job
+    queue_job,
+    get_project
 )
-import os
 
-# ------------------------------ STATE ------------------------------
 user_state = {"user_id": None, "username": ""}
 
-# ------------------------------ PAGES ------------------------------
-
 def page_login(username, password):
     user = get_user_by_username(username)
     if user and verify_password(password, user[1]):
@@ -22,7 +19,6 @@ def page_login(username, password):
         user_state["username"] = username
         return gr.update(visible=False), gr.update(visible=True), f"Welcome back, {username}! 👨‍💻"
     else:
-        # Try to create new user
         user_id = create_user(username, password)
         if user_id:
             user_state["user_id"] = user_id
@@ -46,17 +42,16 @@ def page_projects():
 
 def start_new_project(prompt):
     if not user_state["user_id"]:
-        return "❌ Please login first.", "", ""
+        return "❌ Please login first.", "", "", ""
     
-    # Check project limit (max 3)
     projects = get_user_projects(user_state["user_id"])
     if len(projects) >= 3:
-        return "⚠️ You can only have 3 projects. Delete one to create another.", "", ""
+        return "⚠️ You can only have 3 projects. Delete one to create another.", "", "", ""
 
     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.", "", "", ""
 
 def search_projects_ui(query):
     if not user_state["user_id"]:
@@ -69,7 +64,16 @@ def search_projects_ui(query):
         for p in results
     ])
 
-# ------------------------------ UI ------------------------------
+def show_agent_logs(project_id):
+    if not project_id:
+        return "No project selected."
+    project = get_project(project_id)
+    if not project:
+        return "Project not found."
+    logs = project[6]  # logs field
+    if not logs:
+        return "No logs yet. Still running..."
+    return logs
 
 with gr.Blocks(title="Code Agents Pro") as demo:
     gr.Markdown("# 🤖 Code Agents Pro — AI Software Team, Fully Automated")
@@ -82,7 +86,7 @@ with gr.Blocks(title="Code Agents Pro") as demo:
         login_btn = gr.Button("Login / Sign Up", variant="primary")
         login_msg = gr.Markdown()
 
-    # ---- PAGE 2: PROJECTS ----
+    # ---- PAGE 2: PROJECTS & NEW PROJECT ----
     with gr.Group(visible=False) as page_main_ui:
         gr.Markdown("### 📁 Your Projects (Max 3)")
         refresh_btn = gr.Button("Refresh Projects")
@@ -100,6 +104,29 @@ with gr.Blocks(title="Code Agents Pro") as demo:
         start_btn = gr.Button("🚀 Start AI Team Work", variant="primary")
         status_msg = gr.Markdown()
 
+        # RAM/VRAM monitor
+        ram_monitor = gr.Markdown(value=f"📊 System: RAM {round(psutil.virtual_memory().used / (1024**3), 2)}GB / 18GB | VRAM {round(torch.cuda.memory_reserved() / (1024**3) if torch.cuda.is_available() else 0, 2)}GB")
+
+    # ---- 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=[])
+        view_logs_btn = gr.Button("👁️ View Full Conversation")
+        logs_display = gr.Markdown(label="Agent Messages", value="", lines=30)
+        refresh_logs_btn = gr.Button("🔄 Refresh Logs")
+
+        # 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]
+
+        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)
+
     # ------------------------------ EVENTS ------------------------------
 
     login_btn.click(
@@ -122,10 +149,23 @@ with gr.Blocks(title="Code Agents Pro") as demo:
     start_btn.click(
         start_new_project,
         inputs=project_prompt,
-        outputs=[status_msg, projects_display, search_results]
+        outputs=[status_msg, projects_display, search_results, ram_monitor]
     ).then(
         page_projects,
         outputs=[projects_display, search_results]
     )
 
+    # Switch to logs tab when project starts
+    start_btn.click(
+        lambda: (gr.update(visible=True), gr.update(visible=False)),
+        outputs=[page_logs_ui, page_main_ui]
+    )
+
+    # Back button
+    back_btn = gr.Button("⬅️ Back to Projects", visible=False)
+    back_btn.click(
+        lambda: (gr.update(visible=False), gr.update(visible=True)),
+        outputs=[page_logs_ui, page_main_ui]
+    )
+
 demo.queue().launch(server_name="0.0.0.0", server_port=7860)