See

app.py

@@ -52,7 +52,6 @@ class ModelPredictor:
         combined_predictions = np.concatenate(predictions)
         majority_vote = np.bincount(combined_predictions).argmax()
         return majority_vote
-
     # Based on Equal Interval and Percentage-Based Method 
     # Severe: 13 to 16 votes (upper 25%) Moderate: 9 to 12 votes (upper-middle 25%) Low: 5 to 8 votes (lower-middle 25%) Very Low: 0 to 4 votes (lower 25%)
     def evaluate_severity(self, majority_vote_count):
@@ -74,6 +73,72 @@ model_filenames = [
 model_path = "models/"
 predictor = ModelPredictor(model_path, model_filenames)
 
+def validate_inputs(*args):
+    for arg in args:
+        if arg == '' or arg is None:  # Assuming empty string or None as unselected
+            return False
+    return True
+
+def predict(YNURSMDE, YMDEYR, YSOCMDE, YMDESUD5ANYO, YMSUD5YANY, YUSUITHK, YMDETXRX, YUSUITHKYR, YMDERSUD5ANY, YUSUIPLNYR, YCOUNMDE, YPSY1MDE, YHLTMDE, YDOCMDE, YPSY2MDE, YMDEHARX, LVLDIFMEM2, MDEIMPY, YMDEHPO, YMIMS5YANY, YMDEIMAD5YR, YMIUD5YANY, YMDEHPRX, YMIMI5YANY, YUSUIPLN, YTXMDEYR, YMDEAUD5YR, YRXMDEYR, YMDELT):
+    user_input_data = {
+        'YNURSMDE': [int(YNURSMDE)],
+        'YMDEYR': [int(YMDEYR)],
+        'YSOCMDE': [int(YSOCMDE)],
+        '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)],
+        'YMIMS5YANY': [int(YMIMS5YANY)],
+        'YMDEIMAD5YR': [int(YMDEIMAD5YR)],
+        'YMIUD5YANY': [int(YMIUD5YANY)],
+        '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)
+    majority_vote_count = sum([1 for pred in np.concatenate(predictions) if pred == 1])
+    severity = predictor.evaluate_severity(majority_vote_count)
+
+    results = []
+    unknown_count = 0
+    for i, pred in enumerate(predictions):
+        model_name = model_filenames[i].split('.')[0]
+        pred_value = pred[0]
+        if model_name in predictor.prediction_map:
+            if pred_value < len(predictor.prediction_map[model_name]):
+                results.append(f"Model {model_name}: {predictor.prediction_map[model_name][pred_value]}")
+            else:
+                results.append(f"Model {model_name}: Unknown prediction value {pred_value}")
+                unknown_count += 1
+        else:
+            results.append(f"Model {model_name}: Unknown model")
+            unknown_count += 1
+
+    formatted_results = "\n".join(results)
+    
+    if unknown_count > len(model_filenames) / 2:
+        severity += " (Unknown prediction count is more. So not sure please consult with a human.)"
+
+    return formatted_results, severity
+
 # Mapping user-friendly text to corresponding numeric values for the prediction function
 input_mapping = {
     'YNURSMDE': {"Yes": 1, "No": 0},
@@ -107,48 +172,7 @@ input_mapping = {
     'YMDELT': {"Yes": 1, "No": 2}
 }
 
-def validate_inputs(*args):
-    for arg in args:
-        if arg == '' or arg is None:  # Assuming empty string or None as unselected
-            return False
-    return True
-
-def predict_with_text(YNURSMDE, YMDEYR, YSOCMDE, YMDESUD5ANYO, YMSUD5YANY, YUSUITHK, YMDETXRX, YUSUITHKYR, YMDERSUD5ANY, YUSUIPLNYR, YCOUNMDE, YPSY1MDE, YHLTMDE, YDOCMDE, YPSY2MDE, YMDEHARX, LVLDIFMEM2, MDEIMPY, YMDEHPO, YMIMS5YANY, YMDEIMAD5YR, YMIUD5YANY, YMDEHPRX, YMIMI5YANY, YUSUIPLN, YTXMDEYR, YMDEAUD5YR, YRXMDEYR, YMDELT):
-    if not validate_inputs(YNURSMDE, YMDEYR, YSOCMDE, YMDESUD5ANYO, YMSUD5YANY, YUSUITHK, YMDETXRX, YUSUITHKYR, YMDERSUD5ANY, YUSUIPLNYR, YCOUNMDE, YPSY1MDE, YHLTMDE, YDOCMDE, YPSY2MDE, YMDEHARX, LVLDIFMEM2, MDEIMPY, YMDEHPO, YMIMS5YANY, YMDEIMAD5YR, YMIUD5YANY, YMDEHPRX, YMIMI5YANY, YUSUIPLN, YTXMDEYR, YMDEAUD5YR, YRXMDEYR, YMDELT):
-        return "Please select all required fields.", "Validation Error"
-    user_inputs = {
-        'YNURSMDE': input_mapping['YNURSMDE'][YNURSMDE],
-        'YMDEYR': input_mapping['YMDEYR'][YMDEYR],
-        'YSOCMDE': input_mapping['YSOCMDE'][YSOCMDE],
-        'YMDESUD5ANYO': input_mapping['YMDESUD5ANYO'][YMDESUD5ANYO],
-        'YMSUD5YANY': input_mapping['YMSUD5YANY'][YMSUD5YANY],
-        'YUSUITHK': input_mapping['YUSUITHK'][YUSUITHK],
-        'YMDETXRX': input_mapping['YMDETXRX'][YMDETXRX],
-        'YUSUITHKYR': input_mapping['YUSUITHKYR'][YUSUITHKYR],
-        'YMDERSUD5ANY': input_mapping['YMDERSUD5ANY'][YMDERSUD5ANY],
-        'YUSUIPLNYR': input_mapping['YUSUIPLNYR'][YUSUIPLNYR],
-        'YCOUNMDE': input_mapping['YCOUNMDE'][YCOUNMDE],
-        'YPSY1MDE': input_mapping['YPSY1MDE'][YPSY1MDE],
-        'YHLTMDE': input_mapping['YHLTMDE'][YHLTMDE],
-        'YDOCMDE': input_mapping['YDOCMDE'][YDOCMDE],
-        'YPSY2MDE': input_mapping['YPSY2MDE'][YPSY2MDE],
-        'YMDEHARX': input_mapping['YMDEHARX'][YMDEHARX],
-        'LVLDIFMEM2': input_mapping['LVLDIFMEM2'][LVLDIFMEM2],
-        'MDEIMPY': input_mapping['MDEIMPY'][MDEIMPY],
-        'YMDEHPO': input_mapping['YMDEHPO'][YMDEHPO],
-        'YMIMS5YANY': input_mapping['YMIMS5YANY'][YMIMS5YANY],
-        'YMDEIMAD5YR': input_mapping['YMDEIMAD5YR'][YMDEIMAD5YR],
-        'YMIUD5YANY': input_mapping['YMIUD5YANY'][YMIUD5YANY],
-        'YMDEHPRX': input_mapping['YMDEHPRX'][YMDEHPRX],
-        'YMIMI5YANY': input_mapping['YMIMI5YANY'][YMIMI5YANY],
-        'YUSUIPLN': input_mapping['YUSUIPLN'][YUSUIPLN],
-        'YTXMDEYR': input_mapping['YTXMDEYR'][YTXMDEYR],
-        'YMDEAUD5YR': input_mapping['YMDEAUD5YR'][YMDEAUD5YR],
-        'YRXMDEYR': input_mapping['YRXMDEYR'][YRXMDEYR],
-        'YMDELT': input_mapping['YMDELT'][YMDELT]
-    }
-    return predict(**user_inputs)
-
+# Create Gradio inputs
 inputs = [
     gr.Dropdown(list(input_mapping['YNURSMDE'].keys()), label="YNURSMDE: SAW/TALK TO NURSE/OCC THERA ABOUT MAJOR DEPRESSIVE EPISODE (MDE) IN PAST YEAR (PY)"),
     gr.Dropdown(list(input_mapping['YMDEYR'].keys()), label="YMDEYR: PAST YEAR MAJOR DEPRESSIVE EPISODE"),
@@ -186,4 +210,41 @@ outputs = [
     gr.Textbox(label="Mental Health Severity")
 ]
 
+def predict_with_text(YNURSMDE, YMDEYR, YSOCMDE, YMDESUD5ANYO, YMSUD5YANY, YUSUITHK, YMDETXRX, YUSUITHKYR, YMDERSUD5ANY, YUSUIPLNYR, YCOUNMDE, YPSY1MDE, YHLTMDE, YDOCMDE, YPSY2MDE, YMDEHARX, LVLDIFMEM2, MDEIMPY, YMDEHPO, YMIMS5YANY, YMDEIMAD5YR, YMIUD5YANY, YMDEHPRX, YMIMI5YANY, YUSUIPLN, YTXMDEYR, YMDEAUD5YR, YRXMDEYR, YMDELT):
+    if not validate_inputs(YNURSMDE, YMDEYR, YSOCMDE, YMDESUD5ANYO, YMSUD5YANY, YUSUITHK, YMDETXRX, YUSUITHKYR, YMDERSUD5ANY, YUSUIPLNYR, YCOUNMDE, YPSY1MDE, YHLTMDE, YDOCMDE, YPSY2MDE, YMDEHARX, LVLDIFMEM2, MDEIMPY, YMDEHPO, YMIMS5YANY, YMDEIMAD5YR, YMIUD5YANY, YMDEHPRX, YMIMI5YANY, YUSUIPLN, YTXMDEYR, YMDEAUD5YR, YRXMDEYR, YMDELT):
+        return "Please select all required fields.", "Validation Error"
+    
+    user_inputs = {
+        'YNURSMDE': input_mapping['YNURSMDE'][YNURSMDE],
+        'YMDEYR': input_mapping['YMDEYR'][YMDEYR],
+        'YSOCMDE': input_mapping['YSOCMDE'][YSOCMDE],
+        'YMDESUD5ANYO': input_mapping['YMDESUD5ANYO'][YMDESUD5ANYO],
+        'YMSUD5YANY': input_mapping['YMSUD5YANY'][YMSUD5YANY],
+        'YUSUITHK': input_mapping['YUSUITHK'][YUSUITHK],
+        'YMDETXRX': input_mapping['YMDETXRX'][YMDETXRX],
+        'YUSUITHKYR': input_mapping['YUSUITHKYR'][YUSUITHKYR],
+        'YMDERSUD5ANY': input_mapping['YMDERSUD5ANY'][YMDERSUD5ANY],
+        'YUSUIPLNYR': input_mapping['YUSUIPLNYR'][YUSUIPLNYR],
+        'YCOUNMDE': input_mapping['YCOUNMDE'][YCOUNMDE],
+        'YPSY1MDE': input_mapping['YPSY1MDE'][YPSY1MDE],
+        'YHLTMDE': input_mapping['YHLTMDE'][YHLTMDE],
+        'YDOCMDE': input_mapping['YDOCMDE'][YDOCMDE],
+        'YPSY2MDE': input_mapping['YPSY2MDE'][YPSY2MDE],
+        'YMDEHARX': input_mapping['YMDEHARX'][YMDEHARX],
+        'LVLDIFMEM2': input_mapping['LVLDIFMEM2'][LVLDIFMEM2],
+        'MDEIMPY': input_mapping['MDEIMPY'][MDEIMPY],
+        'YMDEHPO': input_mapping['YMDEHPO'][YMDEHPO],
+        'YMIMS5YANY': input_mapping['YMIMS5YANY'][YMIMS5YANY],
+        'YMDEIMAD5YR': input_mapping['YMDEIMAD5YR'][YMDEIMAD5YR],
+        'YMIUD5YANY': input_mapping['YMIUD5YANY'][YMIUD5YANY],
+        'YMDEHPRX': input_mapping['YMDEHPRX'][YMDEHPRX],
+        'YMIMI5YANY': input_mapping['YMIMI5YANY'][YMIMI5YANY],
+        'YUSUIPLN': input_mapping['YUSUIPLN'][YUSUIPLN],
+        'YTXMDEYR': input_mapping['YTXMDEYR'][YTXMDEYR],
+        'YMDEAUD5YR': input_mapping['YMDEAUD5YR'][YMDEAUD5YR],
+        'YRXMDEYR': input_mapping['YRXMDEYR'][YRXMDEYR],
+        'YMDELT': input_mapping['YMDELT'][YMDELT]
+    }
+    return predict(**user_inputs)
+
 gr.Interface(fn=predict_with_text, inputs=inputs, outputs=outputs, title="Adolescents Mental Health Multi-Output Prediction From NSDUH Data").launch()