Update app.py

app.py

@@ -15,6 +15,7 @@ except Exception as e:
 # Define prediction function
 def predict_stroke(gender, age, hypertension, heart_disease, ever_married,
                    work_type, residence_type, avg_glucose_level, smoking_status, bmi):
+    
     if model is None or preprocessor is None:
         return "Error: Model or preprocessor not loaded."
 
@@ -43,15 +44,15 @@ def predict_stroke(gender, age, hypertension, heart_disease, ever_married,
 iface = gr.Interface(
     fn=predict_stroke,
     inputs=[
-        gr.Textbox(label="Gender (Male/Female)"),
-        gr.Number(label="Age"),
+        gr.Radio(choices=["Male", "Female"], label="Gender"),
+        gr.Slider(minimum=1, maximum=100, step=1, label="Age"),
         gr.Radio(choices=[0, 1], label="Hypertension (0=No, 1=Yes)"),
         gr.Radio(choices=[0, 1], label="Heart Disease (0=No, 1=Yes)"),
-        gr.Textbox(label="Ever Married (Yes/No)"),
-        gr.Textbox(label="Work Type"),
-        gr.Textbox(label="Residence Type (Urban/Rural)"),
+        gr.Dropdown(choices=["Yes", "No"], label="Ever Married"),
+        gr.Dropdown(choices=["Private", "Self-employed", "Govt_job", "children", "Never_worked"], label="Work Type"),
+        gr.Radio(choices=["Urban", "Rural"], label="Residence Type"),
         gr.Number(label="Average Glucose Level"),
-        gr.Textbox(label="Smoking Status"),
+        gr.Dropdown(choices=["never smoked", "formerly smoked", "smokes", "Unknown"], label="Smoking Status"),
         gr.Number(label="BMI")
     ],
     outputs="text",
@@ -60,6 +61,6 @@ iface = gr.Interface(
     allow_flagging="never"
 )
 
-# Launch Gradio app
+# Launch the app
 if __name__ == "__main__":
     iface.launch()