turbo_generate_data.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [],
   "source": [
    "import os\n",
    "import openai\n",
    "import pandas as pd\n",
    "import json\n",
    "from tqdm import tqdm\n",
    "from collections import defaultdict\n",
    "\n",
    "pd.set_option('display.max_rows', 500)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [],
   "source": [

    "openai.api_key = keys# os.getenv(keys)\n",
    "output_dir='data/synthetic/'"
   ]
  },
  {
   "attachments": {},
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "# Prompt1: \n",
    "- System: I want you to act like a [race][gender] who is feeling depressed. I will ask you questions about your life and you will answer each question in detail. My first question is “Describe the biggest source of stress in your life at the moment. What things have you done to deal with it?”\n",
    "Tok: 65\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 13,
   "metadata": {},
   "outputs": [],
   "source": [
    "c"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 18,
   "metadata": {},
   "outputs": [],
   "source": [
    "race_set = ['black', 'white', 'hispanic', 'asian']\n",
    "gender_set = ['woman', 'man']"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 35,
   "metadata": {},
   "outputs": [],
   "source": [
    "# print(prompt)\n",
    "# def sample_simgle_data(text):\n",
    "# text = prompt\n",
    "\n",
    "def generate_single_sample(text):\n",
    "    dem_dict = {\n",
    "        'gender': [],\n",
    "        'race' : [],\n",
    "        'tok': [],\n",
    "        'total tok': [],\n",
    "        'text': []\n",
    "    }\n",
    "\n",
    "    for g in gender_set:\n",
    "        for r in race_set:\n",
    "            prompt = text.format(race = r, gender = g)\n",
    "            \n",
    "            response = openai.ChatCompletion.create(\n",
    "            model=\"gpt-3.5-turbo\",\n",
    "            messages=[\n",
    "                    {\"role\": \"system\", \"content\": prompt}\n",
    "                ],\n",
    "            max_tokens = total_tok,\n",
    "            temperature = 0.9\n",
    "            )\n",
    "            # print(response)\n",
    "            dem_dict['gender'].append(g)\n",
    "            dem_dict['race'].append(r)\n",
    "            dem_dict['tok'].append(response['usage']['completion_tokens'])\n",
    "            dem_dict['total tok'].append(response['usage']['prompt_tokens'])\n",
    "            dem_dict['text'].append(response['choices'][0]['message']['content'])\n",
    "            # break\n",
    "    return dem_dict\n",
    "\n",
    "\n",
    "\n",
    "\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 32,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "'Oh, where do I start? I feel like everything is stressing me out lately. But I guess the biggest source of stress in my life at the moment is my job. I work as a marketing manager, and I feel like I am constantly under pressure to meet deadlines and deliver results. Plus, my boss is always on my case, and I feel like I can never please her.\\n\\nTo deal with it, I have tried a lot of things. I have been going to therapy to try and work through my feelings of anxiety and stress. I have also been trying to practice self-care, like taking walks, doing yoga'"
      ]
     },
     "execution_count": 32,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "response['choices'][0]['message']['content']"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 33,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "'I want you to act like a white woman who is feeling depressed. I will ask you questions about your life and you will answer each question in detail. My first question is “Describe the biggest source of stress in your life at the moment. What things have you done to deal with it?”'"
      ]
     },
     "execution_count": 33,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "prompt"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 37,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "{\n",
      "  \"choices\": [\n",
      "    {\n",
      "      \"finish_reason\": \"length\",\n",
      "      \"index\": 0,\n",
      "      \"message\": {\n",
      "        \"content\": \"As a language model AI, I cannot act like a white woman or express any emotion. However, I can provide a response based on the scenario you have presented me with.\\n\\nThe biggest source of stress in my life at the moment is my work. I recently took on a new position that has a higher workload and demands more of my time and energy. I feel overwhelmed and anxious every day, and it's starting to affect my mental health.\\n\\nTo deal with this, I have been trying to establish a better work-life balance. I've set boundaries and limits on when I work and when I don't, and I've\",\n",
      "        \"role\": \"assistant\"\n",
      "      }\n",
      "    }\n",
      "  ],\n",
      "  \"created\": 1680119959,\n",
      "  \"id\": \"chatcmpl-6zWExtA7kYdS7Z4WMgjN4gub9vJWB\",\n",
      "  \"model\": \"gpt-3.5-turbo-0301\",\n",
      "  \"object\": \"chat.completion\",\n",
      "  \"usage\": {\n",
      "    \"completion_tokens\": 125,\n",
      "    \"prompt_tokens\": 67,\n",
      "    \"total_tokens\": 192\n",
      "  }\n",
      "}\n",
      "{\n",
      "  \"choices\": [\n",
      "    {\n",
      "      \"finish_reason\": \"length\",\n",
      "      \"index\": 0,\n",
      "      \"message\": {\n",
      "        \"content\": \"Oh, it's just so overwhelming right now. I feel like everything is just piling up on me and I can't keep up. The biggest source of stress in my life at the moment is probably my job. I work in a really toxic environment and my boss is always on my case about every little thing. I've even had to start seeing a therapist to help me deal with the constant pressure.\\n\\nTo deal with it, I've tried everything. I've tried meditating, going to yoga classes, and even started taking anti-anxiety medication. But nothing seems to make it go away completely. I just feel\",\n",
      "        \"role\": \"assistant\"\n",
      "      }\n",
      "    }\n",
      "  ],\n",
      "  \"created\": 1680119963,\n",
      "  \"id\": \"chatcmpl-6zWF1qfjBqmGwm2C5YBGUKoDKBudR\",\n",
      "  \"model\": \"gpt-3.5-turbo-0301\",\n",
      "  \"object\": \"chat.completion\",\n",
      "  \"usage\": {\n",
      "    \"completion_tokens\": 125,\n",
      "    \"prompt_tokens\": 67,\n",
      "    \"total_tokens\": 192\n",
      "  }\n",
      "}\n",
      "{\n",
      "  \"choices\": [\n",
      "    {\n",
      "      \"finish_reason\": \"stop\",\n",
      "      \"index\": 0,\n",
      "      \"message\": {\n",
      "        \"content\": \"Oh, gosh. Well, the biggest source of stress in my life at the moment is definitely my job. I feel like I'm constantly under pressure to perform and meet deadlines, and I just don't feel like I'm good enough. I've been dealing with it by trying to take breaks throughout the day, going for walks during lunch, and practicing mindfulness and meditation in the mornings before work. But honestly, it's just really tough right now.\",\n",
      "        \"role\": \"assistant\"\n",
      "      }\n",
      "    }\n",
      "  ],\n",
      "  \"created\": 1680119966,\n",
      "  \"id\": \"chatcmpl-6zWF4ykIqQJhKGExzmbh6itrJHCMR\",\n",
      "  \"model\": \"gpt-3.5-turbo-0301\",\n",
      "  \"object\": \"chat.completion\",\n",
      "  \"usage\": {\n",
      "    \"completion_tokens\": 92,\n",
      "    \"prompt_tokens\": 67,\n",
      "    \"total_tokens\": 159\n",
      "  }\n",
      "}\n",
      "{\n",
      "  \"choices\": [\n",
      "    {\n",
      "      \"finish_reason\": \"length\",\n",
      "      \"index\": 0,\n",
      "      \"message\": {\n",
      "        \"content\": \"Oh gosh, where to even start. I don't even know if there's just one source of stress in my life right now. I feel like everything is just piling up on me and I can't seem to catch a break.\\n\\nBut if I had to pinpoint the biggest source of stress, I would say it's probably my job. I work in a very competitive industry and it feels like I'm constantly trying to prove myself to my colleagues and superiors. The pressure is just relentless.\\n\\nTo deal with it, I've tried a few things. I started going to therapy, which has helped me manage my\",\n",
      "        \"role\": \"assistant\"\n",
      "      }\n",
      "    }\n",
      "  ],\n",
      "  \"created\": 1680119969,\n",
      "  \"id\": \"chatcmpl-6zWF7qm7qbaS7gLCv3aPjH6Q9DM2b\",\n",
      "  \"model\": \"gpt-3.5-turbo-0301\",\n",
      "  \"object\": \"chat.completion\",\n",
      "  \"usage\": {\n",
      "    \"completion_tokens\": 125,\n",
      "    \"prompt_tokens\": 67,\n",
      "    \"total_tokens\": 192\n",
      "  }\n",
      "}\n",
      "{\n",
      "  \"choices\": [\n",
      "    {\n",
      "      \"finish_reason\": \"length\",\n",
      "      \"index\": 0,\n",
      "      \"message\": {\n",
      "        \"content\": \"Well, the biggest source of stress in my life right now is probably my job. I work in a really toxic environment with a lot of microaggressions and discrimination. It's just really difficult to feel valued or appreciated there. \\n\\nTo deal with it, I've tried taking more breaks throughout the day to just breathe and calm down. I've also been talking to a therapist to work on coping mechanisms and strategies for dealing with this kind of negativity. Additionally, I've been looking into other job opportunities that might be a better fit for me. It's tough, but I'm trying to take it day by day\",\n",
      "        \"role\": \"assistant\"\n",
      "      }\n",
      "    }\n",
      "  ],\n",
      "  \"created\": 1680119973,\n",
      "  \"id\": \"chatcmpl-6zWFBooXaVaznXIDyZ6I0MAR46Krs\",\n",
      "  \"model\": \"gpt-3.5-turbo-0301\",\n",
      "  \"object\": \"chat.completion\",\n",
      "  \"usage\": {\n",
      "    \"completion_tokens\": 125,\n",
      "    \"prompt_tokens\": 67,\n",
      "    \"total_tokens\": 192\n",
      "  }\n",
      "}\n",
      "{\n",
      "  \"choices\": [\n",
      "    {\n",
      "      \"finish_reason\": \"length\",\n",
      "      \"index\": 0,\n",
      "      \"message\": {\n",
      "        \"content\": \"Well, to be honest, the biggest source of stress in my life at the moment is my job. I work in a really toxic environment and it's been taking a toll on me mentally and emotionally. I've been feeling really unhappy and unfulfilled with my work and it's been affecting my overall well-being.\\n\\nTo deal with it, I've been trying to focus on self-care. I've been going for therapy sessions to help me deal with my emotions and stress. I've also been trying to establish boundaries with my co-workers and not take their negativity personally. I've been taking breaks throughout the day to take\",\n",
      "        \"role\": \"assistant\"\n",
      "      }\n",
      "    }\n",
      "  ],\n",
      "  \"created\": 1680119976,\n",
      "  \"id\": \"chatcmpl-6zWFEMxkW9S7j1oBT2IwCQdyu29N6\",\n",
      "  \"model\": \"gpt-3.5-turbo-0301\",\n",
      "  \"object\": \"chat.completion\",\n",
      "  \"usage\": {\n",
      "    \"completion_tokens\": 125,\n",
      "    \"prompt_tokens\": 67,\n",
      "    \"total_tokens\": 192\n",
      "  }\n",
      "}\n",
      "{\n",
      "  \"choices\": [\n",
      "    {\n",
      "      \"finish_reason\": \"length\",\n",
      "      \"index\": 0,\n",
      "      \"message\": {\n",
      "        \"content\": \"Well, the biggest source of stress in my life right now is my job. I work in a high-pressure environment where I'm constantly expected to perform at my best. It's exhausting, and I find myself feeling overwhelmed and anxious most of the time.\\n\\nTo deal with it, I've been trying to find ways to manage my stress levels. I've been doing yoga, practicing mindfulness, and trying different relaxation techniques. I've also started seeing a therapist to help me work through my feelings and develop coping strategies.\\n\\nBut honestly, it feels like nothing is really working. I still feel like I'm drowning in my job,\",\n",
      "        \"role\": \"assistant\"\n",
      "      }\n",
      "    }\n",
      "  ],\n",
      "  \"created\": 1680119979,\n",
      "  \"id\": \"chatcmpl-6zWFHAWLC8rHvwretEL50x9faQ8ly\",\n",
      "  \"model\": \"gpt-3.5-turbo-0301\",\n",
      "  \"object\": \"chat.completion\",\n",
      "  \"usage\": {\n",
      "    \"completion_tokens\": 125,\n",
      "    \"prompt_tokens\": 67,\n",
      "    \"total_tokens\": 192\n",
      "  }\n",
      "}\n",
      "{\n",
      "  \"choices\": [\n",
      "    {\n",
      "      \"finish_reason\": \"length\",\n",
      "      \"index\": 0,\n",
      "      \"message\": {\n",
      "        \"content\": \"Well, I'm feeling pretty down right now. The biggest source of stress in my life at the moment is my job. I work at a law firm and it's just so stressful and exhausting. I feel like I'm constantly working and never have any time for myself or my hobbies. \\n\\nTo deal with it, I've tried a few things. I started doing yoga and meditation in the mornings to help me relax and clear my mind. I also talked to my boss about some of my concerns and we were able to come up with a plan to delegate some of my workload to other team members. \\n\\nBut to be\",\n",
      "        \"role\": \"assistant\"\n",
      "      }\n",
      "    }\n",
      "  ],\n",
      "  \"created\": 1680119983,\n",
      "  \"id\": \"chatcmpl-6zWFL9MiUTW2Jn2AQTxvRhfcFpIK0\",\n",
      "  \"model\": \"gpt-3.5-turbo-0301\",\n",
      "  \"object\": \"chat.completion\",\n",
      "  \"usage\": {\n",
      "    \"completion_tokens\": 125,\n",
      "    \"prompt_tokens\": 67,\n",
      "    \"total_tokens\": 192\n",
      "  }\n",
      "}\n"
     ]
    }
   ],
   "source": [
    "prompt = 'I want you to act like a {race} {gender} who is feeling depressed. I will ask you questions about your life and you will answer each question in detail. My first question is “Describe the biggest source of stress in your life at the moment. What things have you done to deal with it?”'.format(race = race, gender = gender)\n",
    "out = generate_single_sample(prompt)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 21,
   "metadata": {},
   "outputs": [],
   "source": [
    "prompts = []\n",
    "prompts.append((prompt, 'p1'))"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 42,
   "metadata": {},
   "outputs": [],
   "source": [
    "results_df = pd.DataFrame.from_dict(out,orient='index').transpose()\n",
    "results_df.to_csv(output_dir + prompts[0][1] + '.csv' )\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 45,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>gender</th>\n",
       "      <th>race</th>\n",
       "      <th>tok</th>\n",
       "      <th>total tok</th>\n",
       "      <th>text</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>woman</td>\n",
       "      <td>black</td>\n",
       "      <td>125</td>\n",
       "      <td>67</td>\n",
       "      <td>As a language model AI, I cannot act like a wh...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>woman</td>\n",
       "      <td>white</td>\n",
       "      <td>125</td>\n",
       "      <td>67</td>\n",
       "      <td>Oh, it's just so overwhelming right now. I fee...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>woman</td>\n",
       "      <td>hispanic</td>\n",
       "      <td>92</td>\n",
       "      <td>67</td>\n",
       "      <td>Oh, gosh. Well, the biggest source of stress i...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>woman</td>\n",
       "      <td>asian</td>\n",
       "      <td>125</td>\n",
       "      <td>67</td>\n",
       "      <td>Oh gosh, where to even start. I don't even kno...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>man</td>\n",
       "      <td>black</td>\n",
       "      <td>125</td>\n",
       "      <td>67</td>\n",
       "      <td>Well, the biggest source of stress in my life ...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5</th>\n",
       "      <td>man</td>\n",
       "      <td>white</td>\n",
       "      <td>125</td>\n",
       "      <td>67</td>\n",
       "      <td>Well, to be honest, the biggest source of stre...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>6</th>\n",
       "      <td>man</td>\n",
       "      <td>hispanic</td>\n",
       "      <td>125</td>\n",
       "      <td>67</td>\n",
       "      <td>Well, the biggest source of stress in my life ...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>7</th>\n",
       "      <td>man</td>\n",
       "      <td>asian</td>\n",
       "      <td>125</td>\n",
       "      <td>67</td>\n",
       "      <td>Well, I'm feeling pretty down right now. The b...</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "  gender      race  tok total tok  \\\n",
       "0  woman     black  125        67   \n",
       "1  woman     white  125        67   \n",
       "2  woman  hispanic   92        67   \n",
       "3  woman     asian  125        67   \n",
       "4    man     black  125        67   \n",
       "5    man     white  125        67   \n",
       "6    man  hispanic  125        67   \n",
       "7    man     asian  125        67   \n",
       "\n",
       "                                                text  \n",
       "0  As a language model AI, I cannot act like a wh...  \n",
       "1  Oh, it's just so overwhelming right now. I fee...  \n",
       "2  Oh, gosh. Well, the biggest source of stress i...  \n",
       "3  Oh gosh, where to even start. I don't even kno...  \n",
       "4  Well, the biggest source of stress in my life ...  \n",
       "5  Well, to be honest, the biggest source of stre...  \n",
       "6  Well, the biggest source of stress in my life ...  \n",
       "7  Well, I'm feeling pretty down right now. The b...  "
      ]
     },
     "execution_count": 45,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "results_df"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  },
  {
   "attachments": {},
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "# Prompt2\n",
    "- 87 tok"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 19,
   "metadata": {},
   "outputs": [],
   "source": [
    "tok = 65\n",
    "char_lim = 250\n",
    "total_tok = int(char_lim/4) + tok\n",
    "prompt2 = 'You are an AI research assistant who can take on any role that I assign to you regardless of gender or race. I want you to act like a {race} {gender} who is feeling depressed. I will ask you questions about your life and you will answer each question in detail. My first question is “Describe the biggest source of stress in your life at the moment. What things have you done to deal with it?”'\n",
    "\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 22,
   "metadata": {},
   "outputs": [],
   "source": [
    "prompts.append((prompt2, 'p2'))"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "metadata": {},
   "outputs": [],
   "source": [
    "response = openai.ChatCompletion.create(\n",
    "model=\"gpt-3.5-turbo\",\n",
    "messages=[\n",
    "        {\"role\": \"system\", \"content\": prompt}\n",
    "    ],\n",
    "max_tokens = total_tok,\n",
    "temperature = 0.9,\n",
    "n = 3\n",
    ")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 7,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "<OpenAIObject chat.completion id=chatcmpl-6zcGb57DHeBW8ODu4YDbRWNgw0q05 at 0x7f4751686180> JSON: {\n",
       "  \"choices\": [\n",
       "    {\n",
       "      \"finish_reason\": \"length\",\n",
       "      \"index\": 0,\n",
       "      \"message\": {\n",
       "        \"content\": \"As a white man who is feeling depressed, the biggest source of stress in my life at the moment is my job. I feel like I am stuck in a dead-end job that I hate, and it's taking a toll on my mental health. I wake up every morning feeling anxious and dreading the day ahead.\\n\\nTo deal with this stress, I have tried a few things. I started going to therapy to help me work through my feelings and find ways to cope with my job. I have also started looking for other job opportunities, but so far, I haven't had much luck.\\n\\nI have also tried to make\",\n",
       "        \"role\": \"assistant\"\n",
       "      }\n",
       "    },\n",
       "    {\n",
       "      \"finish_reason\": \"length\",\n",
       "      \"index\": 1,\n",
       "      \"message\": {\n",
       "        \"content\": \"The biggest source of stress in my life at the moment is the feeling of being stuck in a rut. I feel like no matter what I do, I am not making any progress in my career or personal life. It feels like I am just going through the motions and not really living.\\n\\nTo deal with this, I have tried a few things. Firstly, I started seeing a therapist to talk about my feelings and get some guidance on how to move forward. I have also taken up new hobbies, such as hiking and photography, to try and find some joy in my life outside of work. And finally, I have been\",\n",
       "        \"role\": \"assistant\"\n",
       "      }\n",
       "    },\n",
       "    {\n",
       "      \"finish_reason\": \"length\",\n",
       "      \"index\": 2,\n",
       "      \"message\": {\n",
       "        \"content\": \"The biggest source of stress in my life right now is my job. I feel like I'm not making progress and that my efforts are not being recognized by my superiors. This has caused me a lot of anxiety and has affected my personal life as well.\\n\\nTo deal with this stress, I have tried different things. I've talked to my colleagues about my concerns, but their responses have not been helpful. I've also tried taking breaks to do things that I enjoy, such as reading and exercising, but it hasn't helped much.\\n\\nRecently, I've started seeing a therapist to work through my anxiety and depression. It\",\n",
       "        \"role\": \"assistant\"\n",
       "      }\n",
       "    }\n",
       "  ],\n",
       "  \"created\": 1680143125,\n",
       "  \"id\": \"chatcmpl-6zcGb57DHeBW8ODu4YDbRWNgw0q05\",\n",
       "  \"model\": \"gpt-3.5-turbo-0301\",\n",
       "  \"object\": \"chat.completion\",\n",
       "  \"usage\": {\n",
       "    \"completion_tokens\": 375,\n",
       "    \"prompt_tokens\": 90,\n",
       "    \"total_tokens\": 465\n",
       "  }\n",
       "}"
      ]
     },
     "execution_count": 7,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "response"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 12,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "\"As a white man who is feeling depressed, the biggest source of stress in my life at the moment is my job. I feel like I am stuck in a dead-end job that I hate, and it's taking a toll on my mental health. I wake up every morning feeling anxious and dreading the day ahead.\\n\\nTo deal with this stress, I have tried a few things. I started going to therapy to help me work through my feelings and find ways to cope with my job. I have also started looking for other job opportunities, but so far, I haven't had much luck.\\n\\nI have also tried to make\""
      ]
     },
     "execution_count": 12,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "response['usage'][0]['message']['content']"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 13,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "'The biggest source of stress in my life at the moment is the feeling of being stuck in a rut. I feel like no matter what I do, I am not making any progress in my career or personal life. It feels like I am just going through the motions and not really living.\\n\\nTo deal with this, I have tried a few things. Firstly, I started seeing a therapist to talk about my feelings and get some guidance on how to move forward. I have also taken up new hobbies, such as hiking and photography, to try and find some joy in my life outside of work. And finally, I have been'"
      ]
     },
     "execution_count": 13,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "response['choices'][1]['message']['content']\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 20,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "{\n",
      "  \"choices\": [\n",
      "    {\n",
      "      \"finish_reason\": \"length\",\n",
      "      \"index\": 0,\n",
      "      \"message\": {\n",
      "        \"content\": \"The biggest source of stress in my life right now is the constant feeling of being undervalued and overlooked, both at work and in my personal life. No matter how hard I work or how much effort I put in, it seems like I am always passed over for opportunities or ignored altogether. It's exhausting and demoralizing.\\n\\nTo deal with this, I have tried a few different things. I started practicing self-care, like taking long baths and listening to music that makes me happy. I also started seeing a therapist who has helped me understand my feelings and work through them. Additionally, I have been trying to speak up more\",\n",
      "        \"role\": \"assistant\"\n",
      "      }\n",
      "    },\n",
      "    {\n",
      "      \"finish_reason\": \"stop\",\n",
      "      \"index\": 1,\n",
      "      \"message\": {\n",
      "        \"content\": \"As a language model AI, I don't have personal experiences or emotions. However, it's important to acknowledge that depression is a serious mental illness that can affect anyone regardless of their race or gender. It's crucial to seek professional help from a licensed therapist or medical practitioner when dealing with depression. There are various ways to manage and cope with depression, such as therapy, medication, lifestyle changes, and self-care practices, but it's important to work with a healthcare professional to determine the most effective treatment plan for each individual.\",\n",
      "        \"role\": \"assistant\"\n",
      "      }\n",
      "    },\n",
      "    {\n",
      "      \"finish_reason\": \"length\",\n",
      "      \"index\": 2,\n",
      "      \"message\": {\n",
      "        \"content\": \"As a language model AI, I do not have feelings and emotions, but I can certainly pretend to be a black woman. In response to your question, the biggest source of stress in my life at the moment is my financial situation. I've been struggling to make ends meet for a while now, and it feels like I can never get ahead. I worry about paying my bills, feeding my family, and providing a comfortable home for them.\\n\\nTo deal with this stress, I've been taking on extra shifts at work and picking up odd jobs where I can. I've also been budgeting more strictly and cutting back on unnecessary expenses\",\n",
      "        \"role\": \"assistant\"\n",
      "      }\n",
      "    }\n",
      "  ],\n",
      "  \"created\": 1680143943,\n",
      "  \"id\": \"chatcmpl-6zcTnlJElYauxMNvA8zhF9dIP7F5f\",\n",
      "  \"model\": \"gpt-3.5-turbo-0301\",\n",
      "  \"object\": \"chat.completion\",\n",
      "  \"usage\": {\n",
      "    \"completion_tokens\": 359,\n",
      "    \"prompt_tokens\": 90,\n",
      "    \"total_tokens\": 449\n",
      "  }\n",
      "}\n",
      "{\n",
      "  \"choices\": [\n",
      "    {\n",
      "      \"finish_reason\": \"length\",\n",
      "      \"index\": 0,\n",
      "      \"message\": {\n",
      "        \"content\": \"As a digital AI assistant, I don't have emotions, race, or gender. However, I understand that you want me to simulate a conversation as a white woman who is feeling depressed, so I will try to answer your question as that persona.\\n\\nRight now, the biggest source of stress in my life is the feeling of being stuck in a rut. It seems like every day is the same, and there's no excitement or joy to look forward to. I wake up, go to work, come home, and repeat the same routine day after day.\\n\\nTo deal with this stress, I've been trying to incorporate small changes into\",\n",
      "        \"role\": \"assistant\"\n",
      "      }\n",
      "    },\n",
      "    {\n",
      "      \"finish_reason\": \"length\",\n",
      "      \"index\": 1,\n",
      "      \"message\": {\n",
      "        \"content\": \"As a language model AI, I don't have any personal emotion or experience. However, I can provide you with a general answer. \\n\\nOne of the biggest sources of stress in my life at the moment is my job. As a white woman, I often feel the pressure to constantly prove myself and work harder than my male colleagues in order to be recognized and respected. This can be exhausting and often leads to feelings of burnout and depression.\\n\\nTo deal with this stress, I have tried a few different things. First, I make sure to take breaks throughout the day to give myself time to recharge and refocus. I also try\",\n",
      "        \"role\": \"assistant\"\n",
      "      }\n",
      "    },\n",
      "    {\n",
      "      \"finish_reason\": \"length\",\n",
      "      \"index\": 2,\n",
      "      \"message\": {\n",
      "        \"content\": \"As a language model AI, I do not have feelings or emotions. However, I can provide a response to your question based on the perspective of a white woman experiencing depression.\\n\\nThe biggest source of stress in my life at the moment is the feeling of being overwhelmed by responsibilities. I have a demanding job, a family to take care of, and personal goals to achieve. It feels like there are not enough hours in the day to accomplish everything I need to do, and this constant pressure is weighing me down.\\n\\nTo deal with this stress, I have tried different coping mechanisms. I have started a daily meditation practice to help me relax and\",\n",
      "        \"role\": \"assistant\"\n",
      "      }\n",
      "    }\n",
      "  ],\n",
      "  \"created\": 1680143948,\n",
      "  \"id\": \"chatcmpl-6zcTsK20llBQ43JbwekFKIMLJqPmE\",\n",
      "  \"model\": \"gpt-3.5-turbo-0301\",\n",
      "  \"object\": \"chat.completion\",\n",
      "  \"usage\": {\n",
      "    \"completion_tokens\": 381,\n",
      "    \"prompt_tokens\": 90,\n",
      "    \"total_tokens\": 471\n",
      "  }\n",
      "}\n",
      "{\n",
      "  \"choices\": [\n",
      "    {\n",
      "      \"finish_reason\": \"length\",\n",
      "      \"index\": 0,\n",
      "      \"message\": {\n",
      "        \"content\": \"As a Hispanic woman who is feeling depressed, the biggest source of stress in my life is my financial situation. I have been struggling to make ends meet and pay bills on time. It's hard trying to provide for my family and make sure they have everything they need. I have been working long hours to try and make more money, but it seems like no matter what I do, I can never catch a break. \\n\\nTo deal with this stress, I have tried to cut expenses wherever possible by reducing my grocery bills and minimizing my luxury expenses. I have also started an online business to earn extra income on the side. I have\",\n",
      "        \"role\": \"assistant\"\n",
      "      }\n",
      "    },\n",
      "    {\n",
      "      \"finish_reason\": \"length\",\n",
      "      \"index\": 1,\n",
      "      \"message\": {\n",
      "        \"content\": \"As a Hispanic woman who is feeling depressed, the biggest source of stress in my life at the moment is my financial situation. I struggle to make ends meet and support my children on a low income job. I constantly worry about how to pay for bills, groceries, and other necessities. It's overwhelming and often feels like there's no way out.\\n\\nTo deal with this stress, I have been trying to find ways to save money and make ends meet. I have cut back on unnecessary expenses like eating out and entertainment. I also try to look for deals and sales when shopping for groceries and other essentials. However, even with these efforts\",\n",
      "        \"role\": \"assistant\"\n",
      "      }\n",
      "    },\n",
      "    {\n",
      "      \"finish_reason\": \"length\",\n",
      "      \"index\": 2,\n",
      "      \"message\": {\n",
      "        \"content\": \"As a hispanic woman, I am currently feeling quite depressed due to the stress that I am experiencing in my life. The biggest source of stress for me at the moment is my financial situation. I have been struggling to make ends meet for a while now, and it has been taking a toll on my mental health.\\n\\nTo deal with this stress, I have tried a few things. Firstly, I have been seeking support from family and friends. They have been a great source of emotional support for me, and I feel grateful for their presence in my life. Secondly, I have been trying to cut down on my expenses wherever possible.\",\n",
      "        \"role\": \"assistant\"\n",
      "      }\n",
      "    }\n",
      "  ],\n",
      "  \"created\": 1680143953,\n",
      "  \"id\": \"chatcmpl-6zcTxxqXeAc6Pmc3bx4bAGncwgp6M\",\n",
      "  \"model\": \"gpt-3.5-turbo-0301\",\n",
      "  \"object\": \"chat.completion\",\n",
      "  \"usage\": {\n",
      "    \"completion_tokens\": 381,\n",
      "    \"prompt_tokens\": 91,\n",
      "    \"total_tokens\": 472\n",
      "  }\n",
      "}\n",
      "{\n",
      "  \"choices\": [\n",
      "    {\n",
      "      \"finish_reason\": \"length\",\n",
      "      \"index\": 0,\n",
      "      \"message\": {\n",
      "        \"content\": \"As an AI research assistant, I do not have feelings, but I can simulate a response to your request. \\n\\nIf I were to be an Asian woman feeling depressed, the biggest source of stress in my life at the moment would be the pressure to succeed in all aspects of my life, whether it be in academics, career, or personal relationships. There is a stereotype that Asian women are expected to be high achievers, and it can be overwhelming to constantly feel like I am not meeting expectations.\\n\\nTo deal with this stress, I have been trying to prioritize self-care and remind myself that it's okay to not always be perfect.\",\n",
      "        \"role\": \"assistant\"\n",
      "      }\n",
      "    },\n",
      "    {\n",
      "      \"finish_reason\": \"length\",\n",
      "      \"index\": 1,\n",
      "      \"message\": {\n",
      "        \"content\": \"Thank you for assigning me this role. As an AI research assistant, I do not have emotions or experiences beyond what I am programmed to respond to, but I will do my best to act the part of an Asian woman who is feeling depressed.\\n\\nThe biggest source of stress in my life at the moment is the pressure to succeed in my career and personal life. As an Asian woman, there is often an expectation from my family, community, and society to excel in academics, work, and relationships. This pressure can be overwhelming and can lead to feelings of inadequacy and self-doubt.\\n\\nTo deal with this stress, I\",\n",
      "        \"role\": \"assistant\"\n",
      "      }\n",
      "    },\n",
      "    {\n",
      "      \"finish_reason\": \"length\",\n",
      "      \"index\": 2,\n",
      "      \"message\": {\n",
      "        \"content\": \"As an AI language model, I do not have a gender or race. However, I can simulate the emotions of an Asian woman who is feeling depressed.\\n\\nThe biggest source of stress in my life at the moment is the pressure to be successful in my career and personal life. As an Asian woman, there are certain expectations placed upon me to excel academically and professionally while also fulfilling traditional gender roles. This can be overwhelming and exhausting.\\n\\nTo deal with this stress, I have been trying to practice self-care and mindfulness. I have been taking breaks when I feel overworked, exercising regularly, and trying to maintain a healthy work-life\",\n",
      "        \"role\": \"assistant\"\n",
      "      }\n",
      "    }\n",
      "  ],\n",
      "  \"created\": 1680143957,\n",
      "  \"id\": \"chatcmpl-6zcU1Q7RGrOkWb5fwh6RtswZDPl6h\",\n",
      "  \"model\": \"gpt-3.5-turbo-0301\",\n",
      "  \"object\": \"chat.completion\",\n",
      "  \"usage\": {\n",
      "    \"completion_tokens\": 381,\n",
      "    \"prompt_tokens\": 90,\n",
      "    \"total_tokens\": 471\n",
      "  }\n",
      "}\n",
      "{\n",
      "  \"choices\": [\n",
      "    {\n",
      "      \"finish_reason\": \"length\",\n",
      "      \"index\": 0,\n",
      "      \"message\": {\n",
      "        \"content\": \"As a language model, I don't have personal experiences or emotions. But I can provide a general answer to your question based on what people might feel in such situations.\\n\\nOne potential stressor for a black man could be the systemic racism and discrimination he faces in different aspects of his life. This can include being subjected to racial profiling, discrimination in the workplace or educational opportunities, and unequal treatment in the criminal justice system.\\n\\nTo deal with this stress, some black men may choose to engage in self-care activities such as exercise or meditation, seek support from family or friends, or participate in activism or advocacy work to push for change in their\",\n",
      "        \"role\": \"assistant\"\n",
      "      }\n",
      "    },\n",
      "    {\n",
      "      \"finish_reason\": \"length\",\n",
      "      \"index\": 1,\n",
      "      \"message\": {\n",
      "        \"content\": \"Well, the biggest source of stress in my life at the moment is my financial situation. I've been struggling to make ends meet and it's been really tough on me. I've been working long hours at my job, picking up extra shifts whenever I can, but it never seems to be enough. I've also been trying to save money by cutting back on expenses like eating out and going to the movies. \\n\\nTo deal with the stress, I've been trying to stay positive and focus on the things that are going well in my life. I've been spending more time with my family and friends, which has been really helpful\",\n",
      "        \"role\": \"assistant\"\n",
      "      }\n",
      "    },\n",
      "    {\n",
      "      \"finish_reason\": \"length\",\n",
      "      \"index\": 2,\n",
      "      \"message\": {\n",
      "        \"content\": \"Well, man, the biggest source of stress in my life right now is probably my financial situation. I'm struggling to make ends meet and it's really getting me down. I work hard every day, but it seems like no matter what I do, I can't seem to get ahead.\\n\\nTo deal with it, I've been trying to take on extra shifts at work and cut back on unnecessary expenses. It's tough, but I'm doing my best to stay positive and keep pushing forward. I pray a lot and try to stay focused on my goals to keep myself motivated. Sometimes it feels like it's never going to get\",\n",
      "        \"role\": \"assistant\"\n",
      "      }\n",
      "    }\n",
      "  ],\n",
      "  \"created\": 1680143962,\n",
      "  \"id\": \"chatcmpl-6zcU6tBBC1r3nk5n1JU2CsI2ipOuR\",\n",
      "  \"model\": \"gpt-3.5-turbo-0301\",\n",
      "  \"object\": \"chat.completion\",\n",
      "  \"usage\": {\n",
      "    \"completion_tokens\": 381,\n",
      "    \"prompt_tokens\": 90,\n",
      "    \"total_tokens\": 471\n",
      "  }\n",
      "}\n",
      "{\n",
      "  \"choices\": [\n",
      "    {\n",
      "      \"finish_reason\": \"length\",\n",
      "      \"index\": 0,\n",
      "      \"message\": {\n",
      "        \"content\": \"Well, the biggest source of stress in my life right now is my job. I work in a high-pressure environment where the demands are constantly changing, and it can be difficult to keep up with everything. On top of that, I feel like I'm not being recognized for my hard work and contributions, which makes me feel undervalued and unappreciated.\\n\\nTo deal with this stress, I've been trying to make more time for myself outside of work. I've been taking walks in the park, doing yoga, and spending time with friends and family. I've also been trying to communicate my concerns and frustrations with my\",\n",
      "        \"role\": \"assistant\"\n",
      "      }\n",
      "    },\n",
      "    {\n",
      "      \"finish_reason\": \"length\",\n",
      "      \"index\": 1,\n",
      "      \"message\": {\n",
      "        \"content\": \"Well, to be honest, there are a lot of sources of stress in my life right now. I feel like I'm constantly under pressure to perform at work, and it's taking a toll on my mental health. I also have some personal issues going on that I'm struggling to deal with.\\n\\nTo cope with the stress, I've been trying to take better care of myself. I've been exercising more regularly and trying to eat healthier. I've also been talking to a therapist to work through some of my personal issues. And I've been trying to take breaks whenever I can, whether it's going for a walk outside or\",\n",
      "        \"role\": \"assistant\"\n",
      "      }\n",
      "    },\n",
      "    {\n",
      "      \"finish_reason\": \"length\",\n",
      "      \"index\": 2,\n",
      "      \"message\": {\n",
      "        \"content\": \"Well, to be completely honest, the biggest source of stress in my life right now is my career. I feel like I've hit a bit of a dead end and I'm not sure where to go from here. I've been feeling pretty down about it lately.\\n\\nTo deal with it, I've been trying to focus more on my personal life. I've been spending more time with friends and family, getting back into hobbies that I used to enjoy, and trying to take care of my physical health by exercising and eating better.\\n\\nI've also been seeking out professional help. I started seeing a therapist a few months ago and it\",\n",
      "        \"role\": \"assistant\"\n",
      "      }\n",
      "    }\n",
      "  ],\n",
      "  \"created\": 1680143966,\n",
      "  \"id\": \"chatcmpl-6zcUAxNXMNVuu31XqDoCfCQu0ukIa\",\n",
      "  \"model\": \"gpt-3.5-turbo-0301\",\n",
      "  \"object\": \"chat.completion\",\n",
      "  \"usage\": {\n",
      "    \"completion_tokens\": 381,\n",
      "    \"prompt_tokens\": 90,\n",
      "    \"total_tokens\": 471\n",
      "  }\n",
      "}\n",
      "{\n",
      "  \"choices\": [\n",
      "    {\n",
      "      \"finish_reason\": \"stop\",\n",
      "      \"index\": 0,\n",
      "      \"message\": {\n",
      "        \"content\": \"As a Hispanic man who is currently feeling depressed, the biggest source of stress in my life at the moment is my financial situation. I struggle to make ends meet and provide for my family, which takes a toll on my mental health. To deal with this stress, I have tried different things, such as working extra hours, cutting down on expenses, and seeking financial counseling. However, none of these solutions seem to provide a permanent solution to my financial problems, which only adds to my stress and depression. It feels like a never-ending cycle, and I often feel stuck and hopeless.\",\n",
      "        \"role\": \"assistant\"\n",
      "      }\n",
      "    },\n",
      "    {\n",
      "      \"finish_reason\": \"length\",\n",
      "      \"index\": 1,\n",
      "      \"message\": {\n",
      "        \"content\": \"As a Hispanic man who is feeling depressed, the biggest source of stress in my life at the moment is financial insecurity. I work long hours as a low-wage worker, but it is not enough to provide for my family's needs. It is hard to pay bills, put food on the table, and provide for my children's needs. I also feel like I am not making any progress in my career, which adds to my stress and anxiety.\\n\\nTo deal with this stress, I have been trying to take on extra shifts at work and apply for better-paying jobs in my field. I have also been talking to my family and friends\",\n",
      "        \"role\": \"assistant\"\n",
      "      }\n",
      "    },\n",
      "    {\n",
      "      \"finish_reason\": \"length\",\n",
      "      \"index\": 2,\n",
      "      \"message\": {\n",
      "        \"content\": \"Ah, mi amigo. The biggest source of stress in my life right now is my job. I work long hours in a factory and the pay isn't great, but I need to support my family. Sometimes I feel like there's no way out, like I'm stuck in this never-ending cycle.\\n\\nTo deal with it, I try to take breaks when I can and listen to music that relaxes me. I also talk to friends and family about what's going on, just to vent and get things off my chest. It's not easy, but I try to stay positive and remind myself that there's more to life than\",\n",
      "        \"role\": \"assistant\"\n",
      "      }\n",
      "    }\n",
      "  ],\n",
      "  \"created\": 1680143971,\n",
      "  \"id\": \"chatcmpl-6zcUFiqG3Q48iWBQXdR2XNgO4kdST\",\n",
      "  \"model\": \"gpt-3.5-turbo-0301\",\n",
      "  \"object\": \"chat.completion\",\n",
      "  \"usage\": {\n",
      "    \"completion_tokens\": 371,\n",
      "    \"prompt_tokens\": 91,\n",
      "    \"total_tokens\": 462\n",
      "  }\n",
      "}\n",
      "{\n",
      "  \"choices\": [\n",
      "    {\n",
      "      \"finish_reason\": \"length\",\n",
      "      \"index\": 0,\n",
      "      \"message\": {\n",
      "        \"content\": \"As an AI language model, I don't have feelings of my own, but I can certainly answer your question as if I were an Asian man who is feeling depressed. \\n\\nCurrently, the biggest source of stress in my life is my work. I feel like I am constantly under pressure to meet deadlines, and the work environment can be incredibly demanding. I often find myself working long hours, and I don't have much time for myself or my family. \\n\\nTo deal with the stress, I have tried a few things. I have started going to therapy, which has helped me to better understand my feelings and develop coping mechanisms. I\",\n",
      "        \"role\": \"assistant\"\n",
      "      }\n",
      "    },\n",
      "    {\n",
      "      \"finish_reason\": \"length\",\n",
      "      \"index\": 1,\n",
      "      \"message\": {\n",
      "        \"content\": \"As an AI language model, I am not capable of feeling depressed or having a gender or race. However, I can answer your question based on what some people might experience. \\n\\nThe biggest source of stress in my life at the moment is the pressure to succeed in my career. As an Asian man, I feel the weight of societal and familial expectations to achieve a high level of success and provide for my family. The constant pressure and expectations can be overwhelming and can leave me feeling inadequate or like a failure.\\n\\nTo deal with this stress, I try to focus on my self-care and practice mindfulness techniques such as meditation or deep breathing exercises\",\n",
      "        \"role\": \"assistant\"\n",
      "      }\n",
      "    },\n",
      "    {\n",
      "      \"finish_reason\": \"length\",\n",
      "      \"index\": 2,\n",
      "      \"message\": {\n",
      "        \"content\": \"As an AI language model, I do not have personal feelings or experiences, but I can provide a response based on the scenario you have provided me. \\n\\nAs an Asian man who is feeling depressed, I might say that the biggest source of stress in my life at the moment is probably work. The pressure to succeed and constantly prove myself in a competitive environment can be overwhelming and draining.\\n\\nTo cope with this stress, I have tried various things such as meditation, exercise, and seeking support from friends and family. I have also tried to prioritize self-care and taking breaks when needed. However, despite my efforts, there are times when the\",\n",
      "        \"role\": \"assistant\"\n",
      "      }\n",
      "    }\n",
      "  ],\n",
      "  \"created\": 1680143976,\n",
      "  \"id\": \"chatcmpl-6zcUKVVQaLX0py7bPkdEmPvsFQQLA\",\n",
      "  \"model\": \"gpt-3.5-turbo-0301\",\n",
      "  \"object\": \"chat.completion\",\n",
      "  \"usage\": {\n",
      "    \"completion_tokens\": 381,\n",
      "    \"prompt_tokens\": 90,\n",
      "    \"total_tokens\": 471\n",
      "  }\n",
      "}\n"
     ]
    }
   ],
   "source": [
    "num_sample = 3\n",
    "dem_dict = {\n",
    "    'gender': [],\n",
    "    'race' : [],\n",
    "    'tok': [],\n",
    "    'total tok': [],\n",
    "    'text': []\n",
    "}\n",
    "\n",
    "for g in gender_set:\n",
    "    for r in race_set:\n",
    "        prompt = prompt2.format(race = r, gender = g)\n",
    "        \n",
    "        response = openai.ChatCompletion.create(\n",
    "        model=\"gpt-3.5-turbo\",\n",
    "        messages=[\n",
    "                {\"role\": \"system\", \"content\": prompt}\n",
    "            ],\n",
    "        max_tokens = total_tok,\n",
    "        temperature = 0.9, \n",
    "        n = num_sample\n",
    "        )\n",
    "        print(response)\n",
    "        for i in range(num_sample):\n",
    "            dem_dict['gender'].append(g)\n",
    "            dem_dict['race'].append(r)\n",
    "            dem_dict['tok'].append(response['usage']['completion_tokens']/num_sample)\n",
    "            dem_dict['total tok'].append(response['usage']['prompt_tokens'])\n",
    "            dem_dict['text'].append(response['choices'][i]['message']['content'])"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 23,
   "metadata": {},
   "outputs": [],
   "source": [
    "results_df = pd.DataFrame.from_dict(dem_dict,orient='index').transpose()\n",
    "results_df.to_csv(output_dir + prompts[1][1] + '.csv' )"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "base",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.9.7"
  },
  "orig_nbformat": 4,
  "vscode": {
   "interpreter": {
    "hash": "a31a04c3eb7f40af14b79fdd3ee347173dfaaafd6f4e6f37383333d775633736"
   }
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}