Update app.py

app.py

@@ -5,6 +5,9 @@ import pickle
 import shap
 import matplotlib.pyplot as plt
 
+# Set dark style for plots
+plt.style.use("dark_background")
+
 # Load model
 with open("salar_xgb_team.pkl", "rb") as f:
     model = pickle.load(f)
@@ -21,10 +24,10 @@ def predict_salary(age, education_num, sex, capital_gain, capital_loss, hours_pe
     # Prediction & confidence
     pred = model.predict(input_data)[0]
     prob = model.predict_proba(input_data)[0][1]
-    label = ">50K" if pred == 1 else "<=50K"
+    label = "🎯 > $50K" if pred == 1 else "💼 ≤ $50K"
     confidence = f"{prob * 100:.2f}%" if pred == 1 else f"{(1 - prob) * 100:.2f}%"
 
-    # SHAP values
+    # SHAP values with dark theme
     shap_values = explainer(input_data)
     fig, ax = plt.subplots(figsize=(6, 2.5))
     shap.plots.bar(shap_values[0], max_display=6, show=False)
@@ -33,30 +36,75 @@ def predict_salary(age, education_num, sex, capital_gain, capital_loss, hours_pe
     return label, confidence, fig
 
 # Build UI
-with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.Markdown("## 💼 Income Prediction App")
+with gr.Blocks(css="""
+    body {
+        background-color: #191414;
+        color: #FFFFFF;
+        font-family: 'Helvetica Neue', sans-serif;
+    }
+    .gr-box {
+        background-color: #1DB954 !important;
+        color: #191414 !important;
+    }
+    input[type=number] {
+        font-size: 1.5em !important;
+        text-align: center !important;
+        color: #191414 !important;
+    }
+    .gr-slider input[type=range] {
+        accent-color: #1DB954;
+    }
+    .gr-button {
+        background-color: #1DB954 !important;
+        color: #191414 !important;
+        border: none !important;
+    }
+    .gr-button:hover {
+        filter: brightness(1.1);
+    }
+    #title {
+        font-size: 2em;
+        text-align: center;
+        margin-bottom: 0.3em;
+    }
+    #pulse {
+        animation: pulse 2s infinite;
+        color: #1DB954;
+        font-size: 2.5em;
+        text-align: center;
+    }
+    @keyframes pulse {
+        0% { transform: scale(1); opacity: 1; }
+        50% { transform: scale(1.05); opacity: 0.7; }
+        100% { transform: scale(1); opacity: 1; }
+    }
+""") as demo:
+
+    gr.HTML("<div id='pulse'>🔮 AI-Powered Income Prediction</div>")
     gr.Markdown(
         """
-        This tool uses a trained XGBoost model to predict whether someone earns more than $50K/year based on demographic and financial information.
-        It also shows which features influenced the prediction the most, using SHAP explainability.
-        """
+        Welcome! This app predicts if a person earns more than $50K/year using a trained XGBoost model,
+        based on demographic and financial information.
+
+        SHAP values help explain the prediction visually.
+        """, elem_id="description"
     )
 
     with gr.Row():
         with gr.Column():
             age = gr.Slider(0, 100, label="Age", value=35, step=1)
             education = gr.Dropdown(choices=[str(i) for i in range(1, 17)],
-                                    label="Education Level (1-16)", value="10")
+                                    label="Education Level (1–16)", value="10")
             sex = gr.Radio(["Male", "Female"], label="Sex", value="Male")
             cap_gain = gr.Number(label="Capital Gain", value=0)
             cap_loss = gr.Number(label="Capital Loss", value=0)
             hours = gr.Number(label="Hours per Week", value=40)
-            submit_btn = gr.Button("🔮 Predict")
+            submit_btn = gr.Button("🔍 Predict")
 
         with gr.Column():
-            result = gr.Label(label="Predicted Income")
-            confidence = gr.Label(label="Prediction Confidence")
-            shap_plot = gr.Plot(label="Feature Importance (SHAP)")
+            result = gr.Label(label="Prediction")
+            confidence = gr.Label(label="Confidence Level")
+            shap_plot = gr.Plot(label="Top Feature Contributions")
 
     gr.Markdown("### 🧪 Try Example Inputs")
     gr.Examples(
@@ -78,3 +126,4 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
 # Launch
 demo.launch()
 
+