Update app.py

app.py

@@ -63,23 +63,29 @@ def main():
     years_since_last_promotion = str(years_since_last_promotion)
     years_with_curr_manager = str(years_with_curr_manager)
     
-    # Create a DataFrame to hold the user input data
-    input_data = pd.DataFrame({
-        'Age': [age],
-        'Department': [department],
-        'EnvironmentSatisfaction': [environment_satisfaction],
-        'JobRole': [job_role],
-        'JobSatisfaction': [job_satisfaction],
-        'MonthlyIncome': [monthly_income],
-        'NumCompaniesWorked': [num_companies_worked],
-        'OverTime': [over_time],
-        'PercentSalaryHike': [percent_salary_hike],
-        'RelationshipSatisfaction': [relationship_satisfaction],
-        'TrainingTimesLastYear': [training_times_last_year],
-        'WorkLifeBalance': [work_life_balance],
-        'YearsSinceLastPromotion': [years_since_last_promotion],
-        'YearsWithCurrManager': [years_with_curr_manager]
-    })
+# Create a DataFrame to hold the user input data
+input_data = pd.DataFrame({
+    'Age': [age],
+    'Department': [department],
+    'EnvironmentSatisfaction': [environment_satisfaction],
+    'JobRole': [job_role],
+    'JobSatisfaction': [job_satisfaction],
+    'MonthlyIncome': [monthly_income],
+    'NumCompaniesWorked': [num_companies_worked],
+    'OverTime': [over_time],
+    'PercentSalaryHike': [percent_salary_hike],
+    'RelationshipSatisfaction': [relationship_satisfaction],
+    'TrainingTimesLastYear': [training_times_last_year],
+    'WorkLifeBalance': [work_life_balance],
+    'YearsSinceLastPromotion': [years_since_last_promotion],
+    'YearsWithCurrManager': [years_with_curr_manager]
+})
+
+# Reorder columns to match the expected order
+input_data = input_data[['Age', 'Department', 'EnvironmentSatisfaction', 'JobRole', 'JobSatisfaction',
+                         'MonthlyIncome', 'NumCompaniesWorked', 'OverTime', 'PercentSalaryHike',
+                         'RelationshipSatisfaction', 'TrainingTimesLastYear', 'WorkLifeBalance',
+                         'YearsSinceLastPromotion', 'YearsWithCurrManager']]
 
     # Make predictions
     prediction = model.predict(input_data)