Update app.py

app.py

@@ -27,7 +27,7 @@ model = load_model()
 
 # Prediction function with consistent feature names
 def predict_employee_status(satisfaction_level, last_evaluation, number_project,
-                            average_monthly_hours, time_spent_company, 
+                            average_monthly_hours, time_spend_company, 
                             work_accident, promotion_last_5years, salary, department, threshold=0.5):
     
     # One-hot encode the department
@@ -39,22 +39,22 @@ def predict_employee_status(satisfaction_level, last_evaluation, number_project,
     if department in departments:
         department_features[f"department_{department}"] = 1
 
-    # Generate Interaction Features
+    # Automatically Generate Interaction Features
     satisfaction_evaluation = satisfaction_level * last_evaluation
     work_balance = average_monthly_hours / number_project
 
-    # Prepare the input with all expected features
+    # Prepare the input with all expected features as a DataFrame with column names
     input_data = {
         "satisfaction_level": [satisfaction_level],
         "last_evaluation": [last_evaluation],
         "number_project": [number_project],
         "average_monthly_hours": [average_monthly_hours],
-        "time_spent_company": [time_spent_company],  # Corrected
+        "time_spend_company": [time_spend_company],  # Corrected to match training data
         "Work_accident": [work_accident],
         "promotion_last_5years": [promotion_last_5years],
         "salary": [salary],
-        "satisfaction_evaluation": [satisfaction_evaluation],  # Added
-        "work_balance": [work_balance],  # Added
+        "satisfaction_evaluation": [satisfaction_evaluation],  # Added for model compatibility
+        "work_balance": [work_balance],  # Added for model compatibility
         **department_features
     }
 
@@ -84,7 +84,7 @@ def gradio_interface():
             gr.Number(label="Last Evaluation (0.0 - 1.0)"),
             gr.Number(label="Number of Projects (1 - 10)"),
             gr.Number(label="Average Monthly Hours (80 - 320)"),
-            gr.Number(label="Time Spent at Company (Years)"),  # Corrected
+            gr.Number(label="Time Spend at Company (Years)"),  # Corrected to match the notebook
             gr.Radio([0, 1], label="Work Accident (0 = No, 1 = Yes)"),
             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)"),