Update app.py

app.py

@@ -12,6 +12,16 @@ loaded_model = pickle.load(open("filtered_xgb_model.pkl", 'rb'))
 # Setup SHAP
 explainer = shap.Explainer(loaded_model) # PLEASE DO NOT CHANGE THIS.
 
+example_options = {
+    "🟢 Well Engaged": [4.5, 4.5, 4.6, 4.7, 5, 4.9],
+    "🟡 Marginal": [5, 4, 5, 4, 4, 4],
+    "🔴 At Risk": [4, 4, 4, 4, 5, 5]
+}
+
+# Function to apply the example values
+def fill_example(example_label):
+    return example_options[example_label]
+
 # Create the main function for server
 def main_func(GM3, WorkEnv3, WellBeing2, GM2, JobSecurity, WellBeing1):
     new_row = pd.DataFrame.from_dict({
@@ -37,77 +47,40 @@ def main_func(GM3, WorkEnv3, WellBeing2, GM2, JobSecurity, WellBeing1):
     
     return {"Leave": float(prob[0][0]), "Stay": 1-float(prob[0][0])}, local_plot
 
-# Create the UI
-title = "**Employee Turnover Predictor & Interpreter** 🪐"
-description1 = """
-This app predicts whether an employee is likely to stay or leave based on selected workplace factors. 
-It also provides a SHAP visualization to show how each factor influences the prediction.
-"""
-
-description2 = """
-To use the app, click on one of the examples, or adjust the values of the six employee satisfaction factors, and click on Analyze. ✨ 
-""" 
+with gr.Blocks(title="🌟 Intent to Stay Prediction") as demo:
+    gr.Markdown("# 🌟 Employee Retention Predictor")
+    gr.Markdown("Predict if an employee will **Stay** or **Leave** based on key workplace factors.")
+    gr.Markdown("---")
 
-with gr.Blocks(title=title) as demo:
-    gr.Markdown(f"## {title}")
-    # gr.Markdown("""![marketing](types-of-employee-turnover.jpg)""")
-    gr.Markdown(description1)
-    gr.Markdown("""---""")
-    gr.Markdown(description2)
-    gr.Markdown("""---""")
     with gr.Row():        
         with gr.Column():
-            GM3 = gr.Slider(label="My General Manager is an effective leader", minimum=1, maximum=5, value=4, step=0.1)
-            WorkEnv3 = gr.Slider(label="My Work Environment is comfortable and welcoming", minimum=1, maximum=5, value=4, step=0.1)
-            WellBeing2 = gr.Slider(label="I feel balanced and healthy", minimum=1, maximum=5, value=4, step=0.1)
-            GM2 = gr.Slider(label="My General Manager uses feedback from Team Members to make improvements", minimum=1, maximum=5, value=4, step=0.1)
-            JobSecurity = gr.Slider(label="Job Security", minimum=1, maximum=5, value=4, step=0.1)
-            WellBeing1 = gr.Slider(label="My mental health is good", minimum=1, maximum=5, value=4, step=0.1)
-            submit_btn = gr.Button("Analyze")
-        
-        with gr.Column(visible=True, scale=1, min_width=600) as output_col:
-            label = gr.Label(label="Predicted Intent")
-            local_plot = gr.Plot(label='SHAP Analysis')
+            GM3 = gr.Slider(label="👨‍💼 My General Manager is an effective leader", minimum=1, maximum=5, value=4, step=0.1)
+            WorkEnv3 = gr.Slider(label="🏢 My Work Environment is comfortable and welcoming", minimum=1, maximum=5, value=4, step=0.1)
+            WellBeing2 = gr.Slider(label="💖 I feel balanced and healthy", minimum=1, maximum=5, value=4, step=0.1)
+            GM2 = gr.Slider(label="📊 My General Manager uses feedback from Team Members", minimum=1, maximum=5, value=4, step=0.1)
+            JobSecurity = gr.Slider(label="🔐 Job Security", minimum=1, maximum=5, value=4, step=0.1)
+            WellBeing1 = gr.Slider(label="🧠 My mental health is good", minimum=1, maximum=5, value=4, step=0.1)
+            submit_btn = gr.Button("🚀 Analyze Now", variant="primary")
 
-            submit_btn.click(
-                main_func,
-                [GM3, WorkEnv3, WellBeing2, GM2, JobSecurity, WellBeing1],
-                [label, local_plot], api_name="IntentToStay_Predictor"
-            )
-    
-    gr.Markdown("### Example Predictions:")
+        with gr.Column():
+            label = gr.Label(label="🔮 Prediction Result")
+            local_plot = gr.Plot(label="SHAP Analysis")
 
-    # Add labeled example categories
-    gr.Markdown("""
-    | Example Label   | GM3 | WorkEnv3 | WellBeing2 | GM2 | JobSecurity | WellBeing1 |
-    |----------------|-----|---------|------------|----|-------------|------------|
-    | 🟢 **Well Engaged** | 4.5 | 4.5 | 4.6 | 4.7 | 5.0 | 4.9 |
-    | 🟡 **Marginal** | 5.0 | 4.0 | 5.0 | 4.0 | 4.0 | 4.0 |
-    | 🔴 **At Risk** | 4.0 | 4.0 | 4.0 | 4.0 | 5.0 | 5.0 |
-    """)
+    # Dropdown for labeled examples
+    gr.Markdown("### 🏷️ Select an Example:")
+    example_dropdown = gr.Dropdown(
+        label="Choose a scenario",
+        choices=list(example_options.keys())
+    )
 
-    # Display interactive examples
-    gr.Examples(
-        examples=[
-            [4.5, 4.5, 4.6, 4.7, 5, 4.9],  # Well Engaged
-            [5, 4, 5, 4, 4, 4],  # Marginal
-            [4, 4, 4, 4, 5, 5]  # At Risk
-        ],
-        inputs=[GM3, WorkEnv3, WellBeing2, GM2, JobSecurity, WellBeing1],
-        outputs=[label, local_plot],
-        fn=main_func,
-        cache_examples=True
+    # Apply example values when selected
+    example_dropdown.change(
+        fill_example, 
+        inputs=[example_dropdown], 
+        outputs=[GM3, WorkEnv3, WellBeing2, GM2, JobSecurity, WellBeing1]
     )
 
-    # Add background image using CSS
-    gr.Markdown("""
-    <style>
-        body {
-            background-image: url('https://www.hilton.com/im/en/NoHotel/22512376/hhr-brand-tile-0125.jpg?impolicy=crop&cw=4500&ch=3000&gravity=NorthWest&xposition=0&yposition=0&rw=1280&rh=854');
-            background-size: cover;
-            background-position: center;
-        }
-    </style>
-    """)
+    # Submit button functionality
+    submit_btn.click(main_func, [GM3, WorkEnv3, WellBeing2, GM2, JobSecurity, WellBeing1], [label, local_plot])
 
 demo.launch()