Clean: Remove test data - all user input, AI only assigns

app.py

@@ -4,6 +4,7 @@ from task_manager import TaskManager
 import os
 
 # AI Task Assignment System for Hugging Face Spaces
+# All data is USER INPUT - AI only handles assignment optimization
 
 # Initialize the task manager
 tm = TaskManager()
@@ -11,43 +12,64 @@ tm = TaskManager()
 def show_dashboard():
     """Display system dashboard"""
     try:
-        # Capture dashboard information
         stats = []
         
+        # Show current users
+        if len(tm.engine.users) > 0:
+            stats.append("👥 **REGISTERED USERS**")
+            for _, user in tm.engine.users.iterrows():
+                stats.append(f"- ID {user['user_id']}: {user['name']}")
+        else:
+            stats.append("👥 **No users yet** - Add users in the 'Add User' tab")
+        
+        # Show current tasks
+        stats.append("")
+        if len(tm.engine.tasks) > 0:
+            stats.append("📋 **REGISTERED TASKS**")
+            for _, task in tm.engine.tasks.iterrows():
+                status = "✅ Completed" if task['task_id'] in tm.engine.results['task_id'].values else "⏳ Pending"
+                stats.append(f"- ID {task['task_id']}: {task['type']} (Complexity: {task['complexity']}, Deadline: {task['deadline']}h) [{status}]")
+        else:
+            stats.append("📋 **No tasks yet** - Add tasks in the 'Add Task' tab")
+        
         # Basic stats
+        stats.append("")
         if len(tm.engine.results) > 0:
-            stats.append(f"📊 **SYSTEM STATISTICS**")
+            stats.append("📊 **PERFORMANCE STATISTICS**")
             stats.append(f"- Total completed tasks: {len(tm.engine.results)}")
             stats.append(f"- Average quality: {tm.engine.results['quality'].mean():.2f}/5")
             stats.append(f"- Average time: {tm.engine.results['time_taken'].mean():.1f}h")
-        else:
-            stats.append("📊 No results yet - system ready for assignments")
             
-        # User performance
-        if len(tm.engine.results) > 0:
+            # User performance
             user_stats = tm.engine.results.merge(tm.engine.users, on='user_id').groupby('name').agg({
                 'quality': 'mean',
                 'time_taken': 'mean',
                 'task_id': 'count'
             }).round(2)
             
-            stats.append("\n👥 **USER PERFORMANCE**")
+            stats.append("\n🏆 **USER PERFORMANCE (AI Learning Data)**")
             for user, row in user_stats.iterrows():
-                stats.append(f"- {user}: {row['quality']:.1f}/5 quality, {row['time_taken']:.1f}h avg, {int(row['task_id'])} tasks")
+                skill = "⭐Expert" if row['quality'] >= 4 else "✨Good" if row['quality'] >= 3 else "📚Learning"
+                stats.append(f"- {user}: {row['quality']:.1f}/5 quality, {row['time_taken']:.1f}h avg, {int(row['task_id'])} tasks [{skill}]")
         
         # Active tasks
         if hasattr(tm.engine, 'progress_data') and tm.engine.progress_data:
             active_tasks = [task for task in tm.engine.progress_data.values() 
                            if task['status'] in ['assigned', 'in_progress']]
             if active_tasks:
-                stats.append("\n📋 **ACTIVE TASKS**")
+                stats.append("\n🔄 **ACTIVE TASKS**")
                 for task in active_tasks:
                     status_icon = "🔄" if task['status'] == 'in_progress' else "📋"
-                    stats.append(f"{status_icon} Task {task['task_id']}: {task['user_name']} → {task['task_type']}")
+                    progress = ""
+                    if task.get('progress_updates'):
+                        latest = task['progress_updates'][-1]
+                        progress = f" ({latest['progress_percent']}%)"
+                    stats.append(f"{status_icon} Task {task['task_id']}: {task['user_name']} → {task['task_type']}{progress}")
         
         # AI status
-        ai_status = "🤖 **AI Status**: " + ("Trained ✅" if tm.engine.is_trained else "Random Assignment ⚠️")
-        stats.append(f"\n{ai_status}")
+        stats.append("")
+        ai_status = "🤖 **AI Status**: " + ("Trained ✅ (Making smart assignments)" if tm.engine.is_trained else "Learning Mode ⚠️ (Need completed tasks to learn)")
+        stats.append(ai_status)
         
         return "\n".join(stats)
         
@@ -55,8 +77,14 @@ def show_dashboard():
         return f"Error: {str(e)}"
 
 def assign_tasks():
-    """Assign pending tasks"""
+    """Assign pending tasks using AI"""
     try:
+        if len(tm.engine.users) == 0:
+            return "❌ No users registered! Add users first in the 'Add User' tab."
+        
+        if len(tm.engine.tasks) == 0:
+            return "❌ No tasks registered! Add tasks first in the 'Add Task' tab."
+        
         assignments = []
         
         for _, task in tm.engine.tasks.iterrows():
@@ -66,13 +94,23 @@ def assign_tasks():
             completed = tm.engine.results[tm.engine.results['task_id'] == task_id]
             if len(completed) > 0:
                 continue
+            
+            # Check if already assigned
+            task_key = None
+            for key, data in tm.engine.progress_data.items():
+                if data['task_id'] == task_id and data['status'] in ['assigned', 'in_progress']:
+                    task_key = key
+                    break
+            if task_key:
+                continue
                 
             user_id, user_name = tm.engine.assign_task(task_id)
             if user_name:
-                assignments.append(f"✅ Task {task_id} ({task['type']}) assigned to {user_name}")
+                confidence = "AI Optimized" if tm.engine.is_trained else "Random (AI learning)"
+                assignments.append(f"✅ Task {task_id} ({task['type']}) → **{user_name}** [{confidence}]")
         
         if not assignments:
-            return "📋 No pending tasks to assign"
+            return "📋 No pending tasks to assign (all tasks are either completed or already assigned)"
             
         return "\n".join(assignments)
         
@@ -86,7 +124,10 @@ def add_user(name):
             return "❌ Please enter a valid name"
             
         tm.add_user(name.strip())
-        return f"✅ Added user: {name.strip()}"
+        
+        # Return updated user list
+        user_list = "\n".join([f"- ID {u['user_id']}: {u['name']}" for _, u in tm.engine.users.iterrows()])
+        return f"✅ Added user: **{name.strip()}**\n\n**Current Users:**\n{user_list}"
         
     except Exception as e:
         return f"Error: {str(e)}"
@@ -102,7 +143,10 @@ def add_task(task_type, complexity, deadline):
             return "❌ Deadline must be positive"
             
         tm.add_task(task_type.strip(), complexity, deadline)
-        return f"✅ Added task: {task_type.strip()} (Complexity: {complexity}, Deadline: {deadline}h)"
+        
+        # Return updated task list
+        task_list = "\n".join([f"- ID {t['task_id']}: {t['type']} (Complexity: {t['complexity']})" for _, t in tm.engine.tasks.iterrows()])
+        return f"✅ Added task: **{task_type.strip()}**\n\n**Current Tasks:**\n{task_list}"
         
     except Exception as e:
         return f"Error: {str(e)}"
@@ -110,25 +154,37 @@ def add_task(task_type, complexity, deadline):
 def update_progress(task_id, user_id, progress, notes):
     """Update task progress"""
     try:
+        if task_id <= 0 or user_id <= 0:
+            return "❌ Please enter valid Task ID and User ID"
         if not (0 <= progress <= 100):
             return "❌ Progress must be between 0 and 100"
             
         tm.update_progress(int(task_id), int(user_id), int(progress), notes.strip())
-        return f"✅ Updated progress: Task {task_id} → {progress}%"
+        return f"✅ Progress updated: Task {int(task_id)} → {int(progress)}%"
         
     except Exception as e:
         return f"Error: {str(e)}"
 
 def complete_task(task_id, user_id, time_taken, quality):
-    """Complete a task"""
+    """Complete a task - THIS IS HOW AI LEARNS"""
     try:
+        if task_id <= 0 or user_id <= 0:
+            return "❌ Please enter valid Task ID and User ID"
         if not (1 <= quality <= 5):
             return "❌ Quality must be between 1 and 5"
         if time_taken <= 0:
             return "❌ Time taken must be positive"
             
         tm.enter_result(int(task_id), int(user_id), float(time_taken), int(quality))
-        return f"✅ Task {task_id} completed successfully"
+        
+        return f"""✅ Task {int(task_id)} completed!
+
+**Result recorded:**
+- Time taken: {time_taken}h
+- Quality: {int(quality)}/5
+
+🧠 **AI is learning from this result!**
+Retrain the AI in the 'AI Training' tab to improve future assignments."""
         
     except Exception as e:
         return f"Error: {str(e)}"
@@ -136,103 +192,122 @@ def complete_task(task_id, user_id, time_taken, quality):
 def retrain_ai():
     """Retrain the AI model"""
     try:
+        if len(tm.engine.results) == 0:
+            return "❌ No completed tasks yet! Complete some tasks first so AI can learn from them."
+            
         tm.retrain_ai()
-        return "✅ AI model retrained successfully"
+        return f"""✅ AI model retrained successfully!
+
+**AI learned from {len(tm.engine.results)} completed tasks.**
+
+The AI will now make smarter assignments based on:
+- User performance patterns
+- Task complexity matching
+- Time efficiency
+- Quality consistency
+
+Try assigning new tasks to see improved recommendations!"""
         
     except Exception as e:
         return f"Error: {str(e)}"
 
-def get_users_list():
-    """Get list of users for dropdowns"""
-    try:
-        return [(f"{row['user_id']} - {row['name']}", row['user_id']) for _, row in tm.engine.users.iterrows()]
-    except:
-        return []
-
-def get_tasks_list():
-    """Get list of tasks for dropdowns"""
-    try:
-        return [(f"Task {row['task_id']} - {row['type']}", row['task_id']) for _, row in tm.engine.tasks.iterrows()]
-    except:
-        return []
-
 # Create Gradio interface
 with gr.Blocks(title="🧠 AI Task Assignment System") as app:
     gr.Markdown("""
     # 🧠 AI Task Assignment System
     
-    **Self-learning task assignment engine that gets smarter with every completed task!**
+    **A self-learning task assignment engine powered by AI**
+    
+    ### How It Works:
+    1. **👤 Add Users** - Enter your team members
+    2. **📋 Add Tasks** - Enter tasks with complexity & deadlines  
+    3. **🎯 Get AI Assignments** - AI recommends optimal person for each task
+    4. **✅ Complete Tasks** - Enter results (time taken, quality)
+    5. **🧠 AI Learns** - System improves with every completed task
     
-    - 🎯 Enter people and tasks
-    - 🤖 AI decides optimal assignments  
-    - 📊 System learns from real results
-    - 📈 Continuous improvement
+    > ⚡ **All data is YOUR input** - AI only handles assignment optimization based on observed performance!
     """)
     
     with gr.Tabs():
         # Dashboard Tab
         with gr.Tab("📊 Dashboard"):
+            gr.Markdown("### System Overview - Users, Tasks & Performance")
             dashboard_btn = gr.Button("🔄 Refresh Dashboard", variant="primary")
             dashboard_output = gr.Markdown()
             dashboard_btn.click(show_dashboard, outputs=dashboard_output)
         
-        # Assignment Tab
-        with gr.Tab("🎯 Task Assignment"):
-            gr.Markdown("### Assign pending tasks to optimal users")
-            assign_btn = gr.Button("🎯 Assign Tasks", variant="primary") 
-            assign_output = gr.Markdown()
-            assign_btn.click(assign_tasks, outputs=assign_output)
-        
         # Add User Tab
         with gr.Tab("👤 Add User"):
-            gr.Markdown("### Add new team member")
-            user_name = gr.Textbox(label="User Name", placeholder="Enter name...")
+            gr.Markdown("### Register a new team member")
+            gr.Markdown("*Enter the name of the person you want to add to the system.*")
+            user_name = gr.Textbox(label="User Name", placeholder="Enter name (e.g., John, Sarah, etc.)...")
             add_user_btn = gr.Button("➕ Add User", variant="primary")
             add_user_output = gr.Markdown()
             add_user_btn.click(add_user, inputs=user_name, outputs=add_user_output)
         
         # Add Task Tab  
         with gr.Tab("📋 Add Task"):
-            gr.Markdown("### Add new task")
+            gr.Markdown("### Create a new task")
+            gr.Markdown("*Enter task details - AI will assign it to the best person.*")
+            task_type = gr.Textbox(label="Task Name/Type", placeholder="e.g., Website Design, Data Analysis, Report Writing...")
             with gr.Row():
-                task_type = gr.Textbox(label="Task Type", placeholder="e.g., coding, design, research...")
-                task_complexity = gr.Slider(0, 1, value=0.5, label="Complexity (0=Easy, 1=Hard)")
-                task_deadline = gr.Number(label="Deadline (hours)", value=24)
+                task_complexity = gr.Slider(0, 1, value=0.5, label="Complexity (0=Very Easy, 1=Very Hard)")
+                task_deadline = gr.Number(label="Deadline (hours)", value=24, minimum=1)
             add_task_btn = gr.Button("➕ Add Task", variant="primary")
             add_task_output = gr.Markdown()
             add_task_btn.click(add_task, inputs=[task_type, task_complexity, task_deadline], outputs=add_task_output)
         
+        # Assignment Tab
+        with gr.Tab("🎯 AI Assignment"):
+            gr.Markdown("""### Let AI assign pending tasks
+            
+AI analyzes each user's past performance and assigns tasks to the most suitable person.
+
+*If no performance data exists yet, AI will make random assignments and learn from results.*""")
+            assign_btn = gr.Button("🎯 Assign All Pending Tasks", variant="primary", size="lg") 
+            assign_output = gr.Markdown()
+            assign_btn.click(assign_tasks, outputs=assign_output)
+        
         # Progress Update Tab
         with gr.Tab("📈 Update Progress"):
             gr.Markdown("### Update task progress")
+            gr.Markdown("*Track how work is progressing on assigned tasks.*")
             with gr.Row():
-                prog_task_id = gr.Number(label="Task ID", precision=0)
-                prog_user_id = gr.Number(label="User ID", precision=0) 
-                progress_pct = gr.Slider(0, 100, value=50, label="Progress %")
-            progress_notes = gr.Textbox(label="Notes (optional)", placeholder="Progress notes...")
+                prog_task_id = gr.Number(label="Task ID", precision=0, minimum=1)
+                prog_user_id = gr.Number(label="User ID", precision=0, minimum=1) 
+            progress_pct = gr.Slider(0, 100, value=50, label="Progress %")
+            progress_notes = gr.Textbox(label="Notes (optional)", placeholder="Any updates or blockers...")
             update_prog_btn = gr.Button("📈 Update Progress", variant="primary")
             update_prog_output = gr.Markdown()
             update_prog_btn.click(update_progress, inputs=[prog_task_id, prog_user_id, progress_pct, progress_notes], outputs=update_prog_output)
         
         # Complete Task Tab
         with gr.Tab("✅ Complete Task"):
-            gr.Markdown("### Mark task as completed")
+            gr.Markdown("""### Mark task as completed
+            
+**⚡ This is how AI learns!** Enter the actual results so AI can improve future assignments.""")
+            with gr.Row():
+                comp_task_id = gr.Number(label="Task ID", precision=0, minimum=1)
+                comp_user_id = gr.Number(label="User ID (who completed it)", precision=0, minimum=1)
             with gr.Row():
-                comp_task_id = gr.Number(label="Task ID", precision=0)
-                comp_user_id = gr.Number(label="User ID", precision=0)
-                time_taken = gr.Number(label="Time Taken (hours)", value=1)
-                quality_score = gr.Slider(1, 5, value=3, label="Quality (1=Poor, 5=Excellent)", step=1)
+                time_taken = gr.Number(label="Actual Time Taken (hours)", value=1, minimum=0.1)
+                quality_score = gr.Slider(1, 5, value=3, label="Quality of Work (1=Poor, 5=Excellent)", step=1)
             complete_btn = gr.Button("✅ Complete Task", variant="primary")
             complete_output = gr.Markdown()
             complete_btn.click(complete_task, inputs=[comp_task_id, comp_user_id, time_taken, quality_score], outputs=complete_output)
         
         # AI Training Tab
-        with gr.Tab("🧠 AI Training"):
-            gr.Markdown("""
-            ### Retrain AI Model
-            After completing tasks and entering results, retrain the AI to improve future assignments.
-            """)
-            retrain_btn = gr.Button("🧠 Retrain AI", variant="primary")
+        with gr.Tab("🧠 Train AI"):
+            gr.Markdown("""### Retrain AI with new data
+            
+After completing tasks, retrain the AI to improve its assignment accuracy.
+
+**The AI learns:**
+- Which users perform best on which task types
+- Time efficiency patterns  
+- Quality consistency
+- Optimal user-task matching""")
+            retrain_btn = gr.Button("🧠 Retrain AI Model", variant="primary", size="lg")
             retrain_output = gr.Markdown()
             retrain_btn.click(retrain_ai, outputs=retrain_output)
 
@@ -240,4 +315,4 @@ with gr.Blocks(title="🧠 AI Task Assignment System") as app:
     app.load(show_dashboard, outputs=dashboard_output)
 
 if __name__ == "__main__":
-    app.launch(share=True, theme=gr.themes.Soft())
+    app.launch()

results.csv

@@ -1,7 +1 @@
 task_id,user_id,time_taken,quality
-1,1,8,4
-2,2,3,5
-3,3,25,2
-4,1,6,5
-5,2,2,5
-6,1,15,4

task_progress.json

@@ -1,26 +1 @@
-{
-  "7_2": {
-    "task_id": 7,
-    "user_id": 2,
-    "user_name": "Amit",
-    "task_type": "design",
-    "complexity": 0.5,
-    "deadline": 30.0,
-    "start_time": "2026-01-11T13:40:39.520692",
-    "status": "assigned",
-    "progress_updates": [],
-    "completion_time": null
-  },
-  "8_2": {
-    "task_id": 8,
-    "user_id": 2,
-    "user_name": "Amit",
-    "task_type": "add login page",
-    "complexity": 1.0,
-    "deadline": 6.0,
-    "start_time": "2026-01-11T13:40:39.545921",
-    "status": "assigned",
-    "progress_updates": [],
-    "completion_time": null
-  }
-}
+{}

tasks.csv

@@ -1,10 +1 @@
 task_id,type,complexity,deadline
-1,study,0.6,48.0
-2,fitness,0.4,24.0
-3,project,0.9,72.0
-4,study,0.7,36.0
-5,fitness,0.3,12.0
-6,project,0.8,48.0
-7,design,0.5,30.0
-8,add login page,1.0,6.0
-9,login page,0.5,6.0

users.csv

@@ -1,6 +1 @@
 user_id,name
-1,Rishiraj
-2,Amit
-3,Rahul
-4,Priya
-5,Utkarsh