Feature: Add remove user functionality

app.py

@@ -132,6 +132,28 @@ def add_user(name):
     except Exception as e:
         return f"Error: {str(e)}"
 
+def remove_user(user_id):
+    """Remove a user"""
+    try:
+        if user_id is None or user_id <= 0:
+            return "❌ Please enter a valid User ID"
+        
+        success, user_name = tm.remove_user(int(user_id))
+        
+        if not success:
+            return f"❌ User ID {int(user_id)} not found!"
+        
+        # Return updated user list
+        if len(tm.engine.users) > 0:
+            user_list = "\n".join([f"- ID {u['user_id']}: {u['name']}" for _, u in tm.engine.users.iterrows()])
+        else:
+            user_list = "No users remaining"
+            
+        return f"✅ Removed user: **{user_name}** (ID: {int(user_id)})\n\n**Current Users:**\n{user_list}"
+        
+    except Exception as e:
+        return f"Error: {str(e)}"
+
 def add_task(task_type, complexity, deadline):
     """Add new task"""
     try:
@@ -274,6 +296,16 @@ with gr.Blocks(title="🧠 AI Task Assignment System") as app:
             add_user_output = gr.Markdown()
             add_user_btn.click(add_user, inputs=user_name, outputs=add_user_output)
         
+        # Remove User Tab
+        with gr.Tab("🗑️ Remove User"):
+            gr.Markdown("### Remove a team member")
+            gr.Markdown("*Enter the User ID to remove. Check the Dashboard for User IDs.*")
+            gr.Markdown("⚠️ **Warning:** This will also remove all task history for this user.")
+            remove_user_id = gr.Number(label="User ID to Remove", precision=0, minimum=1)
+            remove_user_btn = gr.Button("🗑️ Remove User", variant="stop")
+            remove_user_output = gr.Markdown()
+            remove_user_btn.click(remove_user, inputs=remove_user_id, outputs=remove_user_output)
+        
         # Add Task Tab  
         with gr.Tab("📋 Add Task"):
             gr.Markdown("### Create a new task")

task_manager.py

@@ -80,6 +80,34 @@ class TaskManager:
         self.engine.tasks.to_csv("tasks.csv", index=False)
         print(f"✅ Added task: {task_type} (ID: {new_id}, Complexity: {complexity}, Deadline: {deadline}h)")
         return new_id
+    
+    def remove_user(self, user_id):
+        """Remove a user by ID"""
+        user_id = int(user_id)
+        user_df = self.engine.users[self.engine.users['user_id'] == user_id]
+        
+        if len(user_df) == 0:
+            print(f"❌ User ID {user_id} not found")
+            return False, None
+            
+        user_name = user_df['name'].iloc[0]
+        
+        # Remove user from dataframe
+        self.engine.users = self.engine.users[self.engine.users['user_id'] != user_id]
+        self.engine.users.to_csv("users.csv", index=False)
+        
+        # Also remove user's results (optional - keeps data integrity)
+        self.engine.results = self.engine.results[self.engine.results['user_id'] != user_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 f"_{user_id}" in k]
+        for key in keys_to_remove:
+            del self.engine.progress_data[key]
+        self.engine.save_progress_data()
+        
+        print(f"✅ Removed user: {user_name} (ID: {user_id})")
+        return True, user_name
 
 if __name__ == "__main__":
     # Example usage