Update app.py

app.py

@@ -2,12 +2,11 @@ import gradio as gr
 import joblib
 import numpy as np
 
-# Load trained model
-model = joblib.load("xgboost_model.pkl")  # Make sure this file is in same directory
+# Load XGBoost model
+model = joblib.load("xgboost_model.pkl")
 
-# Prediction function
-def predict_alzheimers(age, gender, bmi, smoking, alcohol, diabetes, hypertension, family_history, cognitive_score, apoe4):
-    # Convert inputs
+# 🔧 Preprocessing Function
+def preprocess(age, gender, bmi, smoking, alcohol, diabetes, hypertension, family_history, cognitive_score, apoe4):
     gender = 1 if gender == "Male" else 0
     smoking = {"Non-Smoker": 0, "Smoker": 1, "Former Smoker": 2}[smoking]
     alcohol = {"None": 0, "Moderate": 1, "Heavy": 2}[alcohol]
@@ -16,9 +15,13 @@ def predict_alzheimers(age, gender, bmi, smoking, alcohol, diabetes, hypertensio
     family_history = 1 if family_history == "Yes" else 0
     apoe4 = 1 if apoe4 == "Yes" else 0
 
-    features = np.array([[age, gender, bmi, smoking, alcohol, diabetes, hypertension, family_history, cognitive_score, apoe4]])
+    return np.array([[age, gender, bmi, smoking, alcohol, diabetes, hypertension, family_history, cognitive_score, apoe4]])
 
-    # Get probability of Alzheimer's positive (class 1)
+# 🧠 Prediction Function
+def predict_alzheimers(age, gender, bmi, smoking, alcohol, diabetes, hypertension, family_history, cognitive_score, apoe4):
+    features = preprocess(age, gender, bmi, smoking, alcohol, diabetes, hypertension, family_history, cognitive_score, apoe4)
+    
+    # Predict probability of class 1 (positive)
     proba = model.predict_proba(features)[0][1]
     proba_percent = round(proba * 100, 2)
 
@@ -27,8 +30,7 @@ def predict_alzheimers(age, gender, bmi, smoking, alcohol, diabetes, hypertensio
     else:
         return f"✅ Likely Alzheimer’s Negative ({100 - proba_percent}%)"
 
-
-# Gradio UI
+# 🎨 Gradio Interface
 interface = gr.Interface(
     fn=predict_alzheimers,
     inputs=[
@@ -45,7 +47,7 @@ interface = gr.Interface(
     ],
     outputs="text",
     title="Alzheimer’s Prediction App",
-    description="Enter patient details to predict Alzheimer’s risk using XGBoost model"
+    description="🔍 Enter details to predict Alzheimer’s risk using XGBoost model"
 )
 
 if __name__ == "__main__":