{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 2,
   "id": "555dce1f-2f19-428c-bf46-7ad90e294d0e",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "3.9.16 | packaged by conda-forge | (main, Feb  1 2023, 21:38:11) \n",
      "[Clang 14.0.6 ]\n"
     ]
    }
   ],
   "source": [
    "# use svg graphics, display inline\n",
    "%matplotlib inline\n",
    "%config InlineBackend.figure_format = 'svg'\n",
    "\n",
    "import glob\n",
    "import re\n",
    "import copy\n",
    "import sys\n",
    "\n",
    "# basic scientific computing imports\n",
    "import pandas as pd\n",
    "import numpy as np\n",
    "\n",
    "# ai\n",
    "from tqdm.notebook import tnrange, tqdm\n",
    "from langchain.chat_models import ChatOpenAI\n",
    "from langchain.chains import LLMChain\n",
    "from langchain.llms import OpenAI\n",
    "from langchain.prompts import PromptTemplate\n",
    "\n",
    "np.random.seed(42)\n",
    "\n",
    "# Don't usually do this but OpenAI an langchain throw lots of warnings\n",
    "import warnings\n",
    "warnings.filterwarnings(\"ignore\")\n",
    "\n",
    "print(sys.version)"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "3e17a345-8e99-42c3-843b-ea169f083271",
   "metadata": {},
   "source": [
    "# Objective\n",
    "Create a program that designs a creative family photoshoot for every US holiday\n",
    "\n",
    "# Steps\n",
    "- Generate a list of us holidays and dates\n",
    "- Generate a photoshoot for each holiday\n",
    "- Generate a midjourney prompt for each shoot\n",
    "- Generate a shopping list for each one"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "id": "052bdca9-78df-4626-a2cc-3cb3dbbcac08",
   "metadata": {},
   "outputs": [],
   "source": [
    "def convert_string_to_tuples(holiday_string):\n",
    "    return [tuple(line.split(', ')) for line in holiday_string.split('\\n')]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "id": "50507c02-905d-4d33-8ac1-fd1843b84dcd",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "[('01/01', \"New Year's Day\"),\n",
       " ('01/18', 'Martin Luther King Jr. Day'),\n",
       " ('02/12', 'Lunar New Year (approximate date)'),\n",
       " ('02/14', \"Valentine's Day\"),\n",
       " ('03/17', \"St. Patrick's Day\"),\n",
       " ('04/04', 'Easter (approximate date)'),\n",
       " ('05/09', \"Mother's Day\"),\n",
       " ('05/31', 'Memorial Day'),\n",
       " ('06/19', 'Juneteenth'),\n",
       " ('06/20', \"Father's Day\"),\n",
       " ('07/04', 'Independence Day'),\n",
       " ('09/06', 'Labor Day'),\n",
       " ('10/11', 'Columbus Day'),\n",
       " ('10/31', 'Halloween'),\n",
       " ('11/11', 'Veterans Day'),\n",
       " ('11/25', 'Thanksgiving Day'),\n",
       " ('12/24', 'Christmas Eve'),\n",
       " ('12/25', 'Christmas Day'),\n",
       " ('',),\n",
       " ('South Korean Holidays:',),\n",
       " ('01/01', \"New Year's Day\"),\n",
       " ('03/01', 'Independence Movement Day'),\n",
       " ('05/05', \"Children's Day\"),\n",
       " ('06/06', 'Memorial Day'),\n",
       " ('08/15', 'Liberation Day'),\n",
       " ('09/20', 'Chuseok (approximate date)'),\n",
       " ('10/03', 'National Foundation Day'),\n",
       " ('10/09', 'Hangul Proclamation Day')]"
      ]
     },
     "execution_count": 4,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "generate_holidays_prompt = '''\n",
    "Generate a list of all widely celebrated US holidays. Include unofficial but fun holidays like Halloween and Father's day.\n",
    "Be sure to include Lunar New Year (use an approximate date if necessary)\n",
    "Include a few major South Korean holidays as well.\n",
    "\n",
    "Example:\n",
    "01/01, New Year's Day\n",
    "10/31, Halloween\n",
    "\n",
    "List of Holidays:\n",
    "'''\n",
    "\n",
    "llm = OpenAI(model_name='gpt-4', temperature=0.4)\n",
    "holidays_string = llm(generate_holidays_prompt)\n",
    "holidays_data = convert_string_to_tuples(holidays_string)\n",
    "\n",
    "holidays_data"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "f35cc7ad-3d5d-4cb9-8af3-6f701caec9d7",
   "metadata": {},
   "source": [
    "### Clean up the list"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "id": "3058b057-3511-4c18-9e17-952891dfca43",
   "metadata": {},
   "outputs": [],
   "source": [
    "holidays_data = [x for x in holidays_data if '/' in x[0] and len(x) == 2]\n",
    "holidays_data.sort(key=lambda x: x[0])"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "3531a092-a46c-4dee-9dc1-5e078456cc98",
   "metadata": {},
   "source": [
    "### Generate the data"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 8,
   "id": "699c05da-f976-4525-a987-900b1712cb04",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "d1074ee1c22c471cba1d2f53a19c8727",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "  0%|          | 0/26 [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    }
   ],
   "source": [
    "generate_photoshoot_prompt = '''\n",
    "### Task\n",
    "Describe a creative, tasteful portrait photo that incorporates a given US holiday.\n",
    "The subject is an adorable 7 month old baby girl.\n",
    "The photoshoot concept should be creative and simple.\n",
    "\n",
    "### Holiday\n",
    "{holiday}\n",
    "\n",
    "### Photoshoot concept\n",
    "'''\n",
    "\n",
    "\n",
    "photoshoot_concepts = []\n",
    "for _, holiday in tqdm(holidays_data):\n",
    "    # I think this prevents context from accumulating, which is desirable here\n",
    "    llm = OpenAI(model_name='gpt-4', temperature=0.4)\n",
    "    shoot_concept = llm(generate_photoshoot_prompt.format(holiday=holiday))\n",
    "    photoshoot_concepts.append(shoot_concept)\n",
    "    \n",
    "\n",
    "# photoshoot_concepts[0]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 12,
   "id": "e5ef8fc5-4fb8-4023-816f-80586bcb834a",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "8df2e4ec81154ed49e9d21bb55b1d55f",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "  0%|          | 0/26 [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    }
   ],
   "source": [
    "generate_mj_prompt_prompt = '''\n",
    "### Task\n",
    "Convert a description of a photoshoot into a more descriptive photoshoot prompt that is similar to the example prompt. Ensure that it includes the following:\n",
    "- A mention of the subject, a 7 month old baby girl\n",
    "- A description of the scene and the pose\n",
    "- A description of the camera and it's settings\n",
    "\n",
    "Ensure that the prompt is no longer than 5 sentences!\n",
    "\n",
    "### Example Output Prompt\n",
    "Set within a cozy, naturally lit room, adorned with gentle touches of fantastical elements like soft, ethereal fairy lights, and gauzy pastel-hued fabrics. Our model, the 7-month old baby, lies comfortably on a plush, cloud-shaped pillow, adorned with a tiny crown, capturing the essence of a fantasy prince/princess. For lighting, large windows allow the soft glow of early morning light to pour in, supplemented with a softbox for balanced illumination. Canon EOS R5, 50mm f/1.2 lens, sharp details, smooth bokeh effect, 1/250 shutter speed, f/2.8 aperture, ISO 200.\n",
    "\n",
    "### Input description\n",
    "{description}\n",
    "\n",
    "### Output Prompt\n",
    "'''\n",
    "\n",
    "\n",
    "\n",
    "prompts = []\n",
    "for shoot in tqdm(photoshoot_concepts):\n",
    "    llm = OpenAI(model_name='gpt-3.5-turbo', temperature=0.4)\n",
    "    prompt = llm(generate_mj_prompt_prompt.format(description=shoot))\n",
    "    prompts.append(prompt)\n",
    "    \n",
    "\n",
    "# prompts[0]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 17,
   "id": "cce5387b-e484-4ce1-8eee-62e073a62e41",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "d6a5e833eb0d422aadcff80eec9d758a",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "  0%|          | 0/26 [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "Retrying langchain.llms.openai.completion_with_retry.<locals>._completion_with_retry in 4.0 seconds as it raised APIConnectionError: Error communicating with OpenAI: ('Connection aborted.', TimeoutError(60, 'Operation timed out')).\n"
     ]
    }
   ],
   "source": [
    "generate_shopping_list_prompt = '''\n",
    "### Task\n",
    "Generate a list of things a photographer would need to buy to put on the following photoshoot.\n",
    "\n",
    "### Photoshoot description\n",
    "{description}\n",
    "\n",
    "### Shopping list\n",
    "'''\n",
    "\n",
    "shopping_lists = []\n",
    "for shoot in tqdm(photoshoot_concepts):\n",
    "    llm = OpenAI(model_name='gpt-3.5-turbo', temperature=0.4)\n",
    "    shopping_list = llm(generate_shopping_list_prompt.format(description=shoot))\n",
    "    shopping_lists.append(shopping_list)\n",
    "    \n",
    "\n",
    "# shopping_lists[0]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 21,
   "id": "15ec7e8c-8298-4860-a2b1-0067b8cb3552",
   "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>date</th>\n",
       "      <th>holiday</th>\n",
       "      <th>title</th>\n",
       "      <th>photoshoot_description</th>\n",
       "      <th>midjourney_prompt</th>\n",
       "      <th>shopping_list</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>01/01</td>\n",
       "      <td>New Year's Day</td>\n",
       "      <td>Baby's First New Year's Sparkle</td>\n",
       "      <td>Concept: For this New Year's Day themed photos...</td>\n",
       "      <td>Our star of the portrait, a 7-month-old baby g...</td>\n",
       "      <td>1. White or light gray backdrop\\n2. Cute, whit...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>01/01</td>\n",
       "      <td>New Year's Day</td>\n",
       "      <td>A New Year's Baby Dream</td>\n",
       "      <td>Concept: For this New Year's Day themed photos...</td>\n",
       "      <td>Our 7-month-old baby girl will be the star of ...</td>\n",
       "      <td>1. Soft, pastel-colored canvas or printed fabr...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>01/18</td>\n",
       "      <td>Martin Luther King Jr. Day</td>\n",
       "      <td>Dreaming of a Brighter Future</td>\n",
       "      <td>Concept: This portrait photo will be a tribute...</td>\n",
       "      <td>With a light blue sky background and fluffy wh...</td>\n",
       "      <td>1. Light blue backdrop fabric\\n2. Soft, fluffy...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>02/12</td>\n",
       "      <td>Lunar New Year (approximate date)</td>\n",
       "      <td>Baby Blossoms for Lunar New Year</td>\n",
       "      <td>Concept: For this photoshoot, we will incorpor...</td>\n",
       "      <td>Incorporating the Lunar New Year theme, we wil...</td>\n",
       "      <td>1. Silk cherry blossoms\\n2. Small red lanterns...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>02/14</td>\n",
       "      <td>Valentine's Day</td>\n",
       "      <td>Cupid's Littlest Love</td>\n",
       "      <td>Concept: For this Valentine's Day themed photo...</td>\n",
       "      <td>Transform your 7-month-old baby girl into a li...</td>\n",
       "      <td>1. Soft, pastel pink backdrop\\n2. Tiny white h...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5</th>\n",
       "      <td>03/01</td>\n",
       "      <td>Independence Movement Day</td>\n",
       "      <td>Little Lady Liberty</td>\n",
       "      <td>Concept: For this Independence Movement Day th...</td>\n",
       "      <td>Our little Lady Liberty photoshoot will captur...</td>\n",
       "      <td>1. Soft, green onesie\\n2. Small, lightweight, ...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>6</th>\n",
       "      <td>03/17</td>\n",
       "      <td>St. Patrick's Day</td>\n",
       "      <td>Luck of the Irish Baby</td>\n",
       "      <td>Concept: For this St. Patrick's Day-themed pho...</td>\n",
       "      <td>Surrounded by elements of Irish folklore and c...</td>\n",
       "      <td>1. Soft, pastel green backdrop fabric or hand-...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>7</th>\n",
       "      <td>04/04</td>\n",
       "      <td>Easter (approximate date)</td>\n",
       "      <td>Easter Blossoms</td>\n",
       "      <td>Concept: For this Easter-themed photoshoot, th...</td>\n",
       "      <td>Surrounded by the vibrant colors and symbols o...</td>\n",
       "      <td>1. Camera\\n2. Lens (preferably a portrait lens...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>8</th>\n",
       "      <td>05/05</td>\n",
       "      <td>Children's Day</td>\n",
       "      <td>Blooming Joy on Children's Day</td>\n",
       "      <td>Concept: For this Children's Day-themed photos...</td>\n",
       "      <td>Surrounded by a mini flower garden, our 7-mont...</td>\n",
       "      <td>1. Canvas backdrop in soft shades of lavender ...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>9</th>\n",
       "      <td>05/09</td>\n",
       "      <td>Mother's Day</td>\n",
       "      <td>Mommy's Little Flower</td>\n",
       "      <td>Concept: For this Mother's Day themed photosho...</td>\n",
       "      <td>\"Little Flower Blossom\"\\n\\nTransform the 7-mon...</td>\n",
       "      <td>1. Light pastel-colored fabric for backdrop\\n2...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>10</th>\n",
       "      <td>05/31</td>\n",
       "      <td>Memorial Day</td>\n",
       "      <td>Little Patriot</td>\n",
       "      <td>Concept: For this Memorial Day-themed portrait...</td>\n",
       "      <td>In a serene outdoor setting, a lush green park...</td>\n",
       "      <td>1. Camera\\n2. Lens (preferably a portrait lens...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>11</th>\n",
       "      <td>06/06</td>\n",
       "      <td>Memorial Day</td>\n",
       "      <td>Little Patriot</td>\n",
       "      <td>Concept: For this Memorial Day-themed photosho...</td>\n",
       "      <td>In a serene outdoor setting, such as a park or...</td>\n",
       "      <td>1. Soft, navy blue blanket\\n2. Small American ...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>12</th>\n",
       "      <td>06/19</td>\n",
       "      <td>Juneteenth</td>\n",
       "      <td>Freedom's Blossom</td>\n",
       "      <td>Concept: For this Juneteenth-inspired photosho...</td>\n",
       "      <td>In honor of Juneteenth, we will capture the sp...</td>\n",
       "      <td>1. Solid-colored backdrop in a soft shade of l...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>13</th>\n",
       "      <td>06/20</td>\n",
       "      <td>Father's Day</td>\n",
       "      <td>Daddy's Little Helper</td>\n",
       "      <td>Concept: For this Father's Day-themed photosho...</td>\n",
       "      <td>Capture the heartwarming bond between a father...</td>\n",
       "      <td>1. Toy tools (hammer, wrench, screwdriver)\\n2....</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>14</th>\n",
       "      <td>07/04</td>\n",
       "      <td>Independence Day</td>\n",
       "      <td>Baby Firecracker</td>\n",
       "      <td>Concept: For this Independence Day-themed phot...</td>\n",
       "      <td>Capture the joy and excitement of Independence...</td>\n",
       "      <td>1. White backdrop or American flag patterned b...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>15</th>\n",
       "      <td>08/15</td>\n",
       "      <td>Liberation Day</td>\n",
       "      <td>Little Lady Liberty</td>\n",
       "      <td>Concept: For this Liberation Day-themed photos...</td>\n",
       "      <td>Transform the 7-month-old baby girl into a cha...</td>\n",
       "      <td>1. Solid-colored backdrop in a soft shade of b...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>16</th>\n",
       "      <td>09/06</td>\n",
       "      <td>Labor Day</td>\n",
       "      <td>Little Worker Bee</td>\n",
       "      <td>Concept: For this Labor Day-themed photoshoot,...</td>\n",
       "      <td>Transform the 7-month-old baby girl into a cut...</td>\n",
       "      <td>1. Camera and lenses\\n2. Lighting equipment (s...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>17</th>\n",
       "      <td>09/20</td>\n",
       "      <td>Chuseok (approximate date)</td>\n",
       "      <td>Baby's First Chuseok: A Harvest Moon Dream</td>\n",
       "      <td>Concept: For this Chuseok-inspired portrait ph...</td>\n",
       "      <td>In this enchanting portrait photoshoot, we wil...</td>\n",
       "      <td>1. Crescent moon prop (wood or foam)\\n2. Plush...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>18</th>\n",
       "      <td>10/03</td>\n",
       "      <td>National Foundation Day</td>\n",
       "      <td>Little Builder of Tomorrow</td>\n",
       "      <td>Concept: For National Foundation Day, we will ...</td>\n",
       "      <td>In a cozy, well-lit space, we will capture the...</td>\n",
       "      <td>1. Age-appropriate and safe building blocks ma...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>19</th>\n",
       "      <td>10/09</td>\n",
       "      <td>Hangul Proclamation Day</td>\n",
       "      <td>Little Princess of Hangul</td>\n",
       "      <td>Concept: For this Hangul Proclamation Day phot...</td>\n",
       "      <td>Incorporating the beauty and significance of t...</td>\n",
       "      <td>1. Soft, pastel-colored hanok backdrop\\n2. Tra...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>20</th>\n",
       "      <td>10/11</td>\n",
       "      <td>Columbus Day</td>\n",
       "      <td>Little Explorer</td>\n",
       "      <td>Concept: For this Columbus Day-themed photosho...</td>\n",
       "      <td>Transport the 7-month-old baby girl back in ti...</td>\n",
       "      <td>1. Vintage-looking world map backdrop\\n2. Mini...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>21</th>\n",
       "      <td>10/31</td>\n",
       "      <td>Halloween</td>\n",
       "      <td>Little Pumpkin Princess</td>\n",
       "      <td>Concept: For this Halloween-themed portrait ph...</td>\n",
       "      <td>Surrounded by the warm colors of autumn, our 7...</td>\n",
       "      <td>1. Seamless, warm-toned brown or orange fabric...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>22</th>\n",
       "      <td>11/11</td>\n",
       "      <td>Veterans Day</td>\n",
       "      <td>Little Patriot</td>\n",
       "      <td>Concept: For this Veterans Day-themed photosho...</td>\n",
       "      <td>In honor of Veterans Day, we will capture the ...</td>\n",
       "      <td>1. American flag backdrop\\n2. White dress with...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>23</th>\n",
       "      <td>11/25</td>\n",
       "      <td>Thanksgiving Day</td>\n",
       "      <td>Little Turkey's First Thanksgiving</td>\n",
       "      <td>Concept: For this Thanksgiving-themed portrait...</td>\n",
       "      <td>Capture the essence of \"Little Turkey's First ...</td>\n",
       "      <td>1. Brown onesie or bodysuit with turkey face p...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>24</th>\n",
       "      <td>12/24</td>\n",
       "      <td>Christmas Eve</td>\n",
       "      <td>A Starry Christmas Eve</td>\n",
       "      <td>Concept: For this Christmas Eve-themed photosh...</td>\n",
       "      <td>In this magical Christmas Eve-themed photoshoo...</td>\n",
       "      <td>1. Deep blue backdrop\\n2. Twinkling fairy ligh...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>25</th>\n",
       "      <td>12/25</td>\n",
       "      <td>Christmas Day</td>\n",
       "      <td>Santa's Little Helper</td>\n",
       "      <td>Concept: For this charming Christmas Day portr...</td>\n",
       "      <td>In this heartwarming Christmas Day portrait, w...</td>\n",
       "      <td>1. Rustic wooden wall backdrop\\n2. Twinkling f...</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "     date                            holiday  \\\n",
       "0   01/01                     New Year's Day   \n",
       "1   01/01                     New Year's Day   \n",
       "2   01/18         Martin Luther King Jr. Day   \n",
       "3   02/12  Lunar New Year (approximate date)   \n",
       "4   02/14                    Valentine's Day   \n",
       "5   03/01          Independence Movement Day   \n",
       "6   03/17                  St. Patrick's Day   \n",
       "7   04/04          Easter (approximate date)   \n",
       "8   05/05                     Children's Day   \n",
       "9   05/09                       Mother's Day   \n",
       "10  05/31                       Memorial Day   \n",
       "11  06/06                       Memorial Day   \n",
       "12  06/19                         Juneteenth   \n",
       "13  06/20                       Father's Day   \n",
       "14  07/04                   Independence Day   \n",
       "15  08/15                     Liberation Day   \n",
       "16  09/06                          Labor Day   \n",
       "17  09/20         Chuseok (approximate date)   \n",
       "18  10/03            National Foundation Day   \n",
       "19  10/09            Hangul Proclamation Day   \n",
       "20  10/11                       Columbus Day   \n",
       "21  10/31                          Halloween   \n",
       "22  11/11                       Veterans Day   \n",
       "23  11/25                   Thanksgiving Day   \n",
       "24  12/24                      Christmas Eve   \n",
       "25  12/25                      Christmas Day   \n",
       "\n",
       "                                         title  \\\n",
       "0              Baby's First New Year's Sparkle   \n",
       "1                      A New Year's Baby Dream   \n",
       "2                Dreaming of a Brighter Future   \n",
       "3             Baby Blossoms for Lunar New Year   \n",
       "4                        Cupid's Littlest Love   \n",
       "5                          Little Lady Liberty   \n",
       "6                       Luck of the Irish Baby   \n",
       "7                              Easter Blossoms   \n",
       "8               Blooming Joy on Children's Day   \n",
       "9                        Mommy's Little Flower   \n",
       "10                              Little Patriot   \n",
       "11                              Little Patriot   \n",
       "12                           Freedom's Blossom   \n",
       "13                       Daddy's Little Helper   \n",
       "14                            Baby Firecracker   \n",
       "15                         Little Lady Liberty   \n",
       "16                           Little Worker Bee   \n",
       "17  Baby's First Chuseok: A Harvest Moon Dream   \n",
       "18                  Little Builder of Tomorrow   \n",
       "19                   Little Princess of Hangul   \n",
       "20                             Little Explorer   \n",
       "21                     Little Pumpkin Princess   \n",
       "22                              Little Patriot   \n",
       "23          Little Turkey's First Thanksgiving   \n",
       "24                      A Starry Christmas Eve   \n",
       "25                       Santa's Little Helper   \n",
       "\n",
       "                               photoshoot_description  \\\n",
       "0   Concept: For this New Year's Day themed photos...   \n",
       "1   Concept: For this New Year's Day themed photos...   \n",
       "2   Concept: This portrait photo will be a tribute...   \n",
       "3   Concept: For this photoshoot, we will incorpor...   \n",
       "4   Concept: For this Valentine's Day themed photo...   \n",
       "5   Concept: For this Independence Movement Day th...   \n",
       "6   Concept: For this St. Patrick's Day-themed pho...   \n",
       "7   Concept: For this Easter-themed photoshoot, th...   \n",
       "8   Concept: For this Children's Day-themed photos...   \n",
       "9   Concept: For this Mother's Day themed photosho...   \n",
       "10  Concept: For this Memorial Day-themed portrait...   \n",
       "11  Concept: For this Memorial Day-themed photosho...   \n",
       "12  Concept: For this Juneteenth-inspired photosho...   \n",
       "13  Concept: For this Father's Day-themed photosho...   \n",
       "14  Concept: For this Independence Day-themed phot...   \n",
       "15  Concept: For this Liberation Day-themed photos...   \n",
       "16  Concept: For this Labor Day-themed photoshoot,...   \n",
       "17  Concept: For this Chuseok-inspired portrait ph...   \n",
       "18  Concept: For National Foundation Day, we will ...   \n",
       "19  Concept: For this Hangul Proclamation Day phot...   \n",
       "20  Concept: For this Columbus Day-themed photosho...   \n",
       "21  Concept: For this Halloween-themed portrait ph...   \n",
       "22  Concept: For this Veterans Day-themed photosho...   \n",
       "23  Concept: For this Thanksgiving-themed portrait...   \n",
       "24  Concept: For this Christmas Eve-themed photosh...   \n",
       "25  Concept: For this charming Christmas Day portr...   \n",
       "\n",
       "                                    midjourney_prompt  \\\n",
       "0   Our star of the portrait, a 7-month-old baby g...   \n",
       "1   Our 7-month-old baby girl will be the star of ...   \n",
       "2   With a light blue sky background and fluffy wh...   \n",
       "3   Incorporating the Lunar New Year theme, we wil...   \n",
       "4   Transform your 7-month-old baby girl into a li...   \n",
       "5   Our little Lady Liberty photoshoot will captur...   \n",
       "6   Surrounded by elements of Irish folklore and c...   \n",
       "7   Surrounded by the vibrant colors and symbols o...   \n",
       "8   Surrounded by a mini flower garden, our 7-mont...   \n",
       "9   \"Little Flower Blossom\"\\n\\nTransform the 7-mon...   \n",
       "10  In a serene outdoor setting, a lush green park...   \n",
       "11  In a serene outdoor setting, such as a park or...   \n",
       "12  In honor of Juneteenth, we will capture the sp...   \n",
       "13  Capture the heartwarming bond between a father...   \n",
       "14  Capture the joy and excitement of Independence...   \n",
       "15  Transform the 7-month-old baby girl into a cha...   \n",
       "16  Transform the 7-month-old baby girl into a cut...   \n",
       "17  In this enchanting portrait photoshoot, we wil...   \n",
       "18  In a cozy, well-lit space, we will capture the...   \n",
       "19  Incorporating the beauty and significance of t...   \n",
       "20  Transport the 7-month-old baby girl back in ti...   \n",
       "21  Surrounded by the warm colors of autumn, our 7...   \n",
       "22  In honor of Veterans Day, we will capture the ...   \n",
       "23  Capture the essence of \"Little Turkey's First ...   \n",
       "24  In this magical Christmas Eve-themed photoshoo...   \n",
       "25  In this heartwarming Christmas Day portrait, w...   \n",
       "\n",
       "                                        shopping_list  \n",
       "0   1. White or light gray backdrop\\n2. Cute, whit...  \n",
       "1   1. Soft, pastel-colored canvas or printed fabr...  \n",
       "2   1. Light blue backdrop fabric\\n2. Soft, fluffy...  \n",
       "3   1. Silk cherry blossoms\\n2. Small red lanterns...  \n",
       "4   1. Soft, pastel pink backdrop\\n2. Tiny white h...  \n",
       "5   1. Soft, green onesie\\n2. Small, lightweight, ...  \n",
       "6   1. Soft, pastel green backdrop fabric or hand-...  \n",
       "7   1. Camera\\n2. Lens (preferably a portrait lens...  \n",
       "8   1. Canvas backdrop in soft shades of lavender ...  \n",
       "9   1. Light pastel-colored fabric for backdrop\\n2...  \n",
       "10  1. Camera\\n2. Lens (preferably a portrait lens...  \n",
       "11  1. Soft, navy blue blanket\\n2. Small American ...  \n",
       "12  1. Solid-colored backdrop in a soft shade of l...  \n",
       "13  1. Toy tools (hammer, wrench, screwdriver)\\n2....  \n",
       "14  1. White backdrop or American flag patterned b...  \n",
       "15  1. Solid-colored backdrop in a soft shade of b...  \n",
       "16  1. Camera and lenses\\n2. Lighting equipment (s...  \n",
       "17  1. Crescent moon prop (wood or foam)\\n2. Plush...  \n",
       "18  1. Age-appropriate and safe building blocks ma...  \n",
       "19  1. Soft, pastel-colored hanok backdrop\\n2. Tra...  \n",
       "20  1. Vintage-looking world map backdrop\\n2. Mini...  \n",
       "21  1. Seamless, warm-toned brown or orange fabric...  \n",
       "22  1. American flag backdrop\\n2. White dress with...  \n",
       "23  1. Brown onesie or bodysuit with turkey face p...  \n",
       "24  1. Deep blue backdrop\\n2. Twinkling fairy ligh...  \n",
       "25  1. Rustic wooden wall backdrop\\n2. Twinkling f...  "
      ]
     },
     "execution_count": 21,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "df = (\n",
    "    pd.DataFrame({\n",
    "        'date': [x[0] for x in holidays_data],\n",
    "        'holiday': [x[1] for x in holidays_data],\n",
    "        'photoshoot_description': photoshoot_concepts,\n",
    "        'midjourney_prompt': prompts,\n",
    "        'shopping_list': shopping_lists,\n",
    "    })\n",
    "    .assign(\n",
    "        title=lambda d: d.photoshoot_description.str.split('\"', expand=True)[1],\n",
    "        photoshoot_description=lambda d: d.photoshoot_description.str.split('\\n\\n', n=1, expand=True)[1]\n",
    "    )\n",
    "    [['date', 'holiday', 'title', 'photoshoot_description', 'midjourney_prompt', 'shopping_list']]\n",
    ")\n",
    "\n",
    "# df.sort_values('date').to_clipboard(sep='\\t', index=False)\n",
    "# df.to_excel()\n",
    "df"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3 (ipykernel)",
   "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.16"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}