Update app.py

app.py

@@ -1,83 +1,131 @@
 import gradio as gr
-from backend import verify_hf_token, get_user_projects, run_code_agents
+from backend import (
+    create_user,
+    get_user_by_username,
+    verify_password,
+    create_project,
+    get_user_projects,
+    search_projects,
+    queue_job
+)
+import os
 
-user_state = {"user_id": None, "hf_token": "", "current_project": None}
+# ------------------------------ STATE ------------------------------
+user_state = {"user_id": None, "username": ""}
 
-def page_login(hf_token):
-    user_id, msg = verify_hf_token(hf_token)
-    if user_id:
-        user_state["user_id"] = user_id
-        user_state["hf_token"] = hf_token
-        return gr.update(visible=False), gr.update(visible=True), msg
+# ------------------------------ PAGES ------------------------------
+
+def page_login(username, password):
+    user = get_user_by_username(username)
+    if user and verify_password(password, user[1]):
+        user_state["user_id"] = user[0]
+        user_state["username"] = username
+        return gr.update(visible=False), gr.update(visible=True), f"Welcome back, {username}! 👨‍💻"
     else:
-        return gr.update(), gr.update(), msg
+        # Try to create new user
+        user_id = create_user(username, password)
+        if user_id:
+            user_state["user_id"] = user_id
+            user_state["username"] = username
+            return gr.update(visible=False), gr.update(visible=True), f"Account created! Welcome, {username}!"
+        else:
+            return gr.update(), gr.update(), "Username already exists or invalid."
 
 def page_projects():
     if not user_state["user_id"]:
-        return "Please login first."
+        return "Please login first.", ""
     projects = get_user_projects(user_state["user_id"])
     if not projects:
-        return "No projects yet. Describe your idea below!"
-    return "\n".join([
-        f"Project #{p[0]}: {p[1]} ({p[4]}) | [Download]({p[3]})"
+        return "No projects yet. Create one below!", ""
+
+    project_list = "\n".join([
+        f"📁 [{p[0]}] {p[1]} ({p[3]}) - {p[5]}"
         for p in projects
     ])
+    return project_list, ""
 
-def start_code_agents(project_description):
+def start_new_project(prompt):
     if not user_state["user_id"]:
-        return "Please login first.", None, []
-    logs, zip_path, timeline = run_code_agents(user_state["user_id"], project_description)
-    return logs, zip_path, timeline
+        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.", "", ""
 
-with gr.Blocks(title="Code Agents") as demo:
-    gr.Markdown("# Code Agents — AI Teams That Build Software From Prompts")
+    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.", "", ""
 
+def search_projects_ui(query):
+    if not user_state["user_id"]:
+        return "Please login first."
+    results = search_projects(user_state["user_id"], query)
+    if not results:
+        return "No matching projects."
+    return "\n".join([
+        f"📁 [{p[0]}] {p[1]} ({p[3]})"
+        for p in results
+    ])
+
+# ------------------------------ UI ------------------------------
+
+with gr.Blocks(title="Code Agents Pro") as demo:
+    gr.Markdown("# 🤖 Code Agents Pro — AI Software Team, Fully Automated")
+
+    # ---- PAGE 1: LOGIN ----
     with gr.Group() as page_login_ui:
-        gr.Markdown("### Step 1: Login with Hugging Face Token")
-        token_input = gr.Textbox(label="HF Token (starts with 'hf_')", type="password")
-        login_btn = gr.Button("Login to Code Agents", variant="primary")
+        gr.Markdown("### 🔐 Login or Sign Up")
+        username_input = gr.Textbox(label="Username", placeholder="Choose a unique username")
+        password_input = gr.Textbox(label="Password", type="password", placeholder="Minimum 6 characters")
+        login_btn = gr.Button("Login / Sign Up", variant="primary")
         login_msg = gr.Markdown()
 
+    # ---- PAGE 2: PROJECTS ----
     with gr.Group(visible=False) as page_main_ui:
-        gr.Markdown("### Your Projects")
+        gr.Markdown("### 📁 Your Projects (Max 3)")
         refresh_btn = gr.Button("Refresh Projects")
-        projects_display = gr.Markdown(label="Your Projects", value="Loading...")
-        gr.Markdown("### Describe Your Project")
-        project_input = gr.Textbox(
-            label="What do you want to build?",
-            placeholder="Example: A Python CLI tool that downloads YouTube videos as MP3. Use yt-dlp and saves them to ./downloads/",
-            lines=5
+        projects_display = gr.Markdown(label="Your Projects", value="")
+        search_input = gr.Textbox(label="Search Projects", placeholder="Type to search...")
+        search_btn = gr.Button("🔍 Search")
+        search_results = gr.Markdown(label="Search Results")
+
+        gr.Markdown("### ➕ Start New Project")
+        project_prompt = gr.Textbox(
+            label="Describe your project (e.g., 'Build a Flask app that lets users upload images and label them.')",
+            lines=5,
+            placeholder="Be specific about files, libraries, and behavior."
         )
-        start_btn = gr.Button("Start AI Team Work", variant="primary")
-        agent_logs = gr.Markdown("Waiting for agents to begin...")
-        timeline = gr.JSON(label="Timeline", visible=False)
+        start_btn = gr.Button("🚀 Start AI Team Work", variant="primary")
+        status_msg = gr.Markdown()
 
-    with gr.Group(visible=False) as page_download_ui:
-        gr.Markdown("### Download Project")
-        zip_download = gr.File(label="Download Complete Project (.zip)")
-        gr.Markdown("This project is saved under your account. Refresh to see it in 'Your Projects'.")
+    # ------------------------------ EVENTS ------------------------------
 
     login_btn.click(
         page_login,
-        inputs=token_input,
+        inputs=[username_input, password_input],
         outputs=[page_login_ui, page_main_ui, login_msg]
     )
 
     refresh_btn.click(
         page_projects,
-        outputs=projects_display
+        outputs=[projects_display, search_results]
+    )
+
+    search_btn.click(
+        search_projects_ui,
+        inputs=search_input,
+        outputs=search_results
     )
 
     start_btn.click(
-        start_code_agents,
-        inputs=project_input,
-        outputs=[agent_logs, zip_download, timeline]
+        start_new_project,
+        inputs=project_prompt,
+        outputs=[status_msg, projects_display, search_results]
     ).then(
-        lambda: (gr.update(visible=True), gr.update(visible=False)),
-        outputs=[page_download_ui, page_main_ui]
-    ).then(
-        lambda: (gr.update(visible=True), gr.update(visible=False)),
-        outputs=[page_main_ui, page_download_ui]
+        page_projects,
+        outputs=[projects_display, search_results]
     )
 
 demo.queue().launch(server_name="0.0.0.0", server_port=7860)