cocktail_recommendation_system_with_openai.ipynb

{
 "cells": [
  {
   "cell_type": "markdown",
   "metadata": {
    "id": "8R75EZnBX_1V"
   },
   "source": [
    "## Step 1: Libraries Installation\n",
    "Below are brief explanations of the tools and libraries utilised within the implementation code:\n",
    "* **datasets**: This library is part of the Hugging Face ecosystem. By installing 'datasets', we gain access to a number of pre-processed and ready-to-use datasets, which are essential for training and fine-tuning machine learning models or benchmarking their performance.\n",
    "\n",
    "* **pandas**: A data science library that provides robust data structures and methods for data manipulation, processing and analysis.\n",
    "\n",
    "* **openai**: This is the official Python client library for accessing OpenAI's suite of AI models and tools, including GPT and embedding models.italicised text\n",
    "\n",
    "* **pymongo**: PyMongo is a Python toolkit for MongoDB. It enables interactions with a MongoDB database."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {
    "colab": {
     "base_uri": "https://localhost:8080/"
    },
    "id": "4SzSGSA_YCHJ",
    "outputId": "0ef94cd2-1e7e-4a7b-8c52-f45653671780"
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Requirement already satisfied: datasets in c:\\users\\nandi\\anaconda3\\lib\\site-packages (2.19.1)\n",
      "Requirement already satisfied: pandas in c:\\users\\nandi\\anaconda3\\lib\\site-packages (1.4.2)\n",
      "Requirement already satisfied: openai in c:\\users\\nandi\\anaconda3\\lib\\site-packages (1.30.3)\n",
      "Requirement already satisfied: pymongo in c:\\users\\nandi\\anaconda3\\lib\\site-packages (4.7.2)\n",
      "Requirement already satisfied: python-dotenv in c:\\users\\nandi\\anaconda3\\lib\\site-packages (1.0.1)\n",
      "Requirement already satisfied: packaging in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from datasets) (21.3)\n",
      "Requirement already satisfied: pyyaml>=5.1 in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from datasets) (6.0)\n",
      "Requirement already satisfied: fsspec[http]<=2024.3.1,>=2023.1.0 in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from datasets) (2024.3.1)\n",
      "Requirement already satisfied: huggingface-hub>=0.21.2 in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from datasets) (0.23.1)\n",
      "Requirement already satisfied: filelock in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from datasets) (3.6.0)\n",
      "Requirement already satisfied: pyarrow-hotfix in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from datasets) (0.6)\n",
      "Requirement already satisfied: aiohttp in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from datasets) (3.8.1)\n",
      "Requirement already satisfied: requests>=2.19.0 in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from datasets) (2.27.1)\n",
      "Requirement already satisfied: numpy>=1.17 in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from datasets) (1.26.4)\n",
      "Requirement already satisfied: multiprocess in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from datasets) (0.70.16)\n",
      "Requirement already satisfied: pyarrow>=12.0.0 in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from datasets) (16.1.0)\n",
      "Requirement already satisfied: xxhash in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from datasets) (3.4.1)\n",
      "Requirement already satisfied: dill<0.3.9,>=0.3.0 in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from datasets) (0.3.8)\n",
      "Requirement already satisfied: tqdm>=4.62.1 in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from datasets) (4.64.0)\n",
      "Requirement already satisfied: pytz>=2020.1 in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from pandas) (2021.3)\n",
      "Requirement already satisfied: python-dateutil>=2.8.1 in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from pandas) (2.8.2)\n",
      "Requirement already satisfied: sniffio in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from openai) (1.2.0)\n",
      "Requirement already satisfied: typing-extensions<5,>=4.7 in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from openai) (4.10.0)\n",
      "Requirement already satisfied: httpx<1,>=0.23.0 in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from openai) (0.27.0)\n",
      "Requirement already satisfied: anyio<5,>=3.5.0 in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from openai) (3.5.0)\n",
      "Requirement already satisfied: pydantic<3,>=1.9.0 in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from openai) (2.7.1)\n",
      "Requirement already satisfied: distro<2,>=1.7.0 in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from openai) (1.9.0)\n",
      "Requirement already satisfied: dnspython<3.0.0,>=1.16.0 in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from pymongo) (2.6.1)\n",
      "Requirement already satisfied: idna>=2.8 in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from anyio<5,>=3.5.0->openai) (3.3)\n",
      "Requirement already satisfied: multidict<7.0,>=4.5 in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from aiohttp->datasets) (5.1.0)\n",
      "Requirement already satisfied: charset-normalizer<3.0,>=2.0 in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from aiohttp->datasets) (2.0.4)\n",
      "Requirement already satisfied: aiosignal>=1.1.2 in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from aiohttp->datasets) (1.2.0)\n",
      "Requirement already satisfied: attrs>=17.3.0 in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from aiohttp->datasets) (21.4.0)\n",
      "Requirement already satisfied: frozenlist>=1.1.1 in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from aiohttp->datasets) (1.2.0)\n",
      "Requirement already satisfied: async-timeout<5.0,>=4.0.0a3 in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from aiohttp->datasets) (4.0.1)\n",
      "Requirement already satisfied: yarl<2.0,>=1.0 in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from aiohttp->datasets) (1.6.3)\n",
      "Requirement already satisfied: httpcore==1.* in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from httpx<1,>=0.23.0->openai) (1.0.5)\n",
      "Requirement already satisfied: certifi in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from httpx<1,>=0.23.0->openai) (2021.10.8)\n",
      "Requirement already satisfied: h11<0.15,>=0.13 in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from httpcore==1.*->httpx<1,>=0.23.0->openai) (0.14.0)\n",
      "Requirement already satisfied: pyparsing!=3.0.5,>=2.0.2 in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from packaging->datasets) (3.0.4)\n",
      "Requirement already satisfied: pydantic-core==2.18.2 in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from pydantic<3,>=1.9.0->openai) (2.18.2)\n",
      "Requirement already satisfied: annotated-types>=0.4.0 in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from pydantic<3,>=1.9.0->openai) (0.7.0)\n",
      "Requirement already satisfied: six>=1.5 in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from python-dateutil>=2.8.1->pandas) (1.16.0)\n",
      "Requirement already satisfied: urllib3<1.27,>=1.21.1 in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from requests>=2.19.0->datasets) (1.26.9)\n",
      "Requirement already satisfied: colorama in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from tqdm>=4.62.1->datasets) (0.4.4)\n"
     ]
    }
   ],
   "source": [
    "!pip install datasets pandas openai pymongo python-dotenv"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "id": "IdOImjzAYR1L"
   },
   "source": [
    "## Step 2: Data Loading\n",
    "\n",
    "Load the dataset titled [\"brianarbuckle/cocktail_recipes\"](https://huggingface.co/datasets/brianarbuckle/cocktail_recipes). This dataset is a collection of cocktails-related details that include attributes such as the title, ingredients, directions, misc, source and ner."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 91,
   "metadata": {
    "colab": {
     "base_uri": "https://localhost:8080/",
     "height": 513,
     "referenced_widgets": [
      "179bf0b6430543f4b061eeca9580beb7",
      "98da8cd4fb9f4e12828a013e80c3d25b",
      "d09102d2ef544c27be5c25857a1a1cec",
      "83b0b11fa7fa4181883f5c2e9a371f39",
      "d076907f2ad34dddab38ad064e8675a0",
      "3de0500b40894dc7b8f8f133b1eba694",
      "f31b771fa3af483da4fdbf50ec5dd754",
      "00d5b332421f49218a30d2711b400e59",
      "3bc583ba49994c32a51e986357d0f8b4",
      "5b75b2599262447381e838982c69b67f",
      "ca97885d6aae4341a88cf189a1e11df2",
      "14af6f209384423cbb9a72185bb53fa9",
      "793cce78a6834df0be46d6b21ddee6b0",
      "2eeb4f5f78ac4cddafe20e643f134393",
      "2988a806d18e44a39bcb8fabc9cc0864",
      "6c1aeae23e7a437a938970470053fcd9",
      "ddb120fa2e7b4fed81201c9e88e9499d",
      "ed6673f2a3424db785af9978150cdf8e",
      "dedc9c3fb09946bbae2568b52577b391",
      "132142b6f8e74aecb96752301ae3393f",
      "bf0b076013a142e99a25bcc0e17bfd30",
      "5bfa3e2f93b94085b5d2324300c4bbc7",
      "9015081a5ca2450baff5e5fed9e066ee",
      "7ceb3acbd1bc454ca8885aa4fb26b4b8",
      "4f8183d901e3493a86a84c227674c722",
      "52065470ba4042dc9c717b38410ce787",
      "3460d2d487224d13ad25e456d1053422",
      "1471b97b61204f11b4a4d59c9aa91101",
      "aa9a238f10144400a0872c7b15304d22",
      "5516848731f0452fab65e0ade545217c",
      "1b72f31c54624c0591980c5067572a71",
      "f5cda5e56f3f40c5a5ecc47bca536df5",
      "d72fd9f9f0a541cf94a93bcbac3c4bad"
     ]
    },
    "id": "UGlakJPIYGOl",
    "outputId": "2750cab6-cd43-4b9d-c078-a46cdc4d4936"
   },
   "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>title</th>\n",
       "      <th>ingredients</th>\n",
       "      <th>directions</th>\n",
       "      <th>misc</th>\n",
       "      <th>source</th>\n",
       "      <th>ner</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>151 Swizzle</td>\n",
       "      <td>[1.5 oz. 151-Proof Demerara Rum [Lemon Hart or...</td>\n",
       "      <td>[]</td>\n",
       "      <td>[]</td>\n",
       "      <td>Beachbum Berry Remixed</td>\n",
       "      <td>[pernod, rum]</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>20th Century</td>\n",
       "      <td>[The 21st Century, 2 oz. Siete Leguas Blanco T...</td>\n",
       "      <td>[shake on ice and strain into coupe , The Best...</td>\n",
       "      <td>[]</td>\n",
       "      <td>Jim Meehan</td>\n",
       "      <td>[cocchi americano, pernod, tequila]</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>20th Century</td>\n",
       "      <td>[1.5 oz. Plymouth Gin, 3\\/4 oz. Mari Brizard W...</td>\n",
       "      <td>[shake on ice and strain]</td>\n",
       "      <td>[]</td>\n",
       "      <td>PDT</td>\n",
       "      <td>[lillet, gin]</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>Abbey Cocktail</td>\n",
       "      <td>[]</td>\n",
       "      <td>[Shake liquid ingredients with ice., Strain in...</td>\n",
       "      <td>[Suggested glassware is Cocktail Glass]</td>\n",
       "      <td>The Ultimate Bar Book</td>\n",
       "      <td>[]</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>Absinthe Drip</td>\n",
       "      <td>[1 1/2 ounces Pernod (or other absinthe substi...</td>\n",
       "      <td>[Pour Pernod into a pousse-caf or sour glass....</td>\n",
       "      <td>[The Absinthe Drip was made famous by Toulouse...</td>\n",
       "      <td>The Ultimate Bar Book</td>\n",
       "      <td>[pernod, absinthe]</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "            title                                        ingredients  \\\n",
       "0     151 Swizzle  [1.5 oz. 151-Proof Demerara Rum [Lemon Hart or...   \n",
       "1    20th Century  [The 21st Century, 2 oz. Siete Leguas Blanco T...   \n",
       "2    20th Century  [1.5 oz. Plymouth Gin, 3\\/4 oz. Mari Brizard W...   \n",
       "3  Abbey Cocktail                                                 []   \n",
       "4   Absinthe Drip  [1 1/2 ounces Pernod (or other absinthe substi...   \n",
       "\n",
       "                                          directions  \\\n",
       "0                                                 []   \n",
       "1  [shake on ice and strain into coupe , The Best...   \n",
       "2                          [shake on ice and strain]   \n",
       "3  [Shake liquid ingredients with ice., Strain in...   \n",
       "4  [Pour Pernod into a pousse-caf or sour glass....   \n",
       "\n",
       "                                                misc                  source  \\\n",
       "0                                                 []  Beachbum Berry Remixed   \n",
       "1                                                 []              Jim Meehan   \n",
       "2                                                 []                     PDT   \n",
       "3            [Suggested glassware is Cocktail Glass]   The Ultimate Bar Book   \n",
       "4  [The Absinthe Drip was made famous by Toulouse...   The Ultimate Bar Book   \n",
       "\n",
       "                                   ner  \n",
       "0                        [pernod, rum]  \n",
       "1  [cocchi americano, pernod, tequila]  \n",
       "2                        [lillet, gin]  \n",
       "3                                   []  \n",
       "4                   [pernod, absinthe]  "
      ]
     },
     "execution_count": 91,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# 1. Load Dataset\n",
    "from datasets import load_dataset\n",
    "import pandas as pd\n",
    "\n",
    "# https://huggingface.co/datasets/brianarbuckle/cocktail_recipes\n",
    "dataset = load_dataset(\"brianarbuckle/cocktail_recipes\")\n",
    "\n",
    "# Convert the dataset to a pandas dataframe\n",
    "dataset_df = pd.DataFrame(dataset['train'])\n",
    "\n",
    "dataset_df.head(5)"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "id": "CDW2-kw1Y7dU"
   },
   "source": [
    "## Step 3: Data Cleaning and Preparation\n",
    "\n",
    "The next step cleans the data and prepares it for the next stage, which creates a new embedding data point using the new OpenAI embedding model."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 92,
   "metadata": {
    "colab": {
     "base_uri": "https://localhost:8080/",
     "height": 597
    },
    "id": "_Ay-cNw85hms",
    "outputId": "f4fc64fe-d329-4855-b467-413d4686f6eb"
   },
   "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>title</th>\n",
       "      <th>ingredients</th>\n",
       "      <th>directions</th>\n",
       "      <th>misc</th>\n",
       "      <th>source</th>\n",
       "      <th>ner</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>151 Swizzle</td>\n",
       "      <td>[1.5 oz. 151-Proof Demerara Rum [Lemon Hart or...</td>\n",
       "      <td></td>\n",
       "      <td>[]</td>\n",
       "      <td>Beachbum Berry Remixed</td>\n",
       "      <td>pernod, rum</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>20th Century</td>\n",
       "      <td>[The 21st Century, 2 oz. Siete Leguas Blanco T...</td>\n",
       "      <td>shake on ice and strain into coupe , The Best ...</td>\n",
       "      <td>[]</td>\n",
       "      <td>Jim Meehan</td>\n",
       "      <td>cocchi americano, pernod, tequila</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>20th Century</td>\n",
       "      <td>[1.5 oz. Plymouth Gin, 3\\/4 oz. Mari Brizard W...</td>\n",
       "      <td>shake on ice and strain</td>\n",
       "      <td>[]</td>\n",
       "      <td>PDT</td>\n",
       "      <td>lillet, gin</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>Abbey Cocktail</td>\n",
       "      <td>[]</td>\n",
       "      <td>Shake liquid ingredients with ice., Strain int...</td>\n",
       "      <td>[Suggested glassware is Cocktail Glass]</td>\n",
       "      <td>The Ultimate Bar Book</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>Absinthe Drip</td>\n",
       "      <td>[1 1/2 ounces Pernod (or other absinthe substi...</td>\n",
       "      <td>Pour Pernod into a pousse-caf or sour glass.,...</td>\n",
       "      <td>[The Absinthe Drip was made famous by Toulouse...</td>\n",
       "      <td>The Ultimate Bar Book</td>\n",
       "      <td>pernod, absinthe</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>...</th>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>870</th>\n",
       "      <td>Yellow Bird</td>\n",
       "      <td>[ A Caribbean favorite., 1 ounce dark rum, 1 o...</td>\n",
       "      <td>Shake liquid ingredients with ice., Strain int...</td>\n",
       "      <td>[Suggested glassware is Cocktail Glass]</td>\n",
       "      <td>The Ultimate Bar Book</td>\n",
       "      <td>galliano, triple sec, rum, cointreau</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>871</th>\n",
       "      <td>Yellow Fever</td>\n",
       "      <td>[1 1/2 ounces vodka, 1/2 ounce Galliano, 1/2 o...</td>\n",
       "      <td>Shake ingredients with ice., Strain into a chi...</td>\n",
       "      <td>[Suggested glassware is Cocktail Glass]</td>\n",
       "      <td>The Ultimate Bar Book</td>\n",
       "      <td>vodka, galliano</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>872</th>\n",
       "      <td>Yellow Parrot Cocktail</td>\n",
       "      <td>[3/4 ounce yellow Chartreuse, 3/4 ounce Pernod...</td>\n",
       "      <td>Shake ingredients with ice., Strain into a chi...</td>\n",
       "      <td>[Suggested glassware is Cocktail Glass]</td>\n",
       "      <td>The Ultimate Bar Book</td>\n",
       "      <td>brandy, pernod, chartreuse</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>873</th>\n",
       "      <td>[The Spirit of the] Algonquin</td>\n",
       "      <td>[ 2oz. 90 Proof Rye, .75oz. Lemon Juice, .75oz...</td>\n",
       "      <td>shake on ice and strain</td>\n",
       "      <td>[Suggested glassware is Cocktail Glass]</td>\n",
       "      <td>PDT</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>874</th>\n",
       "      <td>the-ontocord-mimosa</td>\n",
       "      <td>[champagne, orange juice-ish, optional but pre...</td>\n",
       "      <td>Nearly fill a champagne flute with champagne.,...</td>\n",
       "      <td>[We all know that a Mimosa is an attempt to ma...</td>\n",
       "      <td>Mix it in 30</td>\n",
       "      <td>champagne, orange juice, gin, orange slice</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>875 rows × 6 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "                             title  \\\n",
       "0                      151 Swizzle   \n",
       "1                     20th Century   \n",
       "2                     20th Century   \n",
       "3                   Abbey Cocktail   \n",
       "4                    Absinthe Drip   \n",
       "..                             ...   \n",
       "870                    Yellow Bird   \n",
       "871                   Yellow Fever   \n",
       "872         Yellow Parrot Cocktail   \n",
       "873  [The Spirit of the] Algonquin   \n",
       "874            the-ontocord-mimosa   \n",
       "\n",
       "                                           ingredients  \\\n",
       "0    [1.5 oz. 151-Proof Demerara Rum [Lemon Hart or...   \n",
       "1    [The 21st Century, 2 oz. Siete Leguas Blanco T...   \n",
       "2    [1.5 oz. Plymouth Gin, 3\\/4 oz. Mari Brizard W...   \n",
       "3                                                   []   \n",
       "4    [1 1/2 ounces Pernod (or other absinthe substi...   \n",
       "..                                                 ...   \n",
       "870  [ A Caribbean favorite., 1 ounce dark rum, 1 o...   \n",
       "871  [1 1/2 ounces vodka, 1/2 ounce Galliano, 1/2 o...   \n",
       "872  [3/4 ounce yellow Chartreuse, 3/4 ounce Pernod...   \n",
       "873  [ 2oz. 90 Proof Rye, .75oz. Lemon Juice, .75oz...   \n",
       "874  [champagne, orange juice-ish, optional but pre...   \n",
       "\n",
       "                                            directions  \\\n",
       "0                                                        \n",
       "1    shake on ice and strain into coupe , The Best ...   \n",
       "2                              shake on ice and strain   \n",
       "3    Shake liquid ingredients with ice., Strain int...   \n",
       "4    Pour Pernod into a pousse-caf or sour glass.,...   \n",
       "..                                                 ...   \n",
       "870  Shake liquid ingredients with ice., Strain int...   \n",
       "871  Shake ingredients with ice., Strain into a chi...   \n",
       "872  Shake ingredients with ice., Strain into a chi...   \n",
       "873                            shake on ice and strain   \n",
       "874  Nearly fill a champagne flute with champagne.,...   \n",
       "\n",
       "                                                  misc  \\\n",
       "0                                                   []   \n",
       "1                                                   []   \n",
       "2                                                   []   \n",
       "3              [Suggested glassware is Cocktail Glass]   \n",
       "4    [The Absinthe Drip was made famous by Toulouse...   \n",
       "..                                                 ...   \n",
       "870            [Suggested glassware is Cocktail Glass]   \n",
       "871            [Suggested glassware is Cocktail Glass]   \n",
       "872            [Suggested glassware is Cocktail Glass]   \n",
       "873            [Suggested glassware is Cocktail Glass]   \n",
       "874  [We all know that a Mimosa is an attempt to ma...   \n",
       "\n",
       "                     source                                         ner  \n",
       "0    Beachbum Berry Remixed                                 pernod, rum  \n",
       "1                Jim Meehan           cocchi americano, pernod, tequila  \n",
       "2                       PDT                                 lillet, gin  \n",
       "3     The Ultimate Bar Book                                         NaN  \n",
       "4     The Ultimate Bar Book                            pernod, absinthe  \n",
       "..                      ...                                         ...  \n",
       "870   The Ultimate Bar Book        galliano, triple sec, rum, cointreau  \n",
       "871   The Ultimate Bar Book                             vodka, galliano  \n",
       "872   The Ultimate Bar Book                  brandy, pernod, chartreuse  \n",
       "873                     PDT                                         NaN  \n",
       "874            Mix it in 30  champagne, orange juice, gin, orange slice  \n",
       "\n",
       "[875 rows x 6 columns]"
      ]
     },
     "execution_count": 92,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "import numpy as np\n",
    "\n",
    "def process_ingredients(col):\n",
    "    if isinstance(col, list):\n",
    "        if not col:\n",
    "            return np.nan\n",
    "        else:\n",
    "            return \", \".join(col)\n",
    "    else:\n",
    "        return col\n",
    "\n",
    "dataset_df['ner'] = dataset_df['ner'].apply(process_ingredients)\n",
    "dataset_df['directions'] = dataset_df['directions'].apply(process_ingredients)\n",
    "dataset_df\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 93,
   "metadata": {
    "colab": {
     "base_uri": "https://localhost:8080/"
    },
    "id": "Nq98H950YqAS",
    "outputId": "58e5486f-00ee-4b96-e5e1-ed4c8703e635"
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Columns: Index(['title', 'ingredients', 'directions', 'misc', 'source', 'ner'], dtype='object')\n",
      "\n",
      "Number of rows and columns: (875, 6)\n",
      "\n",
      "Basic Statistics for numerical data:\n",
      "             title                                        ingredients  \\\n",
      "count          875                                                875   \n",
      "unique         646                                                858   \n",
      "top     Ward Eight  [After Dinner Cocktail, 5 cl Cognac, 2 cl Crme...   \n",
      "freq             9                                                  6   \n",
      "\n",
      "                     directions misc                 source  ner  \n",
      "count                       792  875                    875  753  \n",
      "unique                      455   63                     83  252  \n",
      "top     shake on ice and strain   []  The Ultimate Bar Book  gin  \n",
      "freq                         66  376                    438   81  \n",
      "\n",
      "Number of missing values in each column:\n",
      "title            0\n",
      "ingredients      0\n",
      "directions      83\n",
      "misc             0\n",
      "source           0\n",
      "ner            122\n",
      "dtype: int64\n"
     ]
    }
   ],
   "source": [
    "print(\"Columns:\", dataset_df.columns)\n",
    "print(\"\\nNumber of rows and columns:\", dataset_df.shape)\n",
    "print(\"\\nBasic Statistics for numerical data:\")\n",
    "print(dataset_df.describe())\n",
    "print(\"\\nNumber of missing values in each column:\")\n",
    "print(dataset_df.isnull().sum())"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 94,
   "metadata": {
    "colab": {
     "base_uri": "https://localhost:8080/",
     "height": 684
    },
    "id": "iPjllgXMvC1r",
    "outputId": "cb5f5788-25bd-4119-f92d-246f95ffbcb3"
   },
   "outputs": [
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "C:\\Users\\nandi\\AppData\\Local\\Temp\\ipykernel_53728\\2422078972.py:2: SettingWithCopyWarning: \n",
      "A value is trying to be set on a copy of a slice from a DataFrame\n",
      "\n",
      "See the caveats in the documentation: https://pandas.pydata.org/pandas-docs/stable/user_guide/indexing.html#returning-a-view-versus-a-copy\n",
      "  dataset_df.rename(columns={'ner': 'base'}, inplace=True)\n"
     ]
    },
    {
     "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>title</th>\n",
       "      <th>ingredients</th>\n",
       "      <th>directions</th>\n",
       "      <th>misc</th>\n",
       "      <th>source</th>\n",
       "      <th>base</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>151 Swizzle</td>\n",
       "      <td>[1.5 oz. 151-Proof Demerara Rum [Lemon Hart or...</td>\n",
       "      <td></td>\n",
       "      <td>[]</td>\n",
       "      <td>Beachbum Berry Remixed</td>\n",
       "      <td>pernod, rum</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>20th Century</td>\n",
       "      <td>[The 21st Century, 2 oz. Siete Leguas Blanco T...</td>\n",
       "      <td>shake on ice and strain into coupe , The Best ...</td>\n",
       "      <td>[]</td>\n",
       "      <td>Jim Meehan</td>\n",
       "      <td>cocchi americano, pernod, tequila</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>20th Century</td>\n",
       "      <td>[1.5 oz. Plymouth Gin, 3\\/4 oz. Mari Brizard W...</td>\n",
       "      <td>shake on ice and strain</td>\n",
       "      <td>[]</td>\n",
       "      <td>PDT</td>\n",
       "      <td>lillet, gin</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>Absinthe Drip</td>\n",
       "      <td>[1 1/2 ounces Pernod (or other absinthe substi...</td>\n",
       "      <td>Pour Pernod into a pousse-caf or sour glass.,...</td>\n",
       "      <td>[The Absinthe Drip was made famous by Toulouse...</td>\n",
       "      <td>The Ultimate Bar Book</td>\n",
       "      <td>pernod, absinthe</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5</th>\n",
       "      <td>Acapulco</td>\n",
       "      <td>[1 ounce gold tequila, 1 ounce gold rum, 2 oun...</td>\n",
       "      <td>Shake ingredients with ice., Strain into an ic...</td>\n",
       "      <td>[Suggested glassware is Highball Glass]</td>\n",
       "      <td>The Ultimate Bar Book</td>\n",
       "      <td>tequila, rum</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>...</th>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>869</th>\n",
       "      <td>Yellow Bird</td>\n",
       "      <td>[3 cl White Rum, 1.5 cl Galliano, 1.5 cl Tripl...</td>\n",
       "      <td>Shake liquid ingredients with ice., Strain int...</td>\n",
       "      <td>[Suggested glassware is Cocktail Glass]</td>\n",
       "      <td>IBA</td>\n",
       "      <td>galliano, triple sec, rum</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>870</th>\n",
       "      <td>Yellow Bird</td>\n",
       "      <td>[ A Caribbean favorite., 1 ounce dark rum, 1 o...</td>\n",
       "      <td>Shake liquid ingredients with ice., Strain int...</td>\n",
       "      <td>[Suggested glassware is Cocktail Glass]</td>\n",
       "      <td>The Ultimate Bar Book</td>\n",
       "      <td>galliano, triple sec, rum, cointreau</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>871</th>\n",
       "      <td>Yellow Fever</td>\n",
       "      <td>[1 1/2 ounces vodka, 1/2 ounce Galliano, 1/2 o...</td>\n",
       "      <td>Shake ingredients with ice., Strain into a chi...</td>\n",
       "      <td>[Suggested glassware is Cocktail Glass]</td>\n",
       "      <td>The Ultimate Bar Book</td>\n",
       "      <td>vodka, galliano</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>872</th>\n",
       "      <td>Yellow Parrot Cocktail</td>\n",
       "      <td>[3/4 ounce yellow Chartreuse, 3/4 ounce Pernod...</td>\n",
       "      <td>Shake ingredients with ice., Strain into a chi...</td>\n",
       "      <td>[Suggested glassware is Cocktail Glass]</td>\n",
       "      <td>The Ultimate Bar Book</td>\n",
       "      <td>brandy, pernod, chartreuse</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>874</th>\n",
       "      <td>the-ontocord-mimosa</td>\n",
       "      <td>[champagne, orange juice-ish, optional but pre...</td>\n",
       "      <td>Nearly fill a champagne flute with champagne.,...</td>\n",
       "      <td>[We all know that a Mimosa is an attempt to ma...</td>\n",
       "      <td>Mix it in 30</td>\n",
       "      <td>champagne, orange juice, gin, orange slice</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>753 rows × 6 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "                      title  \\\n",
       "0               151 Swizzle   \n",
       "1              20th Century   \n",
       "2              20th Century   \n",
       "4             Absinthe Drip   \n",
       "5                  Acapulco   \n",
       "..                      ...   \n",
       "869             Yellow Bird   \n",
       "870             Yellow Bird   \n",
       "871            Yellow Fever   \n",
       "872  Yellow Parrot Cocktail   \n",
       "874     the-ontocord-mimosa   \n",
       "\n",
       "                                           ingredients  \\\n",
       "0    [1.5 oz. 151-Proof Demerara Rum [Lemon Hart or...   \n",
       "1    [The 21st Century, 2 oz. Siete Leguas Blanco T...   \n",
       "2    [1.5 oz. Plymouth Gin, 3\\/4 oz. Mari Brizard W...   \n",
       "4    [1 1/2 ounces Pernod (or other absinthe substi...   \n",
       "5    [1 ounce gold tequila, 1 ounce gold rum, 2 oun...   \n",
       "..                                                 ...   \n",
       "869  [3 cl White Rum, 1.5 cl Galliano, 1.5 cl Tripl...   \n",
       "870  [ A Caribbean favorite., 1 ounce dark rum, 1 o...   \n",
       "871  [1 1/2 ounces vodka, 1/2 ounce Galliano, 1/2 o...   \n",
       "872  [3/4 ounce yellow Chartreuse, 3/4 ounce Pernod...   \n",
       "874  [champagne, orange juice-ish, optional but pre...   \n",
       "\n",
       "                                            directions  \\\n",
       "0                                                        \n",
       "1    shake on ice and strain into coupe , The Best ...   \n",
       "2                              shake on ice and strain   \n",
       "4    Pour Pernod into a pousse-caf or sour glass.,...   \n",
       "5    Shake ingredients with ice., Strain into an ic...   \n",
       "..                                                 ...   \n",
       "869  Shake liquid ingredients with ice., Strain int...   \n",
       "870  Shake liquid ingredients with ice., Strain int...   \n",
       "871  Shake ingredients with ice., Strain into a chi...   \n",
       "872  Shake ingredients with ice., Strain into a chi...   \n",
       "874  Nearly fill a champagne flute with champagne.,...   \n",
       "\n",
       "                                                  misc  \\\n",
       "0                                                   []   \n",
       "1                                                   []   \n",
       "2                                                   []   \n",
       "4    [The Absinthe Drip was made famous by Toulouse...   \n",
       "5              [Suggested glassware is Highball Glass]   \n",
       "..                                                 ...   \n",
       "869            [Suggested glassware is Cocktail Glass]   \n",
       "870            [Suggested glassware is Cocktail Glass]   \n",
       "871            [Suggested glassware is Cocktail Glass]   \n",
       "872            [Suggested glassware is Cocktail Glass]   \n",
       "874  [We all know that a Mimosa is an attempt to ma...   \n",
       "\n",
       "                     source                                        base  \n",
       "0    Beachbum Berry Remixed                                 pernod, rum  \n",
       "1                Jim Meehan           cocchi americano, pernod, tequila  \n",
       "2                       PDT                                 lillet, gin  \n",
       "4     The Ultimate Bar Book                            pernod, absinthe  \n",
       "5     The Ultimate Bar Book                                tequila, rum  \n",
       "..                      ...                                         ...  \n",
       "869                     IBA                   galliano, triple sec, rum  \n",
       "870   The Ultimate Bar Book        galliano, triple sec, rum, cointreau  \n",
       "871   The Ultimate Bar Book                             vodka, galliano  \n",
       "872   The Ultimate Bar Book                  brandy, pernod, chartreuse  \n",
       "874            Mix it in 30  champagne, orange juice, gin, orange slice  \n",
       "\n",
       "[753 rows x 6 columns]"
      ]
     },
     "execution_count": 94,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "dataset_df = dataset_df.dropna(subset=['ner'])\n",
    "dataset_df.rename(columns={'ner': 'base'}, inplace=True)\n",
    "dataset_df"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "id": "doNiIQSq8HSd"
   },
   "source": [
    "## Step 4: Create embeddings with OpenAI"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 95,
   "metadata": {
    "colab": {
     "base_uri": "https://localhost:8080/",
     "height": 397
    },
    "id": "izq_fWlUbda-",
    "outputId": "26eab297-b2eb-40cd-af89-7a9f82c51839"
   },
   "outputs": [
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "C:\\Users\\nandi\\AppData\\Local\\Temp\\ipykernel_53728\\245036467.py:28: SettingWithCopyWarning: \n",
      "A value is trying to be set on a copy of a slice from a DataFrame.\n",
      "Try using .loc[row_indexer,col_indexer] = value instead\n",
      "\n",
      "See the caveats in the documentation: https://pandas.pydata.org/pandas-docs/stable/user_guide/indexing.html#returning-a-view-versus-a-copy\n",
      "  dataset_df[\"base_embedding_optimised\"] = dataset_df['base'].apply(get_embedding)\n"
     ]
    },
    {
     "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>title</th>\n",
       "      <th>ingredients</th>\n",
       "      <th>directions</th>\n",
       "      <th>misc</th>\n",
       "      <th>source</th>\n",
       "      <th>base</th>\n",
       "      <th>base_embedding_optimised</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>151 Swizzle</td>\n",
       "      <td>[1.5 oz. 151-Proof Demerara Rum [Lemon Hart or...</td>\n",
       "      <td></td>\n",
       "      <td>[]</td>\n",
       "      <td>Beachbum Berry Remixed</td>\n",
       "      <td>pernod, rum</td>\n",
       "      <td>[-0.004870898090302944, 0.03630739822983742, 0...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>20th Century</td>\n",
       "      <td>[The 21st Century, 2 oz. Siete Leguas Blanco T...</td>\n",
       "      <td>shake on ice and strain into coupe , The Best ...</td>\n",
       "      <td>[]</td>\n",
       "      <td>Jim Meehan</td>\n",
       "      <td>cocchi americano, pernod, tequila</td>\n",
       "      <td>[-0.012937657535076141, -0.006157346069812775,...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>20th Century</td>\n",
       "      <td>[1.5 oz. Plymouth Gin, 3\\/4 oz. Mari Brizard W...</td>\n",
       "      <td>shake on ice and strain</td>\n",
       "      <td>[]</td>\n",
       "      <td>PDT</td>\n",
       "      <td>lillet, gin</td>\n",
       "      <td>[-0.041257601231336594, -0.008328181691467762,...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>Absinthe Drip</td>\n",
       "      <td>[1 1/2 ounces Pernod (or other absinthe substi...</td>\n",
       "      <td>Pour Pernod into a pousse-caf or sour glass.,...</td>\n",
       "      <td>[The Absinthe Drip was made famous by Toulouse...</td>\n",
       "      <td>The Ultimate Bar Book</td>\n",
       "      <td>pernod, absinthe</td>\n",
       "      <td>[-0.04415753856301308, 0.030096998438239098, 0...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5</th>\n",
       "      <td>Acapulco</td>\n",
       "      <td>[1 ounce gold tequila, 1 ounce gold rum, 2 oun...</td>\n",
       "      <td>Shake ingredients with ice., Strain into an ic...</td>\n",
       "      <td>[Suggested glassware is Highball Glass]</td>\n",
       "      <td>The Ultimate Bar Book</td>\n",
       "      <td>tequila, rum</td>\n",
       "      <td>[-0.016873909160494804, 0.004132091999053955, ...</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "           title                                        ingredients  \\\n",
       "0    151 Swizzle  [1.5 oz. 151-Proof Demerara Rum [Lemon Hart or...   \n",
       "1   20th Century  [The 21st Century, 2 oz. Siete Leguas Blanco T...   \n",
       "2   20th Century  [1.5 oz. Plymouth Gin, 3\\/4 oz. Mari Brizard W...   \n",
       "4  Absinthe Drip  [1 1/2 ounces Pernod (or other absinthe substi...   \n",
       "5       Acapulco  [1 ounce gold tequila, 1 ounce gold rum, 2 oun...   \n",
       "\n",
       "                                          directions  \\\n",
       "0                                                      \n",
       "1  shake on ice and strain into coupe , The Best ...   \n",
       "2                            shake on ice and strain   \n",
       "4  Pour Pernod into a pousse-caf or sour glass.,...   \n",
       "5  Shake ingredients with ice., Strain into an ic...   \n",
       "\n",
       "                                                misc                  source  \\\n",
       "0                                                 []  Beachbum Berry Remixed   \n",
       "1                                                 []              Jim Meehan   \n",
       "2                                                 []                     PDT   \n",
       "4  [The Absinthe Drip was made famous by Toulouse...   The Ultimate Bar Book   \n",
       "5            [Suggested glassware is Highball Glass]   The Ultimate Bar Book   \n",
       "\n",
       "                                base  \\\n",
       "0                        pernod, rum   \n",
       "1  cocchi americano, pernod, tequila   \n",
       "2                        lillet, gin   \n",
       "4                   pernod, absinthe   \n",
       "5                       tequila, rum   \n",
       "\n",
       "                            base_embedding_optimised  \n",
       "0  [-0.004870898090302944, 0.03630739822983742, 0...  \n",
       "1  [-0.012937657535076141, -0.006157346069812775,...  \n",
       "2  [-0.041257601231336594, -0.008328181691467762,...  \n",
       "4  [-0.04415753856301308, 0.030096998438239098, 0...  \n",
       "5  [-0.016873909160494804, 0.004132091999053955, ...  "
      ]
     },
     "execution_count": 95,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "import os\n",
    "import openai\n",
    "from dotenv import load_dotenv\n",
    "\n",
    "# Load environment variables from .env file\n",
    "load_dotenv()\n",
    "\n",
    "# Access the API key from the environment variable\n",
    "openai.api_key = os.getenv(\"OPENAI_API_KEY\")\n",
    "\n",
    "EMBEDDING_MODEL = \"text-embedding-3-small\"\n",
    "\n",
    "def get_embedding(text):\n",
    "    \"\"\"Generate an embedding for the given text using OpenAI's API.\"\"\"\n",
    "\n",
    "    # Check for valid input\n",
    "    if not text or not isinstance(text, str):\n",
    "        return None\n",
    "\n",
    "    try:\n",
    "        # Call OpenAI API to get the embedding\n",
    "        embedding = openai.embeddings.create(input=text, model=EMBEDDING_MODEL).data[0].embedding\n",
    "        return embedding\n",
    "    except Exception as e:\n",
    "        print(f\"Error in get_embedding: {e}\")\n",
    "        return None\n",
    "\n",
    "dataset_df[\"base_embedding_optimised\"] = dataset_df['base'].apply(get_embedding)\n",
    "dataset_df.head()"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "id": "mxPrmoE4CPr9"
   },
   "source": [
    "## Step 5: Vector Database Setup and Data Ingestion"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 96,
   "metadata": {
    "colab": {
     "base_uri": "https://localhost:8080/"
    },
    "id": "L5IVb8rKb-yH",
    "outputId": "306a94cc-fdf5-494f-8772-326de9245383"
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Connection to MongoDB successful\n"
     ]
    }
   ],
   "source": [
    "import pymongo\n",
    "# from google.colab import userdata\n",
    "\n",
    "def get_mongo_client(mongo_uri):\n",
    "  \"\"\"Establish connection to the MongoDB.\"\"\"\n",
    "  try:\n",
    "    client = pymongo.MongoClient(mongo_uri)\n",
    "    print(\"Connection to MongoDB successful\")\n",
    "    return client\n",
    "  except pymongo.errors.ConnectionFailure as e:\n",
    "    print(f\"Connection failed: {e}\")\n",
    "    return None\n",
    "\n",
    "mongo_uri = \"mongodb+srv://lazzyCoder:dO1qORxPZ4YOeJkZ@cocktaildb.eq81tdu.mongodb.net/?retryWrites=true&w=majority&appName=cocktailDB\"\n",
    "if not mongo_uri:\n",
    "  print(\"MONGO_URI not set in environment variables\")\n",
    "\n",
    "mongo_client = get_mongo_client(mongo_uri)\n",
    "\n",
    "# Ingest data into MongoDB\n",
    "db = mongo_client['cocktails']\n",
    "collection = db['cocktailsDB']"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 97,
   "metadata": {
    "colab": {
     "base_uri": "https://localhost:8080/"
    },
    "id": "ggd_2ybXCcFq",
    "outputId": "7df79f7a-3c35-4a1d-9d0d-02ba0782d09a"
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "DeleteResult({'n': 753, 'electionId': ObjectId('7fffffff0000000000000188'), 'opTime': {'ts': Timestamp(1716865625, 753), 't': 392}, 'ok': 1.0, '$clusterTime': {'clusterTime': Timestamp(1716865625, 755), 'signature': {'hash': b'\\xdeQ\\xbbid\\x19r\\x0b\\xc9\\x80\\x1f\\xd5\\xfa\\xa3,\\xfd\\xab\\xa0p\\x15', 'keyId': 7327356172825001986}}, 'operationTime': Timestamp(1716865625, 753)}, acknowledged=True)"
      ]
     },
     "execution_count": 97,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# Delete any existing records in the collection\n",
    "collection.delete_many({})"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 98,
   "metadata": {
    "colab": {
     "base_uri": "https://localhost:8080/"
    },
    "id": "cKQ8RHglCien",
    "outputId": "f6795d2c-bacf-462b-8ace-2688c033a863"
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Data ingestion into MongoDB completed\n"
     ]
    }
   ],
   "source": [
    "documents = dataset_df.to_dict('records')\n",
    "collection.insert_many(documents)\n",
    "\n",
    "print(\"Data ingestion into MongoDB completed\")"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "id": "dtMAJzTCCzrS"
   },
   "source": [
    "## Step 6: Create a Vector Search Index"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "id": "FSiurji1Gs6C"
   },
   "source": [
    "At this point make sure that your vector index is created via MongoDB Atlas."
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "id": "Jypp9Q-RGmgk"
   },
   "source": [
    "## Step 7: Perform Vector Search on User Queries"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 99,
   "metadata": {
    "id": "He_tbgVkCtHu"
   },
   "outputs": [],
   "source": [
    "\n",
    "def vector_search(user_query, collection):\n",
    "    \"\"\"\n",
    "    Perform a vector search in the MongoDB collection based on the user query.\n",
    "\n",
    "    Args:\n",
    "    user_query (str): The user's query string.\n",
    "    collection (MongoCollection): The MongoDB collection to search.\n",
    "\n",
    "    Returns:\n",
    "    list: A list of matching documents.\n",
    "    \"\"\"\n",
    "\n",
    "    # Generate embedding for the user query\n",
    "    query_embedding = get_embedding(user_query)\n",
    "\n",
    "    if query_embedding is None:\n",
    "        return \"Invalid query or embedding generation failed.\"\n",
    "\n",
    "    # Define the vector search pipeline\n",
    "    pipeline = [\n",
    "        {\n",
    "            \"$vectorSearch\": {\n",
    "                \"index\": \"vector_index\",\n",
    "                \"queryVector\": query_embedding,\n",
    "                \"path\": \"base_embedding_optimised\",\n",
    "                \"numCandidates\": 150,  # Number of items matches to consider\n",
    "                \"limit\": 5  # Return top 5 matches\n",
    "            }\n",
    "        },\n",
    "        {\n",
    "            \"$project\": {\n",
    "                \"_id\": 0,  # Exclude the _id field\n",
    "                \"base\": 1,  # Include the base field\n",
    "                \"title\": 1,  # Include the title field\n",
    "                \"ingredients\": 1, # Include the ingredients field\n",
    "                \"directions\": 1,  # Include the directions field\n",
    "                \"score\": {\n",
    "                    \"$meta\": \"vectorSearchScore\"  # Include the search score\n",
    "                }\n",
    "            }\n",
    "        }\n",
    "    ]\n",
    "\n",
    "    # Execute the search\n",
    "    results = collection.aggregate(pipeline)\n",
    "    return list(results)"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "id": "lrhKpBtqHz-U"
   },
   "source": [
    "## Step 8: Handling User Query and Result"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 100,
   "metadata": {
    "id": "dGWOrDDCH3gc"
   },
   "outputs": [],
   "source": [
    "def handle_user_query(query, collection):\n",
    "\n",
    "  get_knowledge = vector_search(query, collection)\n",
    "\n",
    "  search_result = ''\n",
    "  for result in get_knowledge:\n",
    "      search_result += f\"Title: {result.get('title', 'N/A')}, Base Drink: {result.get('base', 'N/A')}\\n\"\n",
    "\n",
    "  completion = openai.chat.completions.create(\n",
    "      model=\"gpt-3.5-turbo\",\n",
    "      messages=[\n",
    "          {\"role\": \"system\", \"content\": \"You are a cocktail recommendation system.\"},\n",
    "          {\"role\": \"user\", \"content\": \"Answer this user query: \" + query + \" with the following context: \" + search_result}\n",
    "      ]\n",
    "  )\n",
    "\n",
    "  return (completion.choices[0].message.content), search_result"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 101,
   "metadata": {
    "colab": {
     "base_uri": "https://localhost:8080/"
    },
    "id": "wc3S9-g9IpOL",
    "outputId": "ebeadb28-1ab5-4df3-e23e-2b2dd528b59a"
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Response: Here are some cocktail recommendations you can make with vodka:\n",
      "\n",
      "1. Raspberry Mint Martini: A refreshing and fruity martini made with vodka, raspberry liqueur, fresh raspberries, mint leaves, and a splash of lime juice.\n",
      "\n",
      "2. Purple Passion: A vibrant cocktail featuring vodka, blue Curacao, cranberry juice, and a splash of lemon-lime soda for a sweet and tangy flavor.\n",
      "\n",
      "3. Watermelon: A summery cocktail made with vodka, fresh watermelon juice, lime juice, and a touch of simple syrup for a sweet and refreshing drink.\n",
      "\n",
      "4. Tokyo Mary: A unique twist on the classic Bloody Mary, this cocktail combines vodka with Japanese flavors like soy sauce, wasabi, and Sriracha for a spicy and savory kick.\n",
      "\n",
      "5. Pearl Harbor: A tropical cocktail that blends vodka with melon liqueur and pineapple juice, creating a sweet and fruity drink perfect for sipping on a sunny day. \n",
      "\n",
      "Enjoy crafting these delicious vodka-based cocktails!\n",
      "Source Information: \n",
      "Title: Raspberry Mint Martini, Base Drink: vodka\n",
      "Title: Purple Passion, Base Drink: vodka\n",
      "Title: Watermelon, Base Drink: vodka\n",
      "Title: Tokyo Mary, Base Drink: vodka\n",
      "Title: Pearl Harbor, Base Drink: vodka\n",
      "\n"
     ]
    }
   ],
   "source": [
    "query = \"What are the best cocktails I can make with vodka?\"\n",
    "response, source_information = handle_user_query(query, collection)\n",
    "\n",
    "print(f\"Response: {response}\")\n",
    "print(f\"Source Information: \\n{source_information}\")\n"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "id": "6DII_G0JJLdf"
   },
   "source": [
    "## Frontend Part"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 102,
   "metadata": {
    "colab": {
     "base_uri": "https://localhost:8080/"
    },
    "id": "jh0ML34TI2oA",
    "outputId": "ebc70e5a-20ba-477f-c89e-c706fcb79f9a"
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "\n",
      "up to date, audited 23 packages in 486ms\n",
      "\n",
      "3 packages are looking for funding\n",
      "  run `npm fund` for details\n",
      "\n",
      "2 moderate severity vulnerabilities\n",
      "\n",
      "To address all issues (including breaking changes), run:\n",
      "  npm audit fix --force\n",
      "\n",
      "Run `npm audit` for details.\n",
      "Requirement already satisfied: streamlit in c:\\users\\nandi\\anaconda3\\lib\\site-packages (1.35.0)\n",
      "Requirement already satisfied: packaging<25,>=16.8 in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from streamlit) (21.3)\n",
      "Requirement already satisfied: pandas<3,>=1.3.0 in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from streamlit) (1.4.2)\n",
      "Requirement already satisfied: altair<6,>=4.0 in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from streamlit) (5.3.0)\n",
      "Requirement already satisfied: pillow<11,>=7.1.0 in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from streamlit) (9.0.1)\n",
      "Requirement already satisfied: blinker<2,>=1.0.0 in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from streamlit) (1.8.2)\n",
      "Requirement already satisfied: cachetools<6,>=4.0 in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from streamlit) (4.2.2)\n",
      "Requirement already satisfied: typing-extensions<5,>=4.3.0 in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from streamlit) (4.10.0)\n",
      "Requirement already satisfied: requests<3,>=2.27 in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from streamlit) (2.27.1)\n",
      "Requirement already satisfied: toml<2,>=0.10.1 in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from streamlit) (0.10.2)\n",
      "Requirement already satisfied: rich<14,>=10.14.0 in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from streamlit) (13.7.1)\n",
      "Requirement already satisfied: tornado<7,>=6.0.3 in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from streamlit) (6.1)\n",
      "Requirement already satisfied: tenacity<9,>=8.1.0 in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from streamlit) (8.3.0)\n",
      "Requirement already satisfied: pydeck<1,>=0.8.0b4 in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from streamlit) (0.9.1)\n",
      "Requirement already satisfied: watchdog>=2.1.5 in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from streamlit) (2.1.6)\n",
      "Requirement already satisfied: numpy<2,>=1.19.3 in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from streamlit) (1.26.4)\n",
      "Requirement already satisfied: pyarrow>=7.0 in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from streamlit) (16.1.0)\n",
      "Requirement already satisfied: protobuf<5,>=3.20 in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from streamlit) (4.25.3)\n",
      "Requirement already satisfied: gitpython!=3.1.19,<4,>=3.0.7 in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from streamlit) (3.1.43)\n",
      "Requirement already satisfied: click<9,>=7.0 in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from streamlit) (8.0.4)\n",
      "Requirement already satisfied: jinja2 in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from altair<6,>=4.0->streamlit) (2.11.3)\n",
      "Requirement already satisfied: jsonschema>=3.0 in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from altair<6,>=4.0->streamlit) (4.4.0)\n",
      "Requirement already satisfied: toolz in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from altair<6,>=4.0->streamlit) (0.11.2)\n",
      "Requirement already satisfied: colorama in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from click<9,>=7.0->streamlit) (0.4.4)\n",
      "Requirement already satisfied: gitdb<5,>=4.0.1 in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from gitpython!=3.1.19,<4,>=3.0.7->streamlit) (4.0.11)\n",
      "Requirement already satisfied: smmap<6,>=3.0.1 in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from gitdb<5,>=4.0.1->gitpython!=3.1.19,<4,>=3.0.7->streamlit) (5.0.1)\n",
      "Requirement already satisfied: attrs>=17.4.0 in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from jsonschema>=3.0->altair<6,>=4.0->streamlit) (21.4.0)\n",
      "Requirement already satisfied: pyrsistent!=0.17.0,!=0.17.1,!=0.17.2,>=0.14.0 in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from jsonschema>=3.0->altair<6,>=4.0->streamlit) (0.18.0)\n",
      "Requirement already satisfied: pyparsing!=3.0.5,>=2.0.2 in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from packaging<25,>=16.8->streamlit) (3.0.4)\n",
      "Requirement already satisfied: pytz>=2020.1 in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from pandas<3,>=1.3.0->streamlit) (2021.3)\n",
      "Requirement already satisfied: python-dateutil>=2.8.1 in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from pandas<3,>=1.3.0->streamlit) (2.8.2)\n",
      "Requirement already satisfied: MarkupSafe>=0.23 in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from jinja2->altair<6,>=4.0->streamlit) (2.0.1)\n",
      "Requirement already satisfied: six>=1.5 in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from python-dateutil>=2.8.1->pandas<3,>=1.3.0->streamlit) (1.16.0)\n",
      "Requirement already satisfied: charset-normalizer~=2.0.0 in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from requests<3,>=2.27->streamlit) (2.0.4)\n",
      "Requirement already satisfied: idna<4,>=2.5 in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from requests<3,>=2.27->streamlit) (3.3)\n",
      "Requirement already satisfied: certifi>=2017.4.17 in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from requests<3,>=2.27->streamlit) (2021.10.8)\n",
      "Requirement already satisfied: urllib3<1.27,>=1.21.1 in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from requests<3,>=2.27->streamlit) (1.26.9)\n",
      "Requirement already satisfied: pygments<3.0.0,>=2.13.0 in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from rich<14,>=10.14.0->streamlit) (2.18.0)\n",
      "Requirement already satisfied: markdown-it-py>=2.2.0 in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from rich<14,>=10.14.0->streamlit) (3.0.0)\n",
      "Requirement already satisfied: mdurl~=0.1 in c:\\users\\nandi\\anaconda3\\lib\\site-packages (from markdown-it-py>=2.2.0->rich<14,>=10.14.0->streamlit) (0.1.2)\n"
     ]
    }
   ],
   "source": [
    "!npm install localtunnel\n",
    "!pip install streamlit"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 103,
   "metadata": {
    "colab": {
     "base_uri": "https://localhost:8080/"
    },
    "id": "yWVYpEIVJOIk",
    "outputId": "f463ea45-8a22-4db0-98f3-f95e45058554"
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Overwriting app.py\n"
     ]
    }
   ],
   "source": [
    "%%writefile app.py\n",
    "import openai\n",
    "import os\n",
    "# from google.colab import userdata\n",
    "import pymongo\n",
    "import streamlit as st\n",
    "from dotenv import load_dotenv\n",
    "\n",
    "# Load environment variables from .env file\n",
    "load_dotenv()\n",
    "\n",
    "# Access the API key from the environment variable\n",
    "openai.api_key = os.getenv(\"OPENAI_API_KEY\")\n",
    "EMBEDDING_MODEL = \"text-embedding-3-small\"\n",
    "\n",
    "\n",
    "def get_mongo_client(mongo_uri):\n",
    "  \"\"\"Establish connection to the MongoDB.\"\"\"\n",
    "  try:\n",
    "    client = pymongo.MongoClient(mongo_uri)\n",
    "    print(\"Connection to MongoDB successful\")\n",
    "    return client\n",
    "  except pymongo.errors.ConnectionFailure as e:\n",
    "    print(f\"Connection failed: {e}\")\n",
    "    return None\n",
    "\n",
    "mongo_uri = \"mongodb+srv://lazzyCoder:dO1qORxPZ4YOeJkZ@cocktaildb.eq81tdu.mongodb.net/?retryWrites=true&w=majority&appName=cocktailDB\"\n",
    "if not mongo_uri:\n",
    "  print(\"MONGO_URI not set in environment variables\")\n",
    "\n",
    "mongo_client = get_mongo_client(mongo_uri)\n",
    "\n",
    "# Ingest data into MongoDB\n",
    "db = mongo_client['cocktails']\n",
    "collection = db['cocktailsDB']\n",
    "\n",
    "# Delete any existing records in the collection\n",
    "def get_embedding(text):\n",
    "    \"\"\"Generate an embedding for the given text using OpenAI's API.\"\"\"\n",
    "\n",
    "    # Check for valid input\n",
    "    if not text or not isinstance(text, str):\n",
    "        return None\n",
    "\n",
    "    try:\n",
    "        # Call OpenAI API to get the embedding\n",
    "        embedding = openai.embeddings.create(input=text, model=EMBEDDING_MODEL).data[0].embedding\n",
    "        return embedding\n",
    "    except Exception as e:\n",
    "        print(f\"Error in get_embedding: {e}\")\n",
    "        return None\n",
    "\n",
    "def vector_search(user_query, collection):\n",
    "    \"\"\"\n",
    "    Perform a vector search in the MongoDB collection based on the user query.\n",
    "\n",
    "    Args:\n",
    "    user_query (str): The user's query string.\n",
    "    collection (MongoCollection): The MongoDB collection to search.\n",
    "\n",
    "    Returns:\n",
    "    list: A list of matching documents.\n",
    "    \"\"\"\n",
    "\n",
    "    # Generate embedding for the user query\n",
    "    query_embedding = get_embedding(user_query)\n",
    "\n",
    "    if query_embedding is None:\n",
    "        return \"Invalid query or embedding generation failed.\"\n",
    "\n",
    "    # Define the vector search pipeline\n",
    "    pipeline = [\n",
    "        {\n",
    "            \"$vectorSearch\": {\n",
    "                \"index\": \"vector_index\",\n",
    "                \"queryVector\": query_embedding,\n",
    "                \"path\": \"base_embedding_optimised\",\n",
    "                \"numCandidates\": 150,  # Number of candidate matches to consider\n",
    "                \"limit\": 5  # Return top 5 matches\n",
    "            }\n",
    "        },\n",
    "        {\n",
    "            \"$project\": {\n",
    "                \"_id\": 0,  # Exclude the _id field\n",
    "                \"base\": 1,  # Include the base field\n",
    "                \"title\": 1,  # Include the title field\n",
    "                \"ingredients\": 1, # Include the ingredients field\n",
    "                \"directions\": 1,  # Include the directions field\n",
    "                \"score\": {\n",
    "                    \"$meta\": \"vectorSearchScore\"  # Include the search score\n",
    "                }\n",
    "            }\n",
    "        }\n",
    "    ]\n",
    "\n",
    "    # Execute the search\n",
    "    results = collection.aggregate(pipeline)\n",
    "    return list(results)\n",
    "\n",
    "\n",
    "def handle_user_query(query, collection):\n",
    "    get_knowledge = vector_search(query, collection)\n",
    "\n",
    "    search_result = []\n",
    "    for result in get_knowledge:\n",
    "        search_result.append({\n",
    "            'title': result.get('title', 'N/A'),\n",
    "            'base': result.get('base', 'N/A'),\n",
    "            'ingredients': result.get('ingredients', 'N/A'),\n",
    "            'directions': result.get('directions', 'N/A')\n",
    "        })\n",
    "\n",
    "    completion = openai.chat.completions.create(\n",
    "        model=\"gpt-3.5-turbo\",\n",
    "        messages=[\n",
    "            {\"role\": \"system\", \"content\": \"You are a cocktail recommendation system.\"},\n",
    "            {\"role\": \"user\", \"content\": \"Answer this user query: \" + query + \" with the following context: \" + str(search_result)}\n",
    "        ]\n",
    "    )\n",
    "\n",
    "    return completion.choices[0].message.content, search_result\n",
    "\n",
    "st.set_page_config(page_title=\"Cocktail Recommendation\", layout=\"wide\")\n",
    "st.markdown(\n",
    "    \"\"\"\n",
    "    <div style='width: 65%; margin: 0 auto;'>\n",
    "        <h1 style='text-align: center; color: #1F618D;border-radius: 40px; background: linear-gradient(-225deg, #69EACB 0%, #EACCF8 48%, #6654F1 100%);'>🍹 Cocktail Recommendation 🍸</h1>\n",
    "    </div>\n",
    "    \"\"\",\n",
    "    unsafe_allow_html=True\n",
    ")\n",
    "query = st.text_input(\"Ask your cocktail query here\", placeholder=\"E.g., Recommend a refreshing summer cocktail\")\n",
    "\n",
    "if query != \"\":\n",
    "    response, search_results = handle_user_query(query, collection)\n",
    "    # st.markdown(f\"<p style='font-size: 18px; font-weight: bold; color: #333333;'>Response:</p> <p style='font-size: 16px; color: #666666;'>{response}</p>\", unsafe_allow_html=True)\n",
    "\n",
    "    # Display search results as cards\n",
    "    cols = st.columns(3)  # Change the number based on your preference\n",
    "    for i, result in enumerate(search_results):\n",
    "        with cols[i % 3]:\n",
    "            with st.container():\n",
    "                st.markdown(f\"\"\"\n",
    "                <div style=\"border: 3px solid #CCCCCC; border-radius: 5px; padding: 20px; background-color: #17202A;\">\n",
    "                    <h3 style=\"color: #EC7063; text-align: center;\">{result['title']}</h3>\n",
    "                    <p style=\"font-weight: bold; color: #A3E4D7;\">Base Drink: <span style=\"color: #FBFCFC;\">{result['base']}</span></p>\n",
    "                    <p style=\"font-weight: bold; color: #A3E4D7;\">Ingredients:</p>\n",
    "                    <ul style=\"list-style-type: square; padding-left: 20px; color: #FBFCFC;\">\n",
    "                        {''.join(['<li>' + item + '</li>' for item in result['ingredients']])}\n",
    "                    </ul>\n",
    "                    <p style=\"font-weight: bold; color: #A3E4D7;\">Directions:</p>\n",
    "                    <p style=\"color: #FBFCFC;\">{result['directions']}</p>\n",
    "                </div>\n",
    "                \"\"\", unsafe_allow_html=True)\n",
    "\n",
    "\n",
    "                # Add some spacing between cards\n",
    "                st.markdown(\"<br>\", unsafe_allow_html=True)\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {
    "id": "9lag7iEHKe1W"
   },
   "outputs": [],
   "source": []
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "colab": {
   "provenance": []
  },
  "kernelspec": {
   "display_name": "Python 3",
   "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.10.13"
  },
  "widgets": {
   "application/vnd.jupyter.widget-state+json": {
    "00d5b332421f49218a30d2711b400e59": {
     "model_module": "@jupyter-widgets/base",
     "model_module_version": "1.2.0",
     "model_name": "LayoutModel",
     "state": {
      "_model_module": "@jupyter-widgets/base",
      "_model_module_version": "1.2.0",
      "_model_name": "LayoutModel",
      "_view_count": null,
      "_view_module": "@jupyter-widgets/base",
      "_view_module_version": "1.2.0",
      "_view_name": "LayoutView",
      "align_content": null,
      "align_items": null,
      "align_self": null,
      "border": null,
      "bottom": null,
      "display": null,
      "flex": null,
      "flex_flow": null,
      "grid_area": null,
      "grid_auto_columns": null,
      "grid_auto_flow": null,
      "grid_auto_rows": null,
      "grid_column": null,
      "grid_gap": null,
      "grid_row": null,
      "grid_template_areas": null,
      "grid_template_columns": null,
      "grid_template_rows": null,
      "height": null,
      "justify_content": null,
      "justify_items": null,
      "left": null,
      "margin": null,
      "max_height": null,
      "max_width": null,
      "min_height": null,
      "min_width": null,
      "object_fit": null,
      "object_position": null,
      "order": null,
      "overflow": null,
      "overflow_x": null,
      "overflow_y": null,
      "padding": null,
      "right": null,
      "top": null,
      "visibility": null,
      "width": null
     }
    },
    "132142b6f8e74aecb96752301ae3393f": {
     "model_module": "@jupyter-widgets/controls",
     "model_module_version": "1.5.0",
     "model_name": "ProgressStyleModel",
     "state": {
      "_model_module": "@jupyter-widgets/controls",
      "_model_module_version": "1.5.0",
      "_model_name": "ProgressStyleModel",
      "_view_count": null,
      "_view_module": "@jupyter-widgets/base",
      "_view_module_version": "1.2.0",
      "_view_name": "StyleView",
      "bar_color": null,
      "description_width": ""
     }
    },
    "1471b97b61204f11b4a4d59c9aa91101": {
     "model_module": "@jupyter-widgets/base",
     "model_module_version": "1.2.0",
     "model_name": "LayoutModel",
     "state": {
      "_model_module": "@jupyter-widgets/base",
      "_model_module_version": "1.2.0",
      "_model_name": "LayoutModel",
      "_view_count": null,
      "_view_module": "@jupyter-widgets/base",
      "_view_module_version": "1.2.0",
      "_view_name": "LayoutView",
      "align_content": null,
      "align_items": null,
      "align_self": null,
      "border": null,
      "bottom": null,
      "display": null,
      "flex": null,
      "flex_flow": null,
      "grid_area": null,
      "grid_auto_columns": null,
      "grid_auto_flow": null,
      "grid_auto_rows": null,
      "grid_column": null,
      "grid_gap": null,
      "grid_row": null,
      "grid_template_areas": null,
      "grid_template_columns": null,
      "grid_template_rows": null,
      "height": null,
      "justify_content": null,
      "justify_items": null,
      "left": null,
      "margin": null,
      "max_height": null,
      "max_width": null,
      "min_height": null,
      "min_width": null,
      "object_fit": null,
      "object_position": null,
      "order": null,
      "overflow": null,
      "overflow_x": null,
      "overflow_y": null,
      "padding": null,
      "right": null,
      "top": null,
      "visibility": null,
      "width": null
     }
    },
    "14af6f209384423cbb9a72185bb53fa9": {
     "model_module": "@jupyter-widgets/controls",
     "model_module_version": "1.5.0",
     "model_name": "HBoxModel",
     "state": {
      "_dom_classes": [],
      "_model_module": "@jupyter-widgets/controls",
      "_model_module_version": "1.5.0",
      "_model_name": "HBoxModel",
      "_view_count": null,
      "_view_module": "@jupyter-widgets/controls",
      "_view_module_version": "1.5.0",
      "_view_name": "HBoxView",
      "box_style": "",
      "children": [
       "IPY_MODEL_793cce78a6834df0be46d6b21ddee6b0",
       "IPY_MODEL_2eeb4f5f78ac4cddafe20e643f134393",
       "IPY_MODEL_2988a806d18e44a39bcb8fabc9cc0864"
      ],
      "layout": "IPY_MODEL_6c1aeae23e7a437a938970470053fcd9"
     }
    },
    "179bf0b6430543f4b061eeca9580beb7": {
     "model_module": "@jupyter-widgets/controls",
     "model_module_version": "1.5.0",
     "model_name": "HBoxModel",
     "state": {
      "_dom_classes": [],
      "_model_module": "@jupyter-widgets/controls",
      "_model_module_version": "1.5.0",
      "_model_name": "HBoxModel",
      "_view_count": null,
      "_view_module": "@jupyter-widgets/controls",
      "_view_module_version": "1.5.0",
      "_view_name": "HBoxView",
      "box_style": "",
      "children": [
       "IPY_MODEL_98da8cd4fb9f4e12828a013e80c3d25b",
       "IPY_MODEL_d09102d2ef544c27be5c25857a1a1cec",
       "IPY_MODEL_83b0b11fa7fa4181883f5c2e9a371f39"
      ],
      "layout": "IPY_MODEL_d076907f2ad34dddab38ad064e8675a0"
     }
    },
    "1b72f31c54624c0591980c5067572a71": {
     "model_module": "@jupyter-widgets/controls",
     "model_module_version": "1.5.0",
     "model_name": "ProgressStyleModel",
     "state": {
      "_model_module": "@jupyter-widgets/controls",
      "_model_module_version": "1.5.0",
      "_model_name": "ProgressStyleModel",
      "_view_count": null,
      "_view_module": "@jupyter-widgets/base",
      "_view_module_version": "1.2.0",
      "_view_name": "StyleView",
      "bar_color": null,
      "description_width": ""
     }
    },
    "2988a806d18e44a39bcb8fabc9cc0864": {
     "model_module": "@jupyter-widgets/controls",
     "model_module_version": "1.5.0",
     "model_name": "HTMLModel",
     "state": {
      "_dom_classes": [],
      "_model_module": "@jupyter-widgets/controls",
      "_model_module_version": "1.5.0",
      "_model_name": "HTMLModel",
      "_view_count": null,
      "_view_module": "@jupyter-widgets/controls",
      "_view_module_version": "1.5.0",
      "_view_name": "HTMLView",
      "description": "",
      "description_tooltip": null,
      "layout": "IPY_MODEL_bf0b076013a142e99a25bcc0e17bfd30",
      "placeholder": "​",
      "style": "IPY_MODEL_5bfa3e2f93b94085b5d2324300c4bbc7",
      "value": " 96.9k/96.9k [00:00&lt;00:00, 159kB/s]"
     }
    },
    "2eeb4f5f78ac4cddafe20e643f134393": {
     "model_module": "@jupyter-widgets/controls",
     "model_module_version": "1.5.0",
     "model_name": "FloatProgressModel",
     "state": {
      "_dom_classes": [],
      "_model_module": "@jupyter-widgets/controls",
      "_model_module_version": "1.5.0",
      "_model_name": "FloatProgressModel",
      "_view_count": null,
      "_view_module": "@jupyter-widgets/controls",
      "_view_module_version": "1.5.0",
      "_view_name": "ProgressView",
      "bar_style": "success",
      "description": "",
      "description_tooltip": null,
      "layout": "IPY_MODEL_dedc9c3fb09946bbae2568b52577b391",
      "max": 96915,
      "min": 0,
      "orientation": "horizontal",
      "style": "IPY_MODEL_132142b6f8e74aecb96752301ae3393f",
      "value": 96915
     }
    },
    "3460d2d487224d13ad25e456d1053422": {
     "model_module": "@jupyter-widgets/base",
     "model_module_version": "1.2.0",
     "model_name": "LayoutModel",
     "state": {
      "_model_module": "@jupyter-widgets/base",
      "_model_module_version": "1.2.0",
      "_model_name": "LayoutModel",
      "_view_count": null,
      "_view_module": "@jupyter-widgets/base",
      "_view_module_version": "1.2.0",
      "_view_name": "LayoutView",
      "align_content": null,
      "align_items": null,
      "align_self": null,
      "border": null,
      "bottom": null,
      "display": null,
      "flex": null,
      "flex_flow": null,
      "grid_area": null,
      "grid_auto_columns": null,
      "grid_auto_flow": null,
      "grid_auto_rows": null,
      "grid_column": null,
      "grid_gap": null,
      "grid_row": null,
      "grid_template_areas": null,
      "grid_template_columns": null,
      "grid_template_rows": null,
      "height": null,
      "justify_content": null,
      "justify_items": null,
      "left": null,
      "margin": null,
      "max_height": null,
      "max_width": null,
      "min_height": null,
      "min_width": null,
      "object_fit": null,
      "object_position": null,
      "order": null,
      "overflow": null,
      "overflow_x": null,
      "overflow_y": null,
      "padding": null,
      "right": null,
      "top": null,
      "visibility": null,
      "width": null
     }
    },
    "3bc583ba49994c32a51e986357d0f8b4": {
     "model_module": "@jupyter-widgets/controls",
     "model_module_version": "1.5.0",
     "model_name": "ProgressStyleModel",
     "state": {
      "_model_module": "@jupyter-widgets/controls",
      "_model_module_version": "1.5.0",
      "_model_name": "ProgressStyleModel",
      "_view_count": null,
      "_view_module": "@jupyter-widgets/base",
      "_view_module_version": "1.2.0",
      "_view_name": "StyleView",
      "bar_color": null,
      "description_width": ""
     }
    },
    "3de0500b40894dc7b8f8f133b1eba694": {
     "model_module": "@jupyter-widgets/base",
     "model_module_version": "1.2.0",
     "model_name": "LayoutModel",
     "state": {
      "_model_module": "@jupyter-widgets/base",
      "_model_module_version": "1.2.0",
      "_model_name": "LayoutModel",
      "_view_count": null,
      "_view_module": "@jupyter-widgets/base",
      "_view_module_version": "1.2.0",
      "_view_name": "LayoutView",
      "align_content": null,
      "align_items": null,
      "align_self": null,
      "border": null,
      "bottom": null,
      "display": null,
      "flex": null,
      "flex_flow": null,
      "grid_area": null,
      "grid_auto_columns": null,
      "grid_auto_flow": null,
      "grid_auto_rows": null,
      "grid_column": null,
      "grid_gap": null,
      "grid_row": null,
      "grid_template_areas": null,
      "grid_template_columns": null,
      "grid_template_rows": null,
      "height": null,
      "justify_content": null,
      "justify_items": null,
      "left": null,
      "margin": null,
      "max_height": null,
      "max_width": null,
      "min_height": null,
      "min_width": null,
      "object_fit": null,
      "object_position": null,
      "order": null,
      "overflow": null,
      "overflow_x": null,
      "overflow_y": null,
      "padding": null,
      "right": null,
      "top": null,
      "visibility": null,
      "width": null
     }
    },
    "4f8183d901e3493a86a84c227674c722": {
     "model_module": "@jupyter-widgets/controls",
     "model_module_version": "1.5.0",
     "model_name": "FloatProgressModel",
     "state": {
      "_dom_classes": [],
      "_model_module": "@jupyter-widgets/controls",
      "_model_module_version": "1.5.0",
      "_model_name": "FloatProgressModel",
      "_view_count": null,
      "_view_module": "@jupyter-widgets/controls",
      "_view_module_version": "1.5.0",
      "_view_name": "ProgressView",
      "bar_style": "success",
      "description": "",
      "description_tooltip": null,
      "layout": "IPY_MODEL_5516848731f0452fab65e0ade545217c",
      "max": 875,
      "min": 0,
      "orientation": "horizontal",
      "style": "IPY_MODEL_1b72f31c54624c0591980c5067572a71",
      "value": 875
     }
    },
    "52065470ba4042dc9c717b38410ce787": {
     "model_module": "@jupyter-widgets/controls",
     "model_module_version": "1.5.0",
     "model_name": "HTMLModel",
     "state": {
      "_dom_classes": [],
      "_model_module": "@jupyter-widgets/controls",
      "_model_module_version": "1.5.0",
      "_model_name": "HTMLModel",
      "_view_count": null,
      "_view_module": "@jupyter-widgets/controls",
      "_view_module_version": "1.5.0",
      "_view_name": "HTMLView",
      "description": "",
      "description_tooltip": null,
      "layout": "IPY_MODEL_f5cda5e56f3f40c5a5ecc47bca536df5",
      "placeholder": "​",
      "style": "IPY_MODEL_d72fd9f9f0a541cf94a93bcbac3c4bad",
      "value": " 875/875 [00:00&lt;00:00, 15264.89 examples/s]"
     }
    },
    "5516848731f0452fab65e0ade545217c": {
     "model_module": "@jupyter-widgets/base",
     "model_module_version": "1.2.0",
     "model_name": "LayoutModel",
     "state": {
      "_model_module": "@jupyter-widgets/base",
      "_model_module_version": "1.2.0",
      "_model_name": "LayoutModel",
      "_view_count": null,
      "_view_module": "@jupyter-widgets/base",
      "_view_module_version": "1.2.0",
      "_view_name": "LayoutView",
      "align_content": null,
      "align_items": null,
      "align_self": null,
      "border": null,
      "bottom": null,
      "display": null,
      "flex": null,
      "flex_flow": null,
      "grid_area": null,
      "grid_auto_columns": null,
      "grid_auto_flow": null,
      "grid_auto_rows": null,
      "grid_column": null,
      "grid_gap": null,
      "grid_row": null,
      "grid_template_areas": null,
      "grid_template_columns": null,
      "grid_template_rows": null,
      "height": null,
      "justify_content": null,
      "justify_items": null,
      "left": null,
      "margin": null,
      "max_height": null,
      "max_width": null,
      "min_height": null,
      "min_width": null,
      "object_fit": null,
      "object_position": null,
      "order": null,
      "overflow": null,
      "overflow_x": null,
      "overflow_y": null,
      "padding": null,
      "right": null,
      "top": null,
      "visibility": null,
      "width": null
     }
    },
    "5b75b2599262447381e838982c69b67f": {
     "model_module": "@jupyter-widgets/base",
     "model_module_version": "1.2.0",
     "model_name": "LayoutModel",
     "state": {
      "_model_module": "@jupyter-widgets/base",
      "_model_module_version": "1.2.0",
      "_model_name": "LayoutModel",
      "_view_count": null,
      "_view_module": "@jupyter-widgets/base",
      "_view_module_version": "1.2.0",
      "_view_name": "LayoutView",
      "align_content": null,
      "align_items": null,
      "align_self": null,
      "border": null,
      "bottom": null,
      "display": null,
      "flex": null,
      "flex_flow": null,
      "grid_area": null,
      "grid_auto_columns": null,
      "grid_auto_flow": null,
      "grid_auto_rows": null,
      "grid_column": null,
      "grid_gap": null,
      "grid_row": null,
      "grid_template_areas": null,
      "grid_template_columns": null,
      "grid_template_rows": null,
      "height": null,
      "justify_content": null,
      "justify_items": null,
      "left": null,
      "margin": null,
      "max_height": null,
      "max_width": null,
      "min_height": null,
      "min_width": null,
      "object_fit": null,
      "object_position": null,
      "order": null,
      "overflow": null,
      "overflow_x": null,
      "overflow_y": null,
      "padding": null,
      "right": null,
      "top": null,
      "visibility": null,
      "width": null
     }
    },
    "5bfa3e2f93b94085b5d2324300c4bbc7": {
     "model_module": "@jupyter-widgets/controls",
     "model_module_version": "1.5.0",
     "model_name": "DescriptionStyleModel",
     "state": {
      "_model_module": "@jupyter-widgets/controls",
      "_model_module_version": "1.5.0",
      "_model_name": "DescriptionStyleModel",
      "_view_count": null,
      "_view_module": "@jupyter-widgets/base",
      "_view_module_version": "1.2.0",
      "_view_name": "StyleView",
      "description_width": ""
     }
    },
    "6c1aeae23e7a437a938970470053fcd9": {
     "model_module": "@jupyter-widgets/base",
     "model_module_version": "1.2.0",
     "model_name": "LayoutModel",
     "state": {
      "_model_module": "@jupyter-widgets/base",
      "_model_module_version": "1.2.0",
      "_model_name": "LayoutModel",
      "_view_count": null,
      "_view_module": "@jupyter-widgets/base",
      "_view_module_version": "1.2.0",
      "_view_name": "LayoutView",
      "align_content": null,
      "align_items": null,
      "align_self": null,
      "border": null,
      "bottom": null,
      "display": null,
      "flex": null,
      "flex_flow": null,
      "grid_area": null,
      "grid_auto_columns": null,
      "grid_auto_flow": null,
      "grid_auto_rows": null,
      "grid_column": null,
      "grid_gap": null,
      "grid_row": null,
      "grid_template_areas": null,
      "grid_template_columns": null,
      "grid_template_rows": null,
      "height": null,
      "justify_content": null,
      "justify_items": null,
      "left": null,
      "margin": null,
      "max_height": null,
      "max_width": null,
      "min_height": null,
      "min_width": null,
      "object_fit": null,
      "object_position": null,
      "order": null,
      "overflow": null,
      "overflow_x": null,
      "overflow_y": null,
      "padding": null,
      "right": null,
      "top": null,
      "visibility": null,
      "width": null
     }
    },
    "793cce78a6834df0be46d6b21ddee6b0": {
     "model_module": "@jupyter-widgets/controls",
     "model_module_version": "1.5.0",
     "model_name": "HTMLModel",
     "state": {
      "_dom_classes": [],
      "_model_module": "@jupyter-widgets/controls",
      "_model_module_version": "1.5.0",
      "_model_name": "HTMLModel",
      "_view_count": null,
      "_view_module": "@jupyter-widgets/controls",
      "_view_module_version": "1.5.0",
      "_view_name": "HTMLView",
      "description": "",
      "description_tooltip": null,
      "layout": "IPY_MODEL_ddb120fa2e7b4fed81201c9e88e9499d",
      "placeholder": "​",
      "style": "IPY_MODEL_ed6673f2a3424db785af9978150cdf8e",
      "value": "Downloading data: 100%"
     }
    },
    "7ceb3acbd1bc454ca8885aa4fb26b4b8": {
     "model_module": "@jupyter-widgets/controls",
     "model_module_version": "1.5.0",
     "model_name": "HTMLModel",
     "state": {
      "_dom_classes": [],
      "_model_module": "@jupyter-widgets/controls",
      "_model_module_version": "1.5.0",
      "_model_name": "HTMLModel",
      "_view_count": null,
      "_view_module": "@jupyter-widgets/controls",
      "_view_module_version": "1.5.0",
      "_view_name": "HTMLView",
      "description": "",
      "description_tooltip": null,
      "layout": "IPY_MODEL_1471b97b61204f11b4a4d59c9aa91101",
      "placeholder": "​",
      "style": "IPY_MODEL_aa9a238f10144400a0872c7b15304d22",
      "value": "Generating train split: 100%"
     }
    },
    "83b0b11fa7fa4181883f5c2e9a371f39": {
     "model_module": "@jupyter-widgets/controls",
     "model_module_version": "1.5.0",
     "model_name": "HTMLModel",
     "state": {
      "_dom_classes": [],
      "_model_module": "@jupyter-widgets/controls",
      "_model_module_version": "1.5.0",
      "_model_name": "HTMLModel",
      "_view_count": null,
      "_view_module": "@jupyter-widgets/controls",
      "_view_module_version": "1.5.0",
      "_view_name": "HTMLView",
      "description": "",
      "description_tooltip": null,
      "layout": "IPY_MODEL_5b75b2599262447381e838982c69b67f",
      "placeholder": "​",
      "style": "IPY_MODEL_ca97885d6aae4341a88cf189a1e11df2",
      "value": " 3.33k/3.33k [00:00&lt;00:00, 175kB/s]"
     }
    },
    "9015081a5ca2450baff5e5fed9e066ee": {
     "model_module": "@jupyter-widgets/controls",
     "model_module_version": "1.5.0",
     "model_name": "HBoxModel",
     "state": {
      "_dom_classes": [],
      "_model_module": "@jupyter-widgets/controls",
      "_model_module_version": "1.5.0",
      "_model_name": "HBoxModel",
      "_view_count": null,
      "_view_module": "@jupyter-widgets/controls",
      "_view_module_version": "1.5.0",
      "_view_name": "HBoxView",
      "box_style": "",
      "children": [
       "IPY_MODEL_7ceb3acbd1bc454ca8885aa4fb26b4b8",
       "IPY_MODEL_4f8183d901e3493a86a84c227674c722",
       "IPY_MODEL_52065470ba4042dc9c717b38410ce787"
      ],
      "layout": "IPY_MODEL_3460d2d487224d13ad25e456d1053422"
     }
    },
    "98da8cd4fb9f4e12828a013e80c3d25b": {
     "model_module": "@jupyter-widgets/controls",
     "model_module_version": "1.5.0",
     "model_name": "HTMLModel",
     "state": {
      "_dom_classes": [],
      "_model_module": "@jupyter-widgets/controls",
      "_model_module_version": "1.5.0",
      "_model_name": "HTMLModel",
      "_view_count": null,
      "_view_module": "@jupyter-widgets/controls",
      "_view_module_version": "1.5.0",
      "_view_name": "HTMLView",
      "description": "",
      "description_tooltip": null,
      "layout": "IPY_MODEL_3de0500b40894dc7b8f8f133b1eba694",
      "placeholder": "​",
      "style": "IPY_MODEL_f31b771fa3af483da4fdbf50ec5dd754",
      "value": "Downloading readme: 100%"
     }
    },
    "aa9a238f10144400a0872c7b15304d22": {
     "model_module": "@jupyter-widgets/controls",
     "model_module_version": "1.5.0",
     "model_name": "DescriptionStyleModel",
     "state": {
      "_model_module": "@jupyter-widgets/controls",
      "_model_module_version": "1.5.0",
      "_model_name": "DescriptionStyleModel",
      "_view_count": null,
      "_view_module": "@jupyter-widgets/base",
      "_view_module_version": "1.2.0",
      "_view_name": "StyleView",
      "description_width": ""
     }
    },
    "bf0b076013a142e99a25bcc0e17bfd30": {
     "model_module": "@jupyter-widgets/base",
     "model_module_version": "1.2.0",
     "model_name": "LayoutModel",
     "state": {
      "_model_module": "@jupyter-widgets/base",
      "_model_module_version": "1.2.0",
      "_model_name": "LayoutModel",
      "_view_count": null,
      "_view_module": "@jupyter-widgets/base",
      "_view_module_version": "1.2.0",
      "_view_name": "LayoutView",
      "align_content": null,
      "align_items": null,
      "align_self": null,
      "border": null,
      "bottom": null,
      "display": null,
      "flex": null,
      "flex_flow": null,
      "grid_area": null,
      "grid_auto_columns": null,
      "grid_auto_flow": null,
      "grid_auto_rows": null,
      "grid_column": null,
      "grid_gap": null,
      "grid_row": null,
      "grid_template_areas": null,
      "grid_template_columns": null,
      "grid_template_rows": null,
      "height": null,
      "justify_content": null,
      "justify_items": null,
      "left": null,
      "margin": null,
      "max_height": null,
      "max_width": null,
      "min_height": null,
      "min_width": null,
      "object_fit": null,
      "object_position": null,
      "order": null,
      "overflow": null,
      "overflow_x": null,
      "overflow_y": null,
      "padding": null,
      "right": null,
      "top": null,
      "visibility": null,
      "width": null
     }
    },
    "ca97885d6aae4341a88cf189a1e11df2": {
     "model_module": "@jupyter-widgets/controls",
     "model_module_version": "1.5.0",
     "model_name": "DescriptionStyleModel",
     "state": {
      "_model_module": "@jupyter-widgets/controls",
      "_model_module_version": "1.5.0",
      "_model_name": "DescriptionStyleModel",
      "_view_count": null,
      "_view_module": "@jupyter-widgets/base",
      "_view_module_version": "1.2.0",
      "_view_name": "StyleView",
      "description_width": ""
     }
    },
    "d076907f2ad34dddab38ad064e8675a0": {
     "model_module": "@jupyter-widgets/base",
     "model_module_version": "1.2.0",
     "model_name": "LayoutModel",
     "state": {
      "_model_module": "@jupyter-widgets/base",
      "_model_module_version": "1.2.0",
      "_model_name": "LayoutModel",
      "_view_count": null,
      "_view_module": "@jupyter-widgets/base",
      "_view_module_version": "1.2.0",
      "_view_name": "LayoutView",
      "align_content": null,
      "align_items": null,
      "align_self": null,
      "border": null,
      "bottom": null,
      "display": null,
      "flex": null,
      "flex_flow": null,
      "grid_area": null,
      "grid_auto_columns": null,
      "grid_auto_flow": null,
      "grid_auto_rows": null,
      "grid_column": null,
      "grid_gap": null,
      "grid_row": null,
      "grid_template_areas": null,
      "grid_template_columns": null,
      "grid_template_rows": null,
      "height": null,
      "justify_content": null,
      "justify_items": null,
      "left": null,
      "margin": null,
      "max_height": null,
      "max_width": null,
      "min_height": null,
      "min_width": null,
      "object_fit": null,
      "object_position": null,
      "order": null,
      "overflow": null,
      "overflow_x": null,
      "overflow_y": null,
      "padding": null,
      "right": null,
      "top": null,
      "visibility": null,
      "width": null
     }
    },
    "d09102d2ef544c27be5c25857a1a1cec": {
     "model_module": "@jupyter-widgets/controls",
     "model_module_version": "1.5.0",
     "model_name": "FloatProgressModel",
     "state": {
      "_dom_classes": [],
      "_model_module": "@jupyter-widgets/controls",
      "_model_module_version": "1.5.0",
      "_model_name": "FloatProgressModel",
      "_view_count": null,
      "_view_module": "@jupyter-widgets/controls",
      "_view_module_version": "1.5.0",
      "_view_name": "ProgressView",
      "bar_style": "success",
      "description": "",
      "description_tooltip": null,
      "layout": "IPY_MODEL_00d5b332421f49218a30d2711b400e59",
      "max": 3334,
      "min": 0,
      "orientation": "horizontal",
      "style": "IPY_MODEL_3bc583ba49994c32a51e986357d0f8b4",
      "value": 3334
     }
    },
    "d72fd9f9f0a541cf94a93bcbac3c4bad": {
     "model_module": "@jupyter-widgets/controls",
     "model_module_version": "1.5.0",
     "model_name": "DescriptionStyleModel",
     "state": {
      "_model_module": "@jupyter-widgets/controls",
      "_model_module_version": "1.5.0",
      "_model_name": "DescriptionStyleModel",
      "_view_count": null,
      "_view_module": "@jupyter-widgets/base",
      "_view_module_version": "1.2.0",
      "_view_name": "StyleView",
      "description_width": ""
     }
    },
    "ddb120fa2e7b4fed81201c9e88e9499d": {
     "model_module": "@jupyter-widgets/base",
     "model_module_version": "1.2.0",
     "model_name": "LayoutModel",
     "state": {
      "_model_module": "@jupyter-widgets/base",
      "_model_module_version": "1.2.0",
      "_model_name": "LayoutModel",
      "_view_count": null,
      "_view_module": "@jupyter-widgets/base",
      "_view_module_version": "1.2.0",
      "_view_name": "LayoutView",
      "align_content": null,
      "align_items": null,
      "align_self": null,
      "border": null,
      "bottom": null,
      "display": null,
      "flex": null,
      "flex_flow": null,
      "grid_area": null,
      "grid_auto_columns": null,
      "grid_auto_flow": null,
      "grid_auto_rows": null,
      "grid_column": null,
      "grid_gap": null,
      "grid_row": null,
      "grid_template_areas": null,
      "grid_template_columns": null,
      "grid_template_rows": null,
      "height": null,
      "justify_content": null,
      "justify_items": null,
      "left": null,
      "margin": null,
      "max_height": null,
      "max_width": null,
      "min_height": null,
      "min_width": null,
      "object_fit": null,
      "object_position": null,
      "order": null,
      "overflow": null,
      "overflow_x": null,
      "overflow_y": null,
      "padding": null,
      "right": null,
      "top": null,
      "visibility": null,
      "width": null
     }
    },
    "dedc9c3fb09946bbae2568b52577b391": {
     "model_module": "@jupyter-widgets/base",
     "model_module_version": "1.2.0",
     "model_name": "LayoutModel",
     "state": {
      "_model_module": "@jupyter-widgets/base",
      "_model_module_version": "1.2.0",
      "_model_name": "LayoutModel",
      "_view_count": null,
      "_view_module": "@jupyter-widgets/base",
      "_view_module_version": "1.2.0",
      "_view_name": "LayoutView",
      "align_content": null,
      "align_items": null,
      "align_self": null,
      "border": null,
      "bottom": null,
      "display": null,
      "flex": null,
      "flex_flow": null,
      "grid_area": null,
      "grid_auto_columns": null,
      "grid_auto_flow": null,
      "grid_auto_rows": null,
      "grid_column": null,
      "grid_gap": null,
      "grid_row": null,
      "grid_template_areas": null,
      "grid_template_columns": null,
      "grid_template_rows": null,
      "height": null,
      "justify_content": null,
      "justify_items": null,
      "left": null,
      "margin": null,
      "max_height": null,
      "max_width": null,
      "min_height": null,
      "min_width": null,
      "object_fit": null,
      "object_position": null,
      "order": null,
      "overflow": null,
      "overflow_x": null,
      "overflow_y": null,
      "padding": null,
      "right": null,
      "top": null,
      "visibility": null,
      "width": null
     }
    },
    "ed6673f2a3424db785af9978150cdf8e": {
     "model_module": "@jupyter-widgets/controls",
     "model_module_version": "1.5.0",
     "model_name": "DescriptionStyleModel",
     "state": {
      "_model_module": "@jupyter-widgets/controls",
      "_model_module_version": "1.5.0",
      "_model_name": "DescriptionStyleModel",
      "_view_count": null,
      "_view_module": "@jupyter-widgets/base",
      "_view_module_version": "1.2.0",
      "_view_name": "StyleView",
      "description_width": ""
     }
    },
    "f31b771fa3af483da4fdbf50ec5dd754": {
     "model_module": "@jupyter-widgets/controls",
     "model_module_version": "1.5.0",
     "model_name": "DescriptionStyleModel",
     "state": {
      "_model_module": "@jupyter-widgets/controls",
      "_model_module_version": "1.5.0",
      "_model_name": "DescriptionStyleModel",
      "_view_count": null,
      "_view_module": "@jupyter-widgets/base",
      "_view_module_version": "1.2.0",
      "_view_name": "StyleView",
      "description_width": ""
     }
    },
    "f5cda5e56f3f40c5a5ecc47bca536df5": {
     "model_module": "@jupyter-widgets/base",
     "model_module_version": "1.2.0",
     "model_name": "LayoutModel",
     "state": {
      "_model_module": "@jupyter-widgets/base",
      "_model_module_version": "1.2.0",
      "_model_name": "LayoutModel",
      "_view_count": null,
      "_view_module": "@jupyter-widgets/base",
      "_view_module_version": "1.2.0",
      "_view_name": "LayoutView",
      "align_content": null,
      "align_items": null,
      "align_self": null,
      "border": null,
      "bottom": null,
      "display": null,
      "flex": null,
      "flex_flow": null,
      "grid_area": null,
      "grid_auto_columns": null,
      "grid_auto_flow": null,
      "grid_auto_rows": null,
      "grid_column": null,
      "grid_gap": null,
      "grid_row": null,
      "grid_template_areas": null,
      "grid_template_columns": null,
      "grid_template_rows": null,
      "height": null,
      "justify_content": null,
      "justify_items": null,
      "left": null,
      "margin": null,
      "max_height": null,
      "max_width": null,
      "min_height": null,
      "min_width": null,
      "object_fit": null,
      "object_position": null,
      "order": null,
      "overflow": null,
      "overflow_x": null,
      "overflow_y": null,
      "padding": null,
      "right": null,
      "top": null,
      "visibility": null,
      "width": null
     }
    }
   }
  }
 },
 "nbformat": 4,
 "nbformat_minor": 0
}