Update app.py

app.py

@@ -93,24 +93,38 @@ def main():
         prediction = model.predict(input_data)
         probability = model.predict_proba(input_data)[:, 1]
 
-        # Display prediction and probability
+        # 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)")
-            # Show prediction probability
+            # Display probability
             st.subheader("Prediction Probability")
             st.write(f"The probability of the employee leaving is: {probability[0]*100:.2f}%")
-
-            # Offer recommendations for retaining the employee
+            
+            # Display characteristic-based recommendations
             st.subheader("Suggestions for retaining the employee:")
-            st.markdown("- Invest in orientation programs and career development for entry-level staff, which could contribute to higher retention.")
-            st.markdown("- Implement mentorship programs and career development initiatives aimed at engaging and retaining younger employees.")
-            st.markdown("- Offer robust training and development programs and regular promotions to foster career growth. This investment in skills and career advancement can contribute to higher job satisfaction and retention.")
-            st.markdown("- Recognize the diverse needs of employees based on marital status and consider tailoring benefits or support programs accordingly.")
-            st.markdown("- Consider offering benefits that cater to the unique needs of married, single, and divorced employees.")
-            st.markdown("- Introduce or enhance policies that support work-life balance for employees with families.")
-            st.markdown("- Recognize the unique challenges and opportunities within each department and tailor retention strategies accordingly.")
+            if job_role_satisfaction < 2:
+                st.write("Enhance job satisfaction through job enrichment initiatives and regular assessments of job roles.")
+            if years_since_last_promotion > 5:
+                st.write("Implement a transparent promotion policy and provide opportunities for career advancement to employees who haven't been promoted recently.")
+            if years_with_current_manager > 5:
+                st.write("Offer opportunities for a change in reporting structure to prevent stagnation and promote growth.")
+            if performance_rating < 3:
+                st.write("Provide constructive feedback and support for employees with low performance ratings to improve their performance and morale.")
+            if workload > 8:
+                st.write("Implement workload balancing and stress management programs to support employee well-being and prevent burnout.")
+            if career_development_opportunities == "Limited":
+                st.write("Offer continuous learning and development opportunities to foster employee growth and engagement.")
+            if feedback_mechanism == "Not Effective":
+                st.write("Establish open channels for feedback and regular one-on-one meetings to address employee concerns and development needs.")
+            if company_culture != "Aligned with Values":
+                st.write("Foster a positive work environment aligned with organizational values to enhance employee retention.")
+            if recognition_rewards == "Limited":
+                st.write("Implement reward programs and appreciation initiatives to recognize and acknowledge employees' contributions.")
+            
+            # 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()