Upload app.py!

app.py

@@ -0,0 +1,240 @@
+import pandas as pd
+import gradio as gr
+
+
+# Inputs for UI
+
+with gr.Blocks(
+    theme=gr.themes.Soft(
+        primary_hue=gr.themes.colors.zinc,
+        neutral_hue=gr.themes.colors.slate,
+    )
+) as demo:
+    age = (
+        gr.Number(
+            label="Age",
+            info="Age of client at admission",
+            minimum=0,
+            maximum=100,
+        ),
+    )
+    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",
+        ),
+    )
+    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",
+        ),
+    )
+    marital = (
+        gr.Dropdown(
+            ["Never married", "Now married", "Separated", "Divorced, widowed"],
+            label="Marital Status",
+            info="Client's marital status, compatible with U.S. Census categories",
+        ),
+    )
+    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)",
+        ),
+    )
+    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",
+        ),
+    )
+    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",
+        ),
+    )
+    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,
+        ),
+    )
+    frequency = (
+        gr.Radio(
+            ["No use in the past month", "Some use", "Daily use"],
+            label="Frequency of Primary Substance Use",
+        ),
+    )
+    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",
+        ),
+    )
+    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",
+        ),
+    )
+    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",
+        ),
+    )
+    co_occuring = (
+        gr.Radio(
+            ["Yes", "No"], label="Co-occurring Mental and Substance Use Disorders"
+        ),
+    )
+
+
+def conversion():
+    # Convert inputs to model inputs
+    if age >= 12 or age <= 14:
+        age = 1
+    elif age >= 15 or age <= 17:
+        age = 2
+    elif age >= 18 or age <= 20:
+        age = 3
+    elif age >= 21 or age <= 24:
+        age = 4
+    elif age >= 25 or age <= 29:
+        age = 5
+    elif age >= 30 or age <= 34:
+        age = 6
+    elif age >= 35 or age <= 39:
+        age = 7
+    elif age >= 40 or age <= 44:
+        age = 8
+    elif age >= 45 or age <= 49:
+        age = 9
+    elif age >= 50 or age <= 54:
+        age = 10
+    elif age >= 55 or age <= 64:
+        age = 11
+    else:
+        age = 12
+
+    race = race.index() + 1
+    education = education.index() + 1
+    marital = marital.index() + 1
+    primary_income = primary_income.index() + 1
+    health_insurance = health_insurance.index() + 1
+    primary_substance = primary_substance.index() + 1
+    frequency = frequency(type="index") + 1
+
+    if days_waiting == 0:
+        days_waiting = 0
+    elif days_waiting >= 1 or days_waiting <= 7:
+        days_waiting = 1
+    elif days_waiting >= 8 or days_waiting <= 14:
+        days_waiting = 2
+    elif days_waiting >= 15 or days_waiting <= 30:
+        days_waiting = 3
+    else:
+        days_waiting = 4
+
+    arrests = arrests(type="index")
+    attendance = attendance.index() + 1
+    services = services.index() + 1
+
+    if first_use <= 11:
+        first_use = 1
+    elif first_use >= 12 or first_use <= 14:
+        first_use = 2
+    elif first_use >= 15 or first_use <= 17:
+        first_use = 3
+    elif first_use >= 18 or first_use <= 20:
+        first_use = 4
+    elif first_use >= 21 or first_use <= 24:
+        first_use = 5
+    elif first_use >= 25 or first_use <= 29:
+        first_use = 6
+    else:
+        first_use = 7
+
+
+if __name__ == "__main__":
+    demo.launch()