Update app.py

app.py

@@ -1,10 +1,8 @@
-import pickle
-
 import gradio as gr
+import pickle
 import numpy as np
 import pandas as pd
 
-
 class ModelPredictor:
     def __init__(self, model_path, model_filenames):
         self.model_path = model_path
@@ -72,8 +70,42 @@ model_filenames = [
 model_path = "models/"
 predictor = ModelPredictor(model_path, model_filenames)
 
-def predict(user_inputs):
-    user_input_data = {key: [int(value)] for key, value in user_inputs.items()}
+def predict(YNURSMDE, YMDEYR, YSOCMDE, YUCOSUITHK, YMDESUD5ANYO, YMSUD5YANY, YUSUITHK, YMDETXRX, YUSUITHKYR, YMDERSUD5ANY, YUSUIPLNYR, YCOUNMDE, YPSY1MDE, YHLTMDE, YDOCMDE, YPSY2MDE, YMDEHARX, LVLDIFMEM2, MDEIMPY, YMDEHPO, CONEGMH, COCLNEGMH, YMIMS5YANY, YMDEIMAD5YR, YMIUD5YANY, YUCOSUIPLN, YMDEHPRX, YMIMI5YANY, YUSUIPLN, YTXMDEYR, YMDEAUD5YR, YRXMDEYR, YMDELT):
+    user_input_data = {
+        'YNURSMDE': [int(YNURSMDE)],
+        'YMDEYR': [int(YMDEYR)],
+        'YSOCMDE': [int(YSOCMDE)],
+        'YUCOSUITHK': [int(YUCOSUITHK)],
+        'YMDESUD5ANYO': [int(YMDESUD5ANYO)],
+        'YMSUD5YANY': [int(YMSUD5YANY)],
+        'YUSUITHK': [int(YUSUITHK)],
+        'YMDETXRX': [int(YMDETXRX)],
+        'YUSUITHKYR': [int(YUSUITHKYR)],
+        'YMDERSUD5ANY': [int(YMDERSUD5ANY)],
+        'YUSUIPLNYR': [int(YUSUIPLNYR)],
+        'YCOUNMDE': [int(YCOUNMDE)],
+        'YPSY1MDE': [int(YPSY1MDE)],
+        'YHLTMDE': [int(YHLTMDE)],
+        'YDOCMDE': [int(YDOCMDE)],
+        'YPSY2MDE': [int(YPSY2MDE)],
+        'YMDEHARX': [int(YMDEHARX)],
+        'LVLDIFMEM2': [int(LVLDIFMEM2)],
+        'MDEIMPY': [int(MDEIMPY)],
+        'YMDEHPO': [int(YMDEHPO)],
+        'CONEGMH': [int(CONEGMH)],
+        'COCLNEGMH': [int(COCLNEGMH)],
+        'YMIMS5YANY': [int(YMIMS5YANY)],
+        'YMDEIMAD5YR': [int(YMDEIMAD5YR)],
+        'YMIUD5YANY': [int(YMIUD5YANY)],
+        'YUCOSUIPLN': [int(YUCOSUIPLN)],
+        'YMDEHPRX': [int(YMDEHPRX)],
+        'YMIMI5YANY': [int(YMIMI5YANY)],
+        'YUSUIPLN': [int(YUSUIPLN)],
+        'YTXMDEYR': [int(YTXMDEYR)],
+        'YMDEAUD5YR': [int(YMDEAUD5YR)],
+        'YRXMDEYR': [int(YRXMDEYR)],
+        'YMDELT': [int(YMDELT)]
+    }
     user_input = pd.DataFrame(user_input_data)
     predictions = predictor.make_predictions(user_input)
     majority_vote = predictor.get_majority_vote(predictions)
@@ -95,44 +127,44 @@ def predict(user_inputs):
     return results, severity
 
 inputs = [
-    gr.inputs.Dropdown(["1", "0"], label="YNURSMDE (1 = Yes, 0 = No)"),
-    gr.inputs.Dropdown(["1", "2"], label="YMDEYR (1 = Yes, 2 = No)"),
-    gr.inputs.Dropdown(["1", "0"], label="YSOCMDE (1 = Yes, 0 = No)"),
-    gr.inputs.Dropdown(["1", "2"], label="YUCOSUITHK (1 = Yes, 2 = No)"),
-    gr.inputs.Dropdown(["1", "2", "3", "4"], label="YMDESUD5ANYO (1 = SUD only, no MDE, 2 = MDE only, no SUD, 3 = SUD and MDE, 4 = Neither SUD or MDE)"),
-    gr.inputs.Dropdown(["1", "0"], label="YMSUD5YANY (1 = Yes, 0 = No)"),
-    gr.inputs.Dropdown(["1", "2", "3", "4"], label="YUSUITHK (1 = Yes, 2 = No, 3 = I'm not sure, 4 = I don't want to answer)"),
-    gr.inputs.Dropdown(["1", "0"], label="YMDETXRX (1 = Yes, 0 = No)"),
-    gr.inputs.Dropdown(["1", "2", "3", "4"], label="YUSUITHKYR (1 = Yes, 2 = No, 3 = I'm not sure, 4 = I don't want to answer)"),
-    gr.inputs.Dropdown(["1", "0"], label="YMDERSUD5ANY (1 = Yes, 0 = No)"),
-    gr.inputs.Dropdown(["1", "2", "3", "4"], label="YUSUIPLNYR (1 = Yes, 2 = No, 3 = I'm not sure, 4 = I don't want to answer)"),
-    gr.inputs.Dropdown(["1", "0"], label="YCOUNMDE (1 = Yes, 0 = No)"),
-    gr.inputs.Dropdown(["1", "0"], label="YPSY1MDE (1 = Yes, 0 = No)"),
-    gr.inputs.Dropdown(["1", "0"], label="YHLTMDE (1 = Yes, 0 = No)"),
-    gr.inputs.Dropdown(["1", "0"], label="YDOCMDE (1 = Yes, 0 = No)"),
-    gr.inputs.Dropdown(["1", "0"], label="YPSY2MDE (1 = Yes, 0 = No)"),
-    gr.inputs.Dropdown(["1", "0"], label="YMDEHARX (1 = Yes, 0 = No)"),
-    gr.inputs.Dropdown(["1", "2", "3"], label="LVLDIFMEM2 (1 = No Difficulty, 2 = Some difficulty, 3 = A lot of difficulty or cannot do at all)"),
-    gr.inputs.Dropdown(["1", "2"], label="MDEIMPY (1 = Yes, 2 = No)"),
-    gr.inputs.Dropdown(["1", "0"], label="YMDEHPO (1 = Yes, 0 = No)"),
-    gr.inputs.Dropdown(["1", "2", "3", "4", "5"], label="CONEGMH (1 = Not at all, 2 = A little, 3 = Some, 4 = Quite a bit, 5 = A lot)"),
-    gr.inputs.Dropdown(["1", "2", "3"], label="COCLNEGMH (1 = Not at all, 2 = A little or some, 3 = Quite a bit or a lot)"),
-    gr.inputs.Dropdown(["1", "0"], label="YMIMS5YANY (1 = Yes, 0 = No)"),
-    gr.inputs.Dropdown(["1", "0"], label="YMDEIMAD5YR (1 = Yes, 0 = No)"),
-    gr.inputs.Dropdown(["1", "0"], label="YMIUD5YANY (1 = Yes, 0 = No)"),
-    gr.inputs.Dropdown(["1", "2"], label="YUCOSUIPLN (1 = Yes, 2 = No)"),
-    gr.inputs.Dropdown(["1", "0"], label="YMDEHPRX (1 = Yes, 0 = No)"),
-    gr.inputs.Dropdown(["1", "0"], label="YMIMI5YANY (1 = Yes, 0 = No)"),
-    gr.inputs.Dropdown(["1", "2", "3", "4"], label="YUSUIPLN (1 = Yes, 2 = No, 3 = I'm not sure, 4 = I don't want to answer)"),
-    gr.inputs.Dropdown(["1", "0"], label="YTXMDEYR (1 = Yes, 0 = No)"),
-    gr.inputs.Dropdown(["1", "0"], label="YMDEAUD5YR (1 = Yes, 0 = No)"),
-    gr.inputs.Dropdown(["1", "0"], label="YRXMDEYR (1 = Yes, 0 = No)"),
-    gr.inputs.Dropdown(["1", "2"], label="YMDELT (1 = Yes, 2 = No)")
+    gr.Dropdown(["1", "0"], label="YNURSMDE (1 = Yes, 0 = No)"),
+    gr.Dropdown(["1", "2"], label="YMDEYR (1 = Yes, 2 = No)"),
+    gr.Dropdown(["1", "0"], label="YSOCMDE (1 = Yes, 0 = No)"),
+    gr.Dropdown(["1", "2"], label="YUCOSUITHK (1 = Yes, 2 = No)"),
+    gr.Dropdown(["1", "2", "3", "4"], label="YMDESUD5ANYO (1 = SUD only, no MDE, 2 = MDE only, no SUD, 3 = SUD and MDE, 4 = Neither SUD or MDE)"),
+    gr.Dropdown(["1", "0"], label="YMSUD5YANY (1 = Yes, 0 = No)"),
+    gr.Dropdown(["1", "2", "3", "4"], label="YUSUITHK (1 = Yes, 2 = No, 3 = I'm not sure, 4 = I don't want to answer)"),
+    gr.Dropdown(["1", "0"], label="YMDETXRX (1 = Yes, 0 = No)"),
+    gr.Dropdown(["1", "2", "3", "4"], label="YUSUITHKYR (1 = Yes, 2 = No, 3 = I'm not sure, 4 = I don't want to answer)"),
+    gr.Dropdown(["1", "0"], label="YMDERSUD5ANY (1 = Yes, 0 = No)"),
+    gr.Dropdown(["1", "2", "3", "4"], label="YUSUIPLNYR (1 = Yes, 2 = No, 3 = I'm not sure, 4 = I don't want to answer)"),
+    gr.Dropdown(["1", "0"], label="YCOUNMDE (1 = Yes, 0 = No)"),
+    gr.Dropdown(["1", "0"], label="YPSY1MDE (1 = Yes, 0 = No)"),
+    gr.Dropdown(["1", "0"], label="YHLTMDE (1 = Yes, 0 = No)"),
+    gr.Dropdown(["1", "0"], label="YDOCMDE (1 = Yes, 0 = No)"),
+    gr.Dropdown(["1", "0"], label="YPSY2MDE (1 = Yes, 0 = No)"),
+    gr.Dropdown(["1", "0"], label="YMDEHARX (1 = Yes, 0 = No)"),
+    gr.Dropdown(["1", "2", "3"], label="LVLDIFMEM2 (1 = No Difficulty, 2 = Some difficulty, 3 = A lot of difficulty or cannot do at all)"),
+    gr.Dropdown(["1", "2"], label="MDEIMPY (1 = Yes, 2 = No)"),
+    gr.Dropdown(["1", "0"], label="YMDEHPO (1 = Yes, 0 = No)"),
+    gr.Dropdown(["1", "2", "3", "4", "5"], label="CONEGMH (1 = Not at all, 2 = A little, 3 = Some, 4 = Quite a bit, 5 = A lot)"),
+    gr.Dropdown(["1", "2", "3"], label="COCLNEGMH (1 = Not at all, 2 = A little or some, 3 = Quite a bit or a lot)"),
+    gr.Dropdown(["1", "0"], label="YMIMS5YANY (1 = Yes, 0 = No)"),
+    gr.Dropdown(["1", "0"], label="YMDEIMAD5YR (1 = Yes, 0 = No)"),
+    gr.Dropdown(["1", "0"], label="YMIUD5YANY (1 = Yes, 0 = No)"),
+    gr.Dropdown(["1", "2"], label="YUCOSUIPLN (1 = Yes, 2 = No)"),
+    gr.Dropdown(["1", "0"], label="YMDEHPRX (1 = Yes, 0 = No)"),
+    gr.Dropdown(["1", "0"], label="YMIMI5YANY (1 = Yes, 0 = No)"),
+    gr.Dropdown(["1", "2", "3", "4"], label="YUSUIPLN (1 = Yes, 2 = No, 3 = I'm not sure, 4 = I don't want to answer)"),
+    gr.Dropdown(["1", "0"], label="YTXMDEYR (1 = Yes, 0 = No)"),
+    gr.Dropdown(["1", "0"], label="YMDEAUD5YR (1 = Yes, 0 = No)"),
+    gr.Dropdown(["1", "0"], label="YRXMDEYR (1 = Yes, 0 = No)"),
+    gr.Dropdown(["1", "2"], label="YMDELT (1 = Yes, 2 = No)")
 ]
 
 outputs = [
-    gr.outputs.Textbox(label="Prediction Results"),
-    gr.outputs.Textbox(label="Mental Health Severity")
+    gr.Textbox(label="Prediction Results"),
+    gr.Textbox(label="Mental Health Severity")
 ]
 
 gr.Interface(fn=predict, inputs=inputs, outputs=outputs, title="Adolescent Mental Health Prediction").launch()