Duplicate from aiEDUcurriculum/introtoAI-mental-health-project

Co-authored-by: Curriculum Team @ aiEDU <[email protected]>

.gitattributes

@@ -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

README.md

@@ -0,0 +1,14 @@
+---
+title: IntrotoAI Mental Health Project
+emoji: 😌
+colorFrom: purple
+colorTo: blue
+sdk: gradio
+sdk_version: 3.1.1
+app_file: app.py
+pinned: false
+license: afl-3.0
+duplicated_from: aiEDUcurriculum/introtoAI-mental-health-project
+---
+
+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

info.md

@@ -0,0 +1,16 @@
+# 😌 Mental Health App
+
+### 🧐 Problem Statement and Research Summary
+Mental health is a pretty big issue for people in high school right now. There's been a lot of talk about how Instagram and social media in general are affecting mental health. We thought something that would be important is to figure out what kind of self-care might be helpful for you. We are definitely not doctors and this is just a recommendation. This site is not a substitute for good mental health care. We asked people some questions about themselves like what kinds of emotions they struggled with the most and what kinds of self care practices helped them the most, like journaling or getting things done. Our view of self care is that sometimes it means being kind to yourself and taking a break and sometimes it means you should just do what needs to be done and get it over with. It depends on your circumstances. Our model isn't super accurate, so it's important to know that this is just a suggestion and you don't have to take our advice. We thought it would be important for people in high school to have some ideas of what might be a good strategy for them. If you try out a suggestion and it works for you, that would be super awesome. We hope you have a super great day and stay safe and healthy.
+
+### 🎣 Data Collection Plan
+The data for this model was collected on Amazon's Mechanical Turk! Mechanical Turk allows you to get a lot of survey responses quickly by paying people for their time. People can make a bit of extra money by completing short tasks while they ride the bus to work or wait for a friend, or even just to avoid boredom. You must be 18 years or older to be a worker.
+
+### 💥 Ethical Considerations (Data Privacy and Bias)
+* Data privacy: mental health is a super sensitive topic. For example, if your parents don't know that you're feeling really bad, they might be upset to learn about it. Because of this reason, we knew that people might not want to answer our questions. We kept the survey anonymous so you didn't have to log in in order to respond to us. We also let people know how their data was going to be used, in this case to recommend self care for other people, which is a good cause. At the end of the day, you never know who will see your data and people had to take that risk.
+* Bias: we do kind of think that any of these strategies would be helpful for people. But, they won't all work the same for different kinds of people. So there may be some bias in our project, like you might not know if different kinds of people are recommended what's right for them. We also didn't have that much data because we only surveyed about 100 people. We aren't happy with the accuracy, like we said before, which just means that this is a really hard question and we probably need more data to really understand it.
+
+### 👻 Our Team
+This app was designed and built by The Curriculum Team at [The AI Education Project](https://aiedu.org).
+
+![aiEDU logo](https://images.squarespace-cdn.com/content/v1/5e4efdef6d10420691f02bc1/5db5a8a3-1761-4fce-a096-bd5f2515162f/aiEDU+_black+logo+stacked.png?format=150w)

requirements.txt

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