Update app.py

app.py

@@ -27,12 +27,13 @@ def predict_employee_status(
     average_monthly_hours, time_spend_company,
     work_accident, promotion_last_5years, salary, department, threshold=0.5
 ):
+    # List of all departments as encoded during training
     departments = [
-        'sales', 'accounting', 'hr', 'technical', 'support',
-        'management', 'IT', 'product_mng', 'marketing', 'RandD'
+        'IT', 'RandD', 'accounting', 'hr', 'management',
+        'marketing', 'product_mng', 'sales', 'support', 'technical'
     ]
-    
-    # One-hot encode department (include department_IT explicitly)
+
+    # One-hot encode department
     department_features = {f"department_{dept}": 0 for dept in departments}
     if department in departments:
         department_features[f"department_{department}"] = 1
@@ -79,8 +80,8 @@ def gradio_interface():
             gr.Radio([0, 1], label="Promotion in Last 5 Years (0 = No, 1 = Yes)"),
             gr.Radio([0, 1, 2], label="Salary (0 = Low, 1 = Medium, 2 = High)"),
             gr.Dropdown(
-                ['sales', 'accounting', 'hr', 'technical', 'support',
-                 'management', 'IT', 'product_mng', 'marketing', 'RandD'],
+                ['IT', 'RandD', 'accounting', 'hr', 'management',
+                 'marketing', 'product_mng', 'sales', 'support', 'technical'],
                 label="Department"
             ),
             gr.Slider(0.1, 0.9, value=0.5, step=0.05, label="Prediction Threshold")
@@ -93,3 +94,4 @@ def gradio_interface():
     interface.launch()
 
 gradio_interface()
+