Create app.py

app.py

@@ -0,0 +1,83 @@
+import gradio as gr
+from backend import verify_hf_token, get_user_projects, run_code_agents
+
+user_state = {"user_id": None, "hf_token": "", "current_project": None}
+
+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
+    else:
+        return gr.update(), gr.update(), msg
+
+def page_projects():
+    if not user_state["user_id"]:
+        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]})"
+        for p in projects
+    ])
+
+def start_code_agents(project_description):
+    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
+
+with gr.Blocks(title="Code Agents") as demo:
+    gr.Markdown("# Code Agents — AI Teams That Build Software From Prompts")
+
+    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")
+        login_msg = gr.Markdown()
+
+    with gr.Group(visible=False) as page_main_ui:
+        gr.Markdown("### Your Projects")
+        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
+        )
+        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)
+
+    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'.")
+
+    login_btn.click(
+        page_login,
+        inputs=token_input,
+        outputs=[page_login_ui, page_main_ui, login_msg]
+    )
+
+    refresh_btn.click(
+        page_projects,
+        outputs=projects_display
+    )
+
+    start_btn.click(
+        start_code_agents,
+        inputs=project_input,
+        outputs=[agent_logs, zip_download, timeline]
+    ).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]
+    )
+
+demo.queue().launch(server_name="0.0.0.0", server_port=7860)