fix layout

Pipfile

@@ -8,6 +8,7 @@ gradio = "*"
 pandas = "*"
 joblib = "*"
 scikit-learn = "*"
+gradio-modal = "*"
 
 [dev-packages]
 

Pipfile.lock

@@ -1,7 +1,7 @@
 {
     "_meta": {
         "hash": {
-            "sha256": "1363f39c78ce9be7795c2ab9b0fa2205a144a7ea764d1428a4daabfda8d9f87d"
+            "sha256": "0ed8c82d608eb519e9f57cd59ed5f9ce206291a090c6a4059304804af74e1d1f"
         },
         "pipfile-spec": 6,
         "requires": {
@@ -336,6 +336,15 @@
             "markers": "python_version >= '3.8'",
             "version": "==0.10.0"
         },
+        "gradio-modal": {
+            "hashes": [
+                "sha256:7fb1b30dd967e5d64251b763afc922e189f24cadc1ae428126575c26cdd42b83",
+                "sha256:83c40586b4e85bb78efd0871a2c1b8aca6b7af4547a23b1f575c6226818f4bf1"
+            ],
+            "index": "pypi",
+            "markers": "python_version >= '3.8'",
+            "version": "==0.0.2"
+        },
         "h11": {
             "hashes": [
                 "sha256:8f19fbbe99e72420ff35c00b27a34cb9937e902a8b810e2c88300c6f0a3b699d",
@@ -682,7 +691,7 @@
                 "sha256:f870204a840a60da0b12273ef34f7051e98c3b5961b61b0c2c1be6dfd64fbcd3",
                 "sha256:ffa75af20b44f8dba823498024771d5ac50620e6915abac414251bd971b4529f"
             ],
-            "markers": "python_version < '3.11'",
+            "markers": "python_version >= '3.9'",
             "version": "==1.26.4"
         },
         "orjson": {

app.py

@@ -1,6 +1,7 @@
 import pandas as pd
 import gradio as gr
 import joblib
+from gradio_modal import Modal
 
 
 # Inputs for UI
@@ -113,14 +114,12 @@ def call_the_model(
     }
     input_df = pd.DataFrame(data)
 
-    # print(df)
-    # res = MODEL.predict(input_df)
     prob = MODEL.predict_proba(input_df)
     no, yes = prob[0]
     if no > yes:
         return f"The client is {round(no * 100)}% likely to relapse."
     else:
-        return f"The client is {round(yes * 100)}% likely to complete the treatment program."
+        return f"The client is {round(yes * 100)}% likely to not relapse."
 
 
 def inference(age: gr.Number, race: gr.Dropdown):
@@ -133,159 +132,175 @@ with gr.Blocks(
         neutral_hue=gr.themes.colors.slate,
     )
 ) as demo:
-    age = gr.Number(
-        label="Age", info="Age of client at admission", minimum=0, maximum=100, value=25
-    )
-    race = gr.Dropdown(
-        [
-            "Alaska Native (Aleut, Eskimo, Indian)",
-            "American Indian (other than Alaska Native)",
-            "Asian or Pacific Islander",
-            "Black or African American",
-            "White",
-            "Asian",
-            "Other single race",
-            "Two or more races",
-            "Native Hawaiian or Other Pacific Islander",
-        ],
-        label="Race",
-        type="index",
-        value=0,
-    )
-    education = gr.Dropdown(
-        [
-            "Less than one school grade, no schooling, nursery school, or kindergarten to Grade 8",
-            "Grades 9 to 11",
-            "Grade 12 (or GED)",
-            "1-3 years of college, university, or vocational school",
-            "4 years of college, university, BA/BS, some postgraduate study, or more",
-        ],
-        label="Education",
-        info="The highest school grade completed for adults or children not attending school, or current school grade for school-age children (3-17 years old) attending school",
-        type="index",
-        value=0,
-    )
-    marital = gr.Dropdown(
-        ["Never married", "Now married", "Separated", "Divorced, widowed"],
-        label="Marital Status",
-        info="Client's marital status, compatible with U.S. Census categories",
-        type="index",
-        value=0,
-    )
-    primary_income = gr.Dropdown(
-        [
-            "Wages/salary",
-            "Public assistance",
-            "Retirement/pension, disability",
-            "Other",
-            "None",
-        ],
-        label="Primary Source of Income/Support",
-        info="Client's principal source of financial support (for children younger than 18 years old, the primary parental source of income/support)",
-        type="index",
-        value=0,
-    )
-    health_insurance = gr.Dropdown(
-        [
-            "Private insurance, Blue Cross/Blue Shield, HMO",
-            "Medicaid",
-            "Medicare, other (e.g. TRICARE, CHAMPUS)",
-            "None",
-        ],
-        label="Health Insurance",
-        info="Client's health insurance at admission",
-        type="index",
-        value=0,
-    )
-    primary_substance = gr.Dropdown(
-        [
-            "None",
-            "Alcohol",
-            "Cocaine/crack",
-            "Marijuana/hashish: Includes THC and any other cannabis sativa preparations",
-            "Heroin",
-            "Non-prescription methadone",
-            "Other opiates and synthetics: Includes buprenorphine, butorphanol, codeine, hydrocodone, hydromorphone, meperidine,morphine, opium, oxycodone, pentazocine, propoxyphene, tramadol, and other narcotic analgesics, opiates, or synthetics",
-            "PCP: Phencyclidine",
-            "Hallucinogens: Includes LSD, DMT, mescaline, peyote, psilocybin, STP, and other hallucinogens",
-            "Methamphetamine/speed",
-            "Other amphetamines: Includes amphetamines, MDMA, ‘bath salts’, phenmetrazine, and other amines and related drugs",
-            "Other stimulants: Includes methylphenidate and any other stimulants",
-            "Benzodiazepines: Includes alprazolam, chlordiazepoxide, clonazepam, clorazepate, diazepam, flunitrazepam,flurazepam, halazepam, lorazepam, oxazepam, prazepam, temazepam, triazolam, and other unspecified benzodiazepines",
-            "Other tranquilizers: Includes meprobamate, and other non-benzodiazepine tranquilizers",
-            "Barbiturates: Includes amobarbital, pentobarbital, phenobarbital, secobarbital, etc.",
-            "Other sedatives or hypnotics: Includes chloral hydrate, ethchlorvynol, glutethimide, methaqualone, and othernon-barbiturate sedatives and hypnotics",
-            "Inhalants: Includes aerosols; chloroform, ether, nitrous oxide and other anesthetics; gasoline; glue; nitrites; paint thinnerand other solvents; and other inappropriately inhaled products",
-            "Over-the-counter medications: Includes aspirin, dextromethorphan and other cough syrups, diphenhydramine and otheranti-histamines, ephedrine, sleep aids, and any other legally obtained, non-prescription medication",
-            "Other drugs: Includes diphenylhydantoin/phenytoin, GHB/GBL, ketamine, synthetic cannabinoid 'Spice', carisoprodol(Soma), and other drugs",
-        ],
-        label="Primary Substance Use",
-        info="Client's primary substance use at admission",
-        type="index",
-        value=0,
-    )
-    first_use = gr.Number(
-        label="Age at First Use of Primary Substance",
-        info="For alcohol use, this is the age of first intoxication",
-        minimum=0,
-        maximum=100,
-        value=25,
-    )
-    frequency = gr.Radio(
-        ["No use in the past month", "Some use", "Daily use"],
-        label="Frequency of Primary Substance Use",
-        type="index",
-        value="No use in the past month",
-    )
-    days_waiting = gr.Number(
-        label="Days Waiting to Enter Substance Use Treatment",
-        info="Number of days from the first contact or request for substance use treatment service until the client was admitted and the first clinical substance use treatment service was provided",
-        minimum=0,
-    )
-    arrests = gr.Radio(
-        ["None", "Once", "Two or more times"],
-        label="Arrests",
-        info="Number of arrests in the 30 days prior to admission",
-        type="index",
-        value="None",
-    )
-    attendance = gr.Dropdown(
-        [
-            "No attendance",
-            "1-3 times in the past month",
-            "4-7 times in the past month",
-            "8-30 times in the past month",
-            "Some attendance, frequency is unknown",
-        ],
-        label="Attendance at Substance Use Self-help Groups in Past 30 Days",
-        info="Frequency of attendance at a substance use self-help group in the 30 days prior to the reference date (the date of admission). Includes Alcoholics Anonymous (AA), Narcotics Anonymous (NA), and other self-help/mutual support groups focused on recovery from substance use and dependence",
-        type="index",
-        value=0,
-    )
-    services = gr.Dropdown(
-        [
-            "Detox, 24-hour, hospital inpatient",
-            "Detox, 24-hour, free-standing residential",
-            "Rehab/residential, hospital (non-detox)",
-            "Rehab/residential, short term (30 days or fewer)",
-            "Rehab/residential, long term (more than 30 days)",
-            "Ambulatory, intensive outpatient",
-            "Ambulatory, non-intensive outpatient",
-            "Ambulatory, detoxification",
-        ],
-        label="Type of Treatment Service/Setting",
-        info="Type of treatment service or treatment setting in which the client is placed at the time of admission or transfer",
-        type="index",
-        value=0,
-    )
+    title = gr.Markdown("# NoLapse: Substance Use Relapse Prediction")
+    with gr.Row():
+        with gr.Column():
+            age = gr.Number(
+                label="Age",
+                info="Age of client at admission",
+                minimum=0,
+                maximum=100,
+                value=25,
+            )
+            race = gr.Dropdown(
+                [
+                    "Alaska Native (Aleut, Eskimo, Indian)",
+                    "American Indian (other than Alaska Native)",
+                    "Asian or Pacific Islander",
+                    "Black or African American",
+                    "White",
+                    "Asian",
+                    "Other single race",
+                    "Two or more races",
+                    "Native Hawaiian or Other Pacific Islander",
+                ],
+                label="Race",
+                type="index",
+                value=0,
+            )
+            education = gr.Dropdown(
+                [
+                    "Less than one school grade, no schooling, nursery school, or kindergarten to Grade 8",
+                    "Grades 9 to 11",
+                    "Grade 12 (or GED)",
+                    "1-3 years of college, university, or vocational school",
+                    "4 years of college, university, BA/BS, some postgraduate study, or more",
+                ],
+                label="Education",
+                info="The highest school grade completed for adults or children not attending school, or current school grade for school-age children (3-17 years old) attending school",
+                type="index",
+                value=0,
+            )
+            marital = gr.Dropdown(
+                ["Never married", "Now married", "Separated", "Divorced, widowed"],
+                label="Marital Status",
+                info="Client's marital status, compatible with U.S. Census categories",
+                type="index",
+                value=0,
+            )
+            primary_income = gr.Dropdown(
+                [
+                    "Wages/salary",
+                    "Public assistance",
+                    "Retirement/pension, disability",
+                    "Other",
+                    "None",
+                ],
+                label="Primary Source of Income/Support",
+                info="Client's principal source of financial support (for children younger than 18 years old, the primary parental source of income/support)",
+                type="index",
+                value=0,
+            )
+            health_insurance = gr.Dropdown(
+                [
+                    "Private insurance, Blue Cross/Blue Shield, HMO",
+                    "Medicaid",
+                    "Medicare, other (e.g. TRICARE, CHAMPUS)",
+                    "None",
+                ],
+                label="Health Insurance",
+                info="Client's health insurance at admission",
+                type="index",
+                value=0,
+            )
+            frequency = gr.Radio(
+                ["No use in the past month", "Some use", "Daily use"],
+                label="Frequency of Primary Substance Use",
+                type="index",
+                value="No use in the past month",
+            )
 
-    co_occuring = gr.Checkbox(
-        value=False, label="Co-occurring Mental and Substance Use Disorders"
-    )
+        with gr.Column():
+            primary_substance = gr.Dropdown(
+                [
+                    "None",
+                    "Alcohol",
+                    "Cocaine/crack",
+                    "Marijuana/hashish: Includes THC and any other cannabis sativa preparations",
+                    "Heroin",
+                    "Non-prescription methadone",
+                    "Other opiates and synthetics: Includes buprenorphine, butorphanol, codeine, hydrocodone, hydromorphone, meperidine,morphine, opium, oxycodone, pentazocine, propoxyphene, tramadol, and other narcotic analgesics, opiates, or synthetics",
+                    "PCP: Phencyclidine",
+                    "Hallucinogens: Includes LSD, DMT, mescaline, peyote, psilocybin, STP, and other hallucinogens",
+                    "Methamphetamine/speed",
+                    "Other amphetamines: Includes amphetamines, MDMA, ‘bath salts’, phenmetrazine, and other amines and related drugs",
+                    "Other stimulants: Includes methylphenidate and any other stimulants",
+                    "Benzodiazepines: Includes alprazolam, chlordiazepoxide, clonazepam, clorazepate, diazepam, flunitrazepam,flurazepam, halazepam, lorazepam, oxazepam, prazepam, temazepam, triazolam, and other unspecified benzodiazepines",
+                    "Other tranquilizers: Includes meprobamate, and other non-benzodiazepine tranquilizers",
+                    "Barbiturates: Includes amobarbital, pentobarbital, phenobarbital, secobarbital, etc.",
+                    "Other sedatives or hypnotics: Includes chloral hydrate, ethchlorvynol, glutethimide, methaqualone, and othernon-barbiturate sedatives and hypnotics",
+                    "Inhalants: Includes aerosols; chloroform, ether, nitrous oxide and other anesthetics; gasoline; glue; nitrites; paint thinnerand other solvents; and other inappropriately inhaled products",
+                    "Over-the-counter medications: Includes aspirin, dextromethorphan and other cough syrups, diphenhydramine and otheranti-histamines, ephedrine, sleep aids, and any other legally obtained, non-prescription medication",
+                    "Other drugs: Includes diphenylhydantoin/phenytoin, GHB/GBL, ketamine, synthetic cannabinoid 'Spice', carisoprodol(Soma), and other drugs",
+                ],
+                label="Primary Substance Use",
+                info="Client's primary substance use at admission",
+                type="index",
+                value=0,
+            )
 
-    submit_btn = gr.Button("SUBMIT", variant="primary", size="large")
+            first_use = gr.Number(
+                label="Age at First Use of Primary Substance",
+                info="For alcohol use, this is the age of first intoxication",
+                minimum=0,
+                maximum=100,
+                value=25,
+            )
 
-    result = gr.Label(label="Result")
+            days_waiting = gr.Number(
+                label="Days Waiting to Enter Substance Use Treatment",
+                info="Number of days from the first contact or request for substance use treatment service until the client was admitted and the first clinical substance use treatment service was provided",
+                minimum=0,
+            )
+            arrests = gr.Radio(
+                ["None", "Once", "Two or more times"],
+                label="Arrests",
+                info="Number of arrests in the 30 days prior to admission",
+                type="index",
+                value="None",
+            )
+            attendance = gr.Dropdown(
+                [
+                    "No attendance",
+                    "1-3 times in the past month",
+                    "4-7 times in the past month",
+                    "8-30 times in the past month",
+                    "Some attendance, frequency is unknown",
+                ],
+                label="Attendance at Substance Use Self-help Groups in Past 30 Days",
+                info="Frequency of attendance at a substance use self-help group in the 30 days prior to the reference date (the date of admission). Includes Alcoholics Anonymous (AA), Narcotics Anonymous (NA), and other self-help/mutual support groups focused on recovery from substance use and dependence",
+                type="index",
+                value=0,
+            )
+            services = gr.Dropdown(
+                [
+                    "Detox, 24-hour, hospital inpatient",
+                    "Detox, 24-hour, free-standing residential",
+                    "Rehab/residential, hospital (non-detox)",
+                    "Rehab/residential, short term (30 days or fewer)",
+                    "Rehab/residential, long term (more than 30 days)",
+                    "Ambulatory, intensive outpatient",
+                    "Ambulatory, non-intensive outpatient",
+                    "Ambulatory, detoxification",
+                ],
+                label="Type of Treatment Service/Setting",
+                info="Type of treatment service or treatment setting in which the client is placed at the time of admission or transfer",
+                type="index",
+                value=0,
+            )
+
+            co_occuring = gr.Checkbox(
+                value=False, label="Co-occurring Mental and Substance Use Disorders"
+            )
+
+    submit_btn = gr.Button("SUBMIT", variant="primary", size="lg")
+
+    # result = gr.Label(label="Result")
+    with Modal(visible=False) as modal:
+        modal_content = gr.Label(show_label=False)
+        disclaimer = gr.Markdown(
+            "###### The accuracy of this model is 74%. Take results with caution."
+        )
 
     submit_btn.click(
         call_the_model,
@@ -305,8 +320,8 @@ with gr.Blocks(
             services,
             co_occuring,
         ],
-        [result],
-    )
+        [modal_content],
+    ).then(lambda: Modal(visible=True), None, modal)
 
 
 if __name__ == "__main__":