Feature: Add remove task functionality

app.py

@@ -173,6 +173,28 @@ def add_task(task_type, complexity, deadline):
     except Exception as e:
         return f"Error: {str(e)}"
 
+def remove_task(task_id):
+    """Remove a task"""
+    try:
+        if task_id is None or task_id <= 0:
+            return "❌ Please enter a valid Task ID"
+        
+        success, task_name = tm.remove_task(int(task_id))
+        
+        if not success:
+            return f"❌ Task ID {int(task_id)} not found!"
+        
+        # Return updated task list
+        if len(tm.engine.tasks) > 0:
+            task_list = "\n".join([f"- ID {t['task_id']}: {t['type']} (Complexity: {t['complexity']})" for _, t in tm.engine.tasks.iterrows()])
+        else:
+            task_list = "No tasks remaining"
+            
+        return f"✅ Removed task: **{task_name}** (ID: {int(task_id)})\n\n**Current Tasks:**\n{task_list}"
+        
+    except Exception as e:
+        return f"Error: {str(e)}"
+
 def update_progress(task_id, user_id, progress, notes):
     """Update task progress"""
     try:
@@ -318,6 +340,16 @@ with gr.Blocks(title="🧠 AI Task Assignment System") as app:
             add_task_output = gr.Markdown()
             add_task_btn.click(add_task, inputs=[task_type, task_complexity, task_deadline], outputs=add_task_output)
         
+        # Remove Task Tab
+        with gr.Tab("🗑️ Remove Task"):
+            gr.Markdown("### Remove a task")
+            gr.Markdown("*Enter the Task ID to remove. Check the Dashboard for Task IDs.*")
+            gr.Markdown("⚠️ **Warning:** This will also remove all completion history for this task.")
+            remove_task_id = gr.Number(label="Task ID to Remove", precision=0, minimum=1)
+            remove_task_btn = gr.Button("🗑️ Remove Task", variant="stop")
+            remove_task_output = gr.Markdown()
+            remove_task_btn.click(remove_task, inputs=remove_task_id, outputs=remove_task_output)
+        
         # Assignment Tab
         with gr.Tab("🎯 AI Assignment"):
             gr.Markdown("""### Let AI assign pending tasks

task_manager.py

@@ -108,6 +108,34 @@ class TaskManager:
         
         print(f"✅ Removed user: {user_name} (ID: {user_id})")
         return True, user_name
+    
+    def remove_task(self, task_id):
+        """Remove a task by ID"""
+        task_id = int(task_id)
+        task_df = self.engine.tasks[self.engine.tasks['task_id'] == task_id]
+        
+        if len(task_df) == 0:
+            print(f"❌ Task ID {task_id} not found")
+            return False, None
+            
+        task_name = task_df['type'].iloc[0]
+        
+        # Remove task from dataframe
+        self.engine.tasks = self.engine.tasks[self.engine.tasks['task_id'] != task_id]
+        self.engine.tasks.to_csv("tasks.csv", index=False)
+        
+        # Also remove task's results
+        self.engine.results = self.engine.results[self.engine.results['task_id'] != task_id]
+        self.engine.results.to_csv(self.engine.results_file, index=False)
+        
+        # Remove from progress tracking
+        keys_to_remove = [k for k in self.engine.progress_data.keys() if k.startswith(f"{task_id}_")]
+        for key in keys_to_remove:
+            del self.engine.progress_data[key]
+        self.engine.save_progress_data()
+        
+        print(f"✅ Removed task: {task_name} (ID: {task_id})")
+        return True, task_name
 
 if __name__ == "__main__":
     # Example usage