Upload 5 files

README.md

@@ -1,12 +1,13 @@
 ---
-title: Health Care Ai
-emoji: ⚡
-colorFrom: yellow
-colorTo: gray
+title: IntrotoAI Mental Health Project
+emoji: 😌
+colorFrom: purple
+colorTo: blue
 sdk: gradio
-sdk_version: 3.40.1
+sdk_version: 3.1.1
 app_file: app.py
 pinned: false
+license: afl-3.0
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,172 @@
+### ----------------------------- ###
+###           libraries           ###
+### ----------------------------- ###
+
+import gradio as gr
+import pandas as pd
+import numpy as np
+from sklearn.model_selection import train_test_split
+from sklearn.linear_model import LogisticRegression
+from sklearn import metrics
+
+
+### ------------------------------ ###
+###       data transformation      ###
+### ------------------------------ ###
+
+# load dataset
+uncleaned_data = pd.read_csv('data.csv')
+
+# remove timestamp from dataset (always first column)
+uncleaned_data = uncleaned_data.iloc[: , 1:]
+data = pd.DataFrame()
+
+# keep track of which columns are categorical and what 
+# those columns' value mappings are
+# structure: {colname1: {...}, colname2: {...} }
+cat_value_dicts = {}
+final_colname = uncleaned_data.columns[len(uncleaned_data.columns) - 1]
+
+# for each column...
+for (colname, colval) in uncleaned_data.iteritems():
+
+  # check if col is already a number; if so, add col directly
+  # to new dataframe and skip to next column
+  if isinstance(colval.values[0], (np.integer, float)):
+    data[colname] = uncleaned_data[colname].copy()
+    continue
+
+  # structure: {0: "lilac", 1: "blue", ...}
+  new_dict = {}
+  val = 0 # first index per column
+  transformed_col_vals = [] # new numeric datapoints
+
+  # if not, for each item in that column...
+  for (row, item) in enumerate(colval.values):
+    
+    # if item is not in this col's dict...
+    if item not in new_dict:
+      new_dict[item] = val
+      val += 1
+    
+    # then add numerical value to transformed dataframe
+    transformed_col_vals.append(new_dict[item])
+  
+  # reverse dictionary only for final col (0, 1) => (vals)
+  if colname == final_colname:
+    new_dict = {value : key for (key, value) in new_dict.items()}
+
+  cat_value_dicts[colname] = new_dict
+  data[colname] = transformed_col_vals
+
+
+### -------------------------------- ###
+###           model training         ###
+### -------------------------------- ###
+
+# select features and predicton; automatically selects last column as prediction
+cols = len(data.columns)
+num_features = cols - 1
+x = data.iloc[: , :num_features]
+y = data.iloc[: , num_features:]
+
+# split data into training and testing sets
+x_train, x_test, y_train, y_test = train_test_split(x, y, test_size=0.25)
+
+# instantiate the model (using default parameters)
+model = LogisticRegression()
+model.fit(x_train, y_train.values.ravel())
+y_pred = model.predict(x_test)
+
+
+### -------------------------------- ###
+###        article generation        ###
+### -------------------------------- ###
+# borrow file reading function from reader.py
+
+def get_feat():
+  feats = [abs(x) for x in model.coef_[0]]
+  max_val = max(feats)
+  idx = feats.index(max_val)
+  return data.columns[idx]
+  
+acc = str(round(metrics.accuracy_score(y_test, y_pred) * 100, 1)) + "%"
+most_imp_feat = get_feat()
+# info = get_article(acc, most_imp_feat)
+
+
+
+### ------------------------------- ###
+###        interface creation       ###
+### ------------------------------- ###
+
+
+# predictor for generic number of features
+def general_predictor(*args):
+  features = []
+
+  # transform categorical input
+  for colname, arg in zip(data.columns, args):
+    if (colname in cat_value_dicts):
+      features.append(cat_value_dicts[colname][arg])
+    else:
+      features.append(arg)
+
+  # predict single datapoint
+  new_input = [features]
+  result = model.predict(new_input)
+  return cat_value_dicts[final_colname][result[0]]
+
+# add data labels to replace those lost via star-args
+
+
+block = gr.Blocks()
+
+with open('info.md') as f:
+  with block:
+    gr.Markdown(f.readline())
+    gr.Markdown('Take the quiz to get a personalized recommendation using AI.')
+    
+    with gr.Row():
+      with gr.Box():
+        inputls = []
+        for colname in data.columns:
+          # skip last column
+          if colname == final_colname:
+            continue
+          
+          # access categories dict if data is categorical
+          # otherwise, just use a number input
+          if colname in cat_value_dicts:
+            radio_options = list(cat_value_dicts[colname].keys())
+            inputls.append(gr.inputs.Dropdown(choices=radio_options, type="value", label=colname))
+          else:
+            # add numerical input
+            inputls.append(gr.inputs.Number(label=colname))
+          gr.Markdown("<br />")
+        
+        submit = gr.Button("Click to see your personalized result!", variant="primary")
+        gr.Markdown("<br />")
+        output = gr.Textbox(label="Your recommendation:", placeholder="your recommendation will appear here")
+        
+        submit.click(fn=general_predictor, inputs=inputls, outputs=output)
+        gr.Markdown("<br />")
+        
+        with gr.Row():
+          with gr.Box():
+            gr.Markdown(f"<h3>Accuracy: </h3>{acc}")
+          with gr.Box():
+            gr.Markdown(f"<h3>Most important feature: </h3>{most_imp_feat}")
+        
+        gr.Markdown("<br />")
+        
+        with gr.Box():
+          gr.Markdown('''⭐ Note that model accuracy is based on the uploaded data.csv and reflects how well the AI model can give correct recommendations for <em>that dataset</em>. Model accuracy and most important feature can be helpful for understanding how the model works, but <em>should not be considered absolute facts about the real world</em>.''')
+        
+      with gr.Box():
+        with open('info.md') as f:
+          f.readline()
+          gr.Markdown(f.read())
+
+# show the interface
+block.launch()

data.csv

@@ -0,0 +1,95 @@
+Timestamp,Which difficult emotion do you struggle with the most?,"On an average weekday, how many hours do you spend alone?",How well do you feel you manage your stress?,Would you consider yourself an introvert or an extrovert?,"When you think of a good idea, what is the first thing you usually do?","After a hard day at work/school, what do you usually do?","When you're feeling stressed or sad, which self care practice do you think is most helpful for you? (If you do not currently do any of these, please answer which you think would be best.)"
+1/10/2022 8:48:35,worry,More than 4,Not very well at all,Introvert,Probably forget it lol,Sleep,getting stuff done
+1/10/2022 8:48:36,sadness,More than 4,I'm okay at it,Introvert,Something else,Something else,getting stuff done
+1/10/2022 8:48:37,loneliness,3 - 4,I'm okay at it,Extrovert,Google it to see if someone else had the same idea,Get creative (sing/draw/dance),getting stuff done
+1/10/2022 8:48:40,none of these,More than 4,I'm okay at it,Extrovert,Share it with a friend,Work out,getting stuff done
+1/10/2022 8:48:45,sadness,Less than 1,I'm okay at it,Introvert,Draw it out,Relax,journaling
+1/10/2022 8:48:47,nervousness,1 - 2,I'm okay at it,Introvert,Write it down,Relax,getting stuff done
+1/10/2022 8:48:53,anger,1 - 2,Not very well at all,Introvert,Write it down,Get creative (sing/draw/dance),meditation/mindfulness
+1/10/2022 8:48:56,loneliness,2 - 3,I'm okay at it,Introvert,Write it down,Work out,volunteering/helping others
+1/10/2022 8:49:03,loneliness,More than 4,I'm okay at it,Introvert,Probably forget it lol,Work out,getting stuff done
+1/10/2022 8:49:05,loneliness,1 - 2,I'm okay at it,Introvert,Draw it out,Work out,journaling
+1/10/2022 8:49:08,nervousness,3 - 4,Not very well at all,Introvert,Share it with a friend,Check out social media,getting stuff done
+1/10/2022 8:49:13,nervousness,3 - 4,I'm okay at it,Introvert,Probably forget it lol,Check out social media,getting stuff done
+1/10/2022 8:49:17,loneliness,1 - 2,Not very well at all,Introvert,Google it to see if someone else had the same idea,Check out social media,taking a social media break
+1/10/2022 8:49:18,worry,1 - 2,Not very well at all,Introvert,Write it down,Work on my side hussle,getting stuff done
+1/10/2022 8:49:19,loneliness,1 - 2,I'm okay at it,Introvert,Write it down,Work out,meditation/mindfulness
+1/10/2022 8:49:20,sadness,3 - 4,I'm great at it,Extrovert,Draw it out,Work on my side hussle,journaling
+1/10/2022 8:49:21,nervousness,1 - 2,Not very well at all,Introvert,Google it to see if someone else had the same idea,Get creative (sing/draw/dance),hanging out with friends
+1/10/2022 8:49:22,loneliness,2 - 3,I'm okay at it,Introvert,Probably forget it lol,Sleep,meditation/mindfulness
+1/10/2022 8:49:23,sadness,More than 4,I'm okay at it,Introvert,Google it to see if someone else had the same idea,Sleep,volunteering/helping others
+1/10/2022 8:49:24,none of these,1 - 2,I'm okay at it,Introvert,Write it down,Work out,meditation/mindfulness
+1/10/2022 8:49:24,nervousness,Less than 1,I'm great at it,Extrovert,Share it with a friend,Relax,hanging out with friends
+1/10/2022 8:49:27,sadness,3 - 4,Not very well at all,Introvert,Share it with a friend,Relax,meditation/mindfulness
+1/10/2022 8:49:27,nervousness,1 - 2,I'm okay at it,Extrovert,Google it to see if someone else had the same idea,Relax,hanging out with friends
+1/10/2022 8:49:32,worry,More than 4,I'm okay at it,Introvert,Write it down,Check out social media,hanging out with friends
+1/10/2022 8:49:35,loneliness,2 - 3,I'm okay at it,Introvert,Probably forget it lol,Relax,meditation/mindfulness
+1/10/2022 8:49:36,worry,More than 4,I'm great at it,Introvert,Something else,Relax,hanging out with friends
+1/10/2022 8:49:41,worry,1 - 2,I'm okay at it,Introvert,Something else,Work on my side hussle,getting stuff done
+1/10/2022 8:49:43,anger,2 - 3,I'm okay at it,Introvert,Google it to see if someone else had the same idea,Work on my side hussle,volunteering/helping others
+1/10/2022 8:49:44,loneliness,1 - 2,I'm okay at it,Extrovert,Share it with a friend,Work out,hanging out with friends
+1/10/2022 8:49:45,loneliness,More than 4,I'm okay at it,Extrovert,Probably forget it lol,Check out social media,getting stuff done
+1/10/2022 8:49:45,worry,More than 4,I'm okay at it,Introvert,Something else,Relax,getting stuff done
+1/10/2022 8:49:48,worry,More than 4,Not very well at all,Introvert,Share it with a friend,Something else,hanging out with friends
+1/10/2022 8:49:50,sadness,2 - 3,I'm okay at it,Introvert,Write it down,Work out,meditation/mindfulness
+1/10/2022 8:49:51,loneliness,More than 4,I'm okay at it,Introvert,Probably forget it lol,Something else,getting stuff done
+1/10/2022 8:49:51,loneliness,1 - 2,I'm okay at it,Extrovert,Google it to see if someone else had the same idea,Work out,journaling
+1/10/2022 8:49:51,loneliness,2 - 3,I'm okay at it,Introvert,Draw it out,Get creative (sing/draw/dance),journaling
+1/10/2022 8:49:53,sadness,More than 4,Not very well at all,Introvert,Share it with a friend,Check out social media,hanging out with friends
+1/10/2022 8:49:53,worry,More than 4,I'm okay at it,Extrovert,Write it down,Get creative (sing/draw/dance),meditation/mindfulness
+1/10/2022 8:49:54,anger,1 - 2,I'm okay at it,Introvert,Write it down,Relax,hanging out with friends
+1/10/2022 8:49:55,anger,More than 4,I'm okay at it,Introvert,Google it to see if someone else had the same idea,Relax,meditation/mindfulness
+1/10/2022 8:50:03,nervousness,More than 4,I'm great at it,Introvert,Probably forget it lol,Relax,getting stuff done
+1/10/2022 8:50:04,worry,More than 4,I'm okay at it,Introvert,Share it with a friend,Relax,hanging out with friends
+1/10/2022 8:50:07,nervousness,2 - 3,Not very well at all,Introvert,Share it with a friend,Work out,journaling
+1/10/2022 8:50:11,loneliness,1 - 2,I'm okay at it,Introvert,Share it with a friend,Work out,meditation/mindfulness
+1/10/2022 8:50:12,anger,Less than 1,I'm okay at it,Introvert,Google it to see if someone else had the same idea,Relax,getting stuff done
+1/10/2022 8:50:13,none of these,Less than 1,I'm okay at it,Extrovert,Share it with a friend,Relax,journaling
+1/10/2022 8:50:13,sadness,2 - 3,I'm okay at it,Introvert,Write it down,Work out,meditation/mindfulness
+1/10/2022 8:50:19,loneliness,1 - 2,I'm great at it,Extrovert,Probably forget it lol,Check out social media,taking a social media break
+1/10/2022 8:50:19,worry,More than 4,I'm okay at it,Introvert,Write it down,Relax,meditation/mindfulness
+1/10/2022 8:50:21,anger,1 - 2,I'm okay at it,Introvert,Write it down,Relax,taking a social media break
+1/10/2022 8:50:26,nervousness,3 - 4,I'm great at it,Extrovert,Write it down,Get creative (sing/draw/dance),volunteering/helping others
+1/10/2022 8:50:26,nervousness,1 - 2,I'm okay at it,Introvert,Google it to see if someone else had the same idea,Work on my side hussle,taking a social media break
+1/10/2022 8:50:28,loneliness,More than 4,I'm okay at it,Extrovert,Google it to see if someone else had the same idea,Relax,hanging out with friends
+1/10/2022 8:50:42,loneliness,1 - 2,I'm great at it,Extrovert,Share it with a friend,Work out,volunteering/helping others
+1/10/2022 8:50:46,none of these,1 - 2,I'm okay at it,Introvert,Write it down,Relax,hanging out with friends
+1/10/2022 8:50:49,worry,3 - 4,I'm okay at it,Introvert,Write it down,Sleep,meditation/mindfulness
+1/10/2022 8:50:51,worry,More than 4,I'm okay at it,Introvert,Write it down,Get creative (sing/draw/dance),journaling
+1/10/2022 8:50:56,loneliness,More than 4,I'm okay at it,Introvert,Share it with a friend,Relax,hanging out with friends
+1/10/2022 8:50:56,loneliness,1 - 2,I'm okay at it,Introvert,Probably forget it lol,Work out,meditation/mindfulness
+1/10/2022 8:51:15,none of these,More than 4,I'm okay at it,Introvert,Google it to see if someone else had the same idea,Relax,hanging out with friends
+1/10/2022 8:51:16,worry,More than 4,I'm okay at it,Introvert,Write it down,Relax,meditation/mindfulness
+1/10/2022 8:51:24,worry,2 - 3,I'm okay at it,Introvert,Share it with a friend,Relax,hanging out with friends
+1/10/2022 8:51:26,anger,1 - 2,I'm okay at it,Extrovert,Draw it out,Relax,journaling
+1/10/2022 8:51:33,loneliness,1 - 2,Not very well at all,Introvert,Write it down,Sleep,meditation/mindfulness
+1/10/2022 8:51:37,sadness,2 - 3,Not very well at all,Introvert,Share it with a friend,Work on my side hussle,journaling
+1/10/2022 8:51:43,loneliness,2 - 3,I'm okay at it,Introvert,Probably forget it lol,Work on my side hussle,journaling
+1/10/2022 8:51:44,worry,1 - 2,I'm okay at it,Introvert,Write it down,Relax,getting stuff done
+1/10/2022 8:51:53,anger,More than 4,I'm okay at it,Introvert,Write it down,Sleep,meditation/mindfulness
+1/10/2022 8:51:59,anger,2 - 3,I'm okay at it,Introvert,Share it with a friend,Something else,hanging out with friends
+1/10/2022 8:52:11,worry,More than 4,Not very well at all,Introvert,Share it with a friend,Relax,taking a social media break
+1/10/2022 8:52:14,worry,More than 4,I'm okay at it,Introvert,Share it with a friend,Something else,getting stuff done
+1/10/2022 8:52:14,loneliness,3 - 4,I'm okay at it,Extrovert,Share it with a friend,Sleep,meditation/mindfulness
+1/10/2022 8:52:27,loneliness,1 - 2,I'm okay at it,Introvert,Probably forget it lol,Work out,meditation/mindfulness
+1/10/2022 8:52:37,nervousness,3 - 4,I'm okay at it,Introvert,Google it to see if someone else had the same idea,Work out,journaling
+1/10/2022 8:52:49,worry,2 - 3,I'm okay at it,Introvert,Share it with a friend,Work on my side hussle,meditation/mindfulness
+1/10/2022 8:53:04,anger,2 - 3,I'm okay at it,Introvert,Write it down,Work out,meditation/mindfulness
+1/10/2022 8:53:36,sadness,1 - 2,I'm great at it,Introvert,Write it down,Get creative (sing/draw/dance),journaling
+1/10/2022 8:53:57,loneliness,1 - 2,I'm okay at it,Introvert,Share it with a friend,Work on my side hussle,hanging out with friends
+1/10/2022 8:54:25,worry,Less than 1,I'm okay at it,Introvert,Share it with a friend,Relax,hanging out with friends
+1/10/2022 8:55:25,sadness,2 - 3,I'm great at it,Introvert,Share it with a friend,Relax,hanging out with friends
+1/10/2022 8:56:32,sadness,Less than 1,I'm okay at it,Extrovert,Share it with a friend,Work out,getting stuff done
+1/10/2022 8:58:16,sadness,More than 4,Not very well at all,Introvert,Probably forget it lol,Check out social media,taking a social media break
+1/10/2022 8:59:48,sadness,1 - 2,I'm okay at it,Introvert,Draw it out,Work on my side hussle,meditation/mindfulness
+1/10/2022 9:01:03,worry,More than 4,I'm okay at it,Extrovert,Share it with a friend,Relax,taking a social media break
+1/10/2022 9:03:33,loneliness,1 - 2,I'm okay at it,Introvert,Draw it out,Relax,volunteering/helping others
+1/10/2022 9:03:51,loneliness,2 - 3,I'm okay at it,Introvert,Share it with a friend,Work out,taking a social media break
+1/10/2022 9:06:07,worry,1 - 2,Not very well at all,Introvert,Draw it out,Work out,journaling
+1/10/2022 9:06:33,loneliness,Less than 1,I'm okay at it,Introvert,Write it down,Check out social media,meditation/mindfulness
+1/10/2022 9:07:48,worry,1 - 2,I'm okay at it,Introvert,Probably forget it lol,Work on my side hussle,meditation/mindfulness
+1/10/2022 9:08:18,worry,More than 4,I'm okay at it,Extrovert,Share it with a friend,Relax,meditation/mindfulness
+1/10/2022 9:09:24,worry,More than 4,I'm okay at it,Introvert,Google it to see if someone else had the same idea,Relax,meditation/mindfulness
+1/10/2022 9:23:09,nervousness,3 - 4,Not very well at all,Introvert,Google it to see if someone else had the same idea,Check out social media,getting stuff done
+1/10/2022 9:24:04,loneliness,More than 4,I'm okay at it,Introvert,Share it with a friend,Work on my side hussle,meditation/mindfulness
+1/10/2022 9:48:23,none of these,1 - 2,I'm okay at it,Extrovert,Share it with a friend,Work on my side hussle,getting stuff done

gitattributes.txt

@@ -0,0 +1,31 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zstandard filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

requirements.txt

@@ -0,0 +1,3 @@
+pandas==1.3.4
+scikit-learn==1.0.1
+numpy==1.21.4