app

app.py

@@ -27,14 +27,28 @@ with gr.Blocks() as app:
     gr.Markdown("# Health Risk Prediction App")
     gr.Markdown("### Input the feature values below to predict health risks.")
 
-    input_components = []
-    for feature in ['State', 'Sex', 'GeneralHealth', 'PhysicalHealthDays', 'MentalHealthDays', 'LastCheckupTime', 'PhysicalActivities', 'SleepHours', 'HadStroke', 'HadArthritis', 'HadDiabetes', 'SmokerStatus', 'ECigaretteUsage', 'RaceEthnicityCategory', 'AgeCategory', 'HeightInMeters', 'WeightInKilograms', 'BMI', 'AlcoholDrinkers', 'HighRiskLastYear']:
-        if "Days" in feature or "Hours" in feature or "BMI" in feature:
-            input_components.append(gr.Slider(0, 100, step=1, label=feature))
-        elif "Sex" in feature:
-            input_components.append(gr.Radio(["Male", "Female"], label=feature))
-        else:
-            input_components.append(gr.Textbox(label=feature))
+    inputs = [
+        gr.Dropdown(["Alabama", "Alaska", "Arizona", "Arkansas", "California"], label="State"),  # Example states
+        gr.Radio(["Male", "Female"], label="Sex"),
+        gr.Radio(["Excellent", "Very Good", "Good", "Fair", "Poor"], label="GeneralHealth"),
+        gr.Slider(0, 30, step=1, label="PhysicalHealthDays"),
+        gr.Slider(0, 30, step=1, label="MentalHealthDays"),
+        gr.Radio(["Within last year", "1-2 years ago", "3-5 years ago", "5+ years ago"], label="LastCheckupTime"),
+        gr.Radio(["Yes", "No"], label="PhysicalActivities"),
+        gr.Slider(0, 24, step=1, label="SleepHours"),
+        gr.Radio(["Yes", "No"], label="HadStroke"),
+        gr.Radio(["Yes", "No"], label="HadArthritis"),
+        gr.Radio(["Yes", "No"], label="HadDiabetes"),
+        gr.Radio(["Smoker", "Non-Smoker"], label="SmokerStatus"),
+        gr.Radio(["Yes", "No"], label="ECigaretteUsage"),
+        gr.Dropdown(["White", "Black", "Asian", "Hispanic", "Other"], label="RaceEthnicityCategory"),
+        gr.Dropdown(["18-24", "25-34", "35-44", "45-54", "55-64", "65+"], label="AgeCategory"),
+        gr.Slider(1.0, 2.5, step=0.01, label="HeightInMeters"),
+        gr.Slider(30, 200, step=1, label="WeightInKilograms"),
+        gr.Slider(10, 50, step=0.1, label="BMI"),
+        gr.Radio(["Yes", "No"], label="AlcoholDrinkers"),
+        gr.Radio(["Yes", "No"], label="HighRiskLastYear"),
+    ]
 
     predict_button = gr.Button("Predict")
     output = gr.Textbox(label="Prediction Result")
@@ -42,7 +56,7 @@ with gr.Blocks() as app:
     # Connect prediction logic
     predict_button.click(
         fn=predict_with_model,
-        inputs=input_components,
+        inputs=inputs,
         outputs=output,
     )