Update app/main.py

app/main.py

@@ -1,22 +1,24 @@
 from fastapi import FastAPI
-from fastapi.responses import HTMLResponse
 from pydantic import BaseModel
 from supabase import create_client, Client
 import os
 import gradio as gr
 import requests
 
+# --- Supabase Config ---
 SUPABASE_URL = os.getenv("SUPABASE_URL")
 SUPABASE_KEY = os.getenv("SUPABASE_KEY")
 supabase: Client = create_client(SUPABASE_URL, SUPABASE_KEY)
 
+# --- FastAPI Setup ---
 app = FastAPI()
 
-### --- Task API --- ###
+# --- Pydantic Task Model ---
 class Task(BaseModel):
     title: str
     completed: bool = False
 
+# --- API Endpoints ---
 @app.post("/tasks/")
 def add_task(task: Task):
     response = supabase.table("tasks").insert(task.dict()).execute()
@@ -37,64 +39,65 @@ def delete_task(task_id: str):
     supabase.table("tasks").delete().eq("id", task_id).execute()
     return {"detail": "Task deleted"}
 
-### --- Gradio frontend logic --- ###
-API_URL = "http://localhost:7860"
+# --- Gradio Frontend Logic ---
+API_URL = "http://localhost:7860"  # adjust if needed for Hugging Face Spaces
 
 def get_tasks_data():
     response = requests.get(f"{API_URL}/tasks/")
     return response.json()
 
-def format_task(task):
-    return f"{task['id']} | {task['title']} - {'✔️' if task['completed'] else '❌'}"
+def get_task_choices():
+    """Returns [(label, uuid)] for the dropdown."""
+    tasks = get_tasks_data()
+    return [(f"{t['title']} - {'✔️' if t['completed'] else '❌'}", t['id']) for t in tasks]
 
-def get_tasks_ui():
-    return [format_task(t) for t in get_tasks_data()]
+def get_task_display():
+    """Returns formatted list of tasks for display."""
+    return [f"{t['title']} - {'✔️' if t['completed'] else '❌'}" for t in get_tasks_data()]
 
 def add_task_ui(title):
     if not title.strip():
-        return "Title cannot be empty", get_tasks_ui(), gr.update(choices=get_tasks_ui())
+        return "Title cannot be empty", get_task_display(), gr.update(choices=get_task_choices())
     requests.post(f"{API_URL}/tasks/", json={"title": title, "completed": False})
-    return "", get_tasks_ui(), gr.update(choices=get_tasks_ui())
+    return "", get_task_display(), gr.update(choices=get_task_choices())
 
-def complete_task_ui(task_id_combo):
-    task_id = task_id_combo.split(" | ")[0]
+def complete_task_ui(task_id):
     requests.patch(f"{API_URL}/tasks/{task_id}", params={"completed": True})
-    return get_tasks_ui(), gr.update(choices=get_tasks_ui())
+    return get_task_display(), gr.update(choices=get_task_choices())
 
-def delete_task_ui(task_id_combo):
-    task_id = task_id_combo.split(" | ")[0]
+def delete_task_ui(task_id):
     requests.delete(f"{API_URL}/tasks/{task_id}")
-    return get_tasks_ui(), gr.update(choices=get_tasks_ui())
-
-def get_task_choices():
-    tasks = get_tasks_data()
-    return [(f"{t['title']} - {'✔️' if t['completed'] else '❌'}", t['id']) for t in tasks]
+    return get_task_display(), gr.update(choices=get_task_choices())
 
+# --- Gradio UI ---
 with gr.Blocks() as demo:
-    gr.Markdown("## ✅ Task List Manager (FastAPI + Supabase)")
+    gr.Markdown("## ✅ Task List Manager")
 
     with gr.Row():
         task_input = gr.Textbox(label="New Task", placeholder="e.g., Write blog post")
         add_button = gr.Button("Add Task")
 
     task_selector = gr.Dropdown(label="Select a Task", choices=[])
+
     with gr.Row():
         complete_button = gr.Button("Mark as Completed")
         delete_button = gr.Button("Delete Task")
 
     task_display = gr.Textbox(label="All Tasks", lines=10)
 
-    # Button click logic
+    # Button Logic
     add_button.click(add_task_ui, inputs=task_input, outputs=[task_input, task_display, task_selector])
     complete_button.click(complete_task_ui, inputs=task_selector, outputs=[task_display, task_selector])
     delete_button.click(delete_task_ui, inputs=task_selector, outputs=[task_display, task_selector])
+
+    # Load initial data
     demo.load(
         lambda: (
-            [f"{t['title']} - {'✔️' if t['completed'] else '❌'}" for t in get_tasks_data()], 
+            get_task_display(),
             gr.update(choices=get_task_choices())
         ),
         outputs=[task_display, task_selector]
     )
 
-# Mount Gradio at root path
+# Mount Gradio app into FastAPI
 app = gr.mount_gradio_app(app, demo, path="/")