Upload app.py with huggingface_hub

app.py

@@ -31,20 +31,20 @@ def fetch_all():
     cur.execute("SELECT id, title, completed FROM tasks")
     rows = cur.fetchall()
     conn.close()
-    data = [[r["id"], r["title"], bool(r["completed"])] for r in rows]
-    return data
+    return [[r["id"], r["title"], bool(r["completed"])] for r in rows]
 
 def get_task_options():
-    data = fetch_all()
-    return [str(row[0]) for row in data]
+    """Return list of task ids as strings for dropdowns."""
+    tasks = fetch_all()
+    return [str(row[0]) for row in tasks]
 
 def refresh_tasks():
-    data = fetch_all()
-    return data, ["id", "title", "completed"]
+    """Return only the data for the Dataframe component."""
+    return fetch_all()
 
 def add_task(title):
     if not title:
-        return fetch_all(), ["id", "title", "completed"]
+        return refresh_tasks()
     conn = get_connection()
     cur = conn.cursor()
     cur.execute("INSERT INTO tasks (title, completed) VALUES (?, ?)", (title, 0))
@@ -59,7 +59,10 @@ def update_task(task_id, new_title, completed):
     cur = conn.cursor()
     if new_title:
         cur.execute("UPDATE tasks SET title = ? WHERE id = ?", (new_title, int(task_id)))
-    cur.execute("UPDATE tasks SET completed = ? WHERE id = ?", (1 if completed else 0, int(task_id)))
+    cur.execute(
+        "UPDATE tasks SET completed = ? WHERE id = ?",
+        (1 if completed else 0, int(task_id)),
+    )
     conn.commit()
     conn.close()
     return refresh_tasks()
@@ -74,7 +77,7 @@ def delete_task(task_id):
     conn.close()
     return refresh_tasks()
 
-# Initialize database on first run
+# Initialise DB on first run
 if not os.path.isfile(DB_PATH):
     init_db()
 
@@ -85,6 +88,7 @@ with gr.Blocks() as demo:
             label="Tasks",
             headers=["id", "title", "completed"],
             datatype=["number", "str", "bool"],
+            value=[],
             interactive=False,
         )
         refresh_btn = gr.Button("Refresh")
@@ -92,22 +96,33 @@ with gr.Blocks() as demo:
         add_title = gr.Textbox(label="New Task Title", placeholder="Enter task title")
         add_btn = gr.Button("Add Task")
     with gr.Row():
-        upd_id = gr.Dropdown(label="Task ID to Update", choices=get_task_options())
+        upd_id = gr.Dropdown(
+            label="Task ID to Update",
+            choices=get_task_options(),
+        )
         upd_title = gr.Textbox(label="New Title (optional)", placeholder="Leave empty to keep current")
         upd_completed = gr.Checkbox(label="Completed")
         upd_btn = gr.Button("Update Task")
     with gr.Row():
-        del_id = gr.Dropdown(label="Task ID to Delete", choices=get_task_options())
+        del_id = gr.Dropdown(
+            label="Task ID to Delete",
+            choices=get_task_options(),
+        )
         del_btn = gr.Button("Delete Task")
 
     # Bind actions
-    refresh_btn.click(fn=refresh_tasks, inputs=None, outputs=[task_table])
-    add_btn.click(fn=add_task, inputs=add_title, outputs=[task_table])
-    upd_btn.click(fn=update_task, inputs=[upd_id, upd_title, upd_completed], outputs=[task_table])
-    del_btn.click(fn=delete_task, inputs=del_id, outputs=[task_table])
-
-    # Update dropdown options after any change
-    for btn in [add_btn, upd_btn, del_btn]:
-        btn.click(fn=get_task_options, inputs=None, outputs=[upd_id, del_id])
+    refresh_btn.click(fn=refresh_tasks, inputs=None, outputs=task_table)
+
+    add_btn.click(fn=add_task, inputs=add_title, outputs=task_table).then(
+        fn=get_task_options, inputs=None, outputs=[upd_id, del_id]
+    )
+
+    upd_btn.click(fn=update_task, inputs=[upd_id, upd_title, upd_completed], outputs=task_table).then(
+        fn=get_task_options, inputs=None, outputs=[upd_id, del_id]
+    )
+
+    del_btn.click(fn=delete_task, inputs=del_id, outputs=task_table).then(
+        fn=get_task_options, inputs=None, outputs=[upd_id, del_id]
+    )
 
 demo.launch()