Update app.py

app.py

@@ -93,40 +93,40 @@ def main():
         prediction = model.predict(input_data)
         probability = model.predict_proba(input_data)[:, 1]
 
-        # Display prediction
-        if prediction[0] == 0:
-            st.success("Employee is predicted to stay (Attrition = No)")
-        else:
-            st.error("Employee is predicted to leave (Attrition = Yes)")
-            # Display probability
-            st.subheader("Prediction Probability")
-            st.write(f"The probability of the employee leaving is: {probability[0]*100:.2f}%")
+        # Convert integer inputs to integers
+        age = int(age)
+        monthly_income = int(monthly_income)
+        num_companies_worked = int(num_companies_worked)
+        percent_salary_hike = int(percent_salary_hike)
+        training_times_last_year = int(training_times_last_year)
+        years_since_last_promotion = int(years_since_last_promotion)
+        years_with_curr_manager = int(years_with_curr_manager)
+        
+        # Display characteristic-based recommendations
+        st.subheader("Suggestions for retaining the employee:")
+        if job_satisfaction == 1 or environment_satisfaction == 1:
+            st.write("Enhance job and environment satisfaction through initiatives such as recognition programs and improving workplace conditions.")
+        if years_since_last_promotion > 5:
+            st.write("Implement a transparent promotion policy and provide opportunities for career advancement.")
+        if years_with_curr_manager > 5:
+            st.write("Offer opportunities for a change in reporting structure to prevent stagnation and promote growth.")
+        if percent_salary_hike < 5:
+            st.write("Consider adjusting salary and benefits packages to remain competitive and reward employee loyalty.")
+        if training_times_last_year < 2:
+            st.write("Invest in employee development through training programs and continuous learning opportunities.")
+        if over_time:
+            st.write("Evaluate workload distribution and consider implementing measures to prevent overwork, such as workload balancing and flexible scheduling.")
+        if relationship_satisfaction == 1:
+            st.write("Foster positive relationships and a supportive work environment through team-building activities and open communication channels.")
+        if monthly_income < 5000:
+            st.write("Review compensation structures and adjust salaries to align with industry standards and employee expectations.")
+        if num_companies_worked > 5:
+            st.write("Identify reasons for high turnover and address issues related to job stability, career progression, and organizational culture.")
+        if work_life_balance == 1:
+            st.write("Promote work-life balance initiatives, such as flexible work arrangements and wellness programs, to support employee well-being.")
             
-            # Display characteristic-based recommendations
-            st.subheader("Suggestions for retaining the employee:")
-            if job_satisfaction == 1 or environment_satisfaction == 1:
-                st.write("Enhance job and environment satisfaction through initiatives such as recognition programs and improving workplace conditions.")
-            if years_since_last_promotion > 5:
-                st.write("Implement a transparent promotion policy and provide opportunities for career advancement.")
-            if years_with_curr_manager > 5:
-                st.write("Offer opportunities for a change in reporting structure to prevent stagnation and promote growth.")
-            if percent_salary_hike < 5:
-                st.write("Consider adjusting salary and benefits packages to remain competitive and reward employee loyalty.")
-            if training_times_last_year < 2:
-                st.write("Invest in employee development through training programs and continuous learning opportunities.")
-            if over_time:
-                st.write("Evaluate workload distribution and consider implementing measures to prevent overwork, such as workload balancing and flexible scheduling.")
-            if relationship_satisfaction == 1:
-                st.write("Foster positive relationships and a supportive work environment through team-building activities and open communication channels.")
-            if monthly_income < 5000:
-                st.write("Review compensation structures and adjust salaries to align with industry standards and employee expectations.")
-            if num_companies_worked > 5:
-                st.write("Identify reasons for high turnover and address issues related to job stability, career progression, and organizational culture.")
-            if work_life_balance == 1:
-                st.write("Promote work-life balance initiatives, such as flexible work arrangements and wellness programs, to support employee well-being.")
-            
-            # General recommendation for all negative predictions
-            st.write("Conduct exit interviews to gather feedback and identify areas for improvement in retention strategies.")
+        # General recommendation for all negative predictions
+        st.write("Conduct exit interviews to gather feedback and identify areas for improvement in retention strategies.")
 
 if __name__ == "__main__":
     main()