new

app.py

@@ -51,12 +51,12 @@ with gr.Blocks() as demo:
         """
     )
 
-    task_table = gr.Dataframe(
-        value=get_tasks(),
+    # Create an empty DataFrame display
+    task_table = gr.DataFrame(
         headers=TODO_COLUMNS,
-        interactive=False,
         label="Current Tasks",
-        elem_id="task-table"
+        elem_id="task-table",
+        interactive=False
     )
 
     # Add Task Section
@@ -89,19 +89,28 @@ with gr.Blocks() as demo:
         with gr.Column(scale=1):
             add_button = gr.Button("Add Task", elem_id="button")
 
+    # Button click to add task and update DataFrame
+    def update_task_table(task, priority, status, due_date):
+        add_task(task, priority, status, due_date)
+        return get_tasks()
+
     add_button.click(
-        fn=add_task,
+        fn=update_task_table,
         inputs=[task_input, priority_input, status_input, due_date_input],
         outputs=task_table
     )
 
     # Delete Task Section
-    with gr.Row():
-        delete_index = gr.Number(label="Task Index to Delete", value=0, elem_id="input-box")
-        delete_button = gr.Button("Delete Task", elem_id="button")
+    delete_index = gr.Number(label="Task Index to Delete", value=0, elem_id="input-box")
+    delete_button = gr.Button("Delete Task", elem_id="button")
+
+    # Button click to delete task and update DataFrame
+    def update_after_delete(index):
+        delete_task(index)
+        return get_tasks()
 
     delete_button.click(
-        fn=delete_task,
+        fn=update_after_delete,
         inputs=delete_index,
         outputs=task_table
     )