{
 "cells": [
  {
   "cell_type": "code",
   "outputs": [],
   "source": [
    "import instructor\n",
    "from pydantic import BaseModel\n",
    "from openai import OpenAI\n",
    "\n",
    "import dotenv\n",
    "from tqdm.auto import trange\n",
    "\n",
    "import json"
   ],
   "metadata": {
    "collapsed": false,
    "ExecuteTime": {
     "end_time": "2024-05-23T21:23:26.793952Z",
     "start_time": "2024-05-23T21:23:26.789582Z"
    }
   },
   "id": "1b2180343e5180dd",
   "execution_count": 36
  },
  {
   "cell_type": "code",
   "outputs": [
    {
     "data": {
      "text/plain": "True"
     },
     "execution_count": 20,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "dotenv.load_dotenv()"
   ],
   "metadata": {
    "collapsed": false,
    "ExecuteTime": {
     "end_time": "2024-05-23T20:49:24.686772Z",
     "start_time": "2024-05-23T20:49:24.683815Z"
    }
   },
   "id": "1296d472faf81f48",
   "execution_count": 20
  },
  {
   "cell_type": "code",
   "outputs": [],
   "source": [
    "def get_initial_messages(transcript: str):\n",
    "    return [\n",
    "        {\n",
    "            \"role\": \"system\",\n",
    "            \"content\": [\n",
    "                {\n",
    "                    \"type\": \"text\",\n",
    "                    \"text\": \"You are a trivia bot writing fun questions for a trivia game. You have been given a transcript of a video. You need to write two questions based on the transcript. Do not refer explicitly to the transcript - it should be a valid question at a trivia night, for fans of the show, which is Hot Ones, Season 19. The questions should be in a JSON format.\"\n",
    "                }\n",
    "            ]\n",
    "        },\n",
    "        {\n",
    "            \"role\": \"user\",\n",
    "            \"content\": [\n",
    "                {\n",
    "                    \"type\": \"text\",\n",
    "                    \"text\": f\"Transcript: {transcript}\"\n",
    "                }\n",
    "            ]\n",
    "        }\n",
    "    ]"
   ],
   "metadata": {
    "collapsed": false,
    "ExecuteTime": {
     "end_time": "2024-05-23T21:19:34.503329Z",
     "start_time": "2024-05-23T21:19:34.498Z"
    }
   },
   "id": "e511c7f86475ac01",
   "execution_count": 30
  },
  {
   "cell_type": "code",
   "outputs": [],
   "source": [
    "class TriviaQuestions(BaseModel):\n",
    "    question1: str\n",
    "    answer1: str\n",
    "    question2: str\n",
    "    answer2: str\n"
   ],
   "metadata": {
    "collapsed": false,
    "ExecuteTime": {
     "end_time": "2024-05-23T21:19:34.945726Z",
     "start_time": "2024-05-23T21:19:34.944360Z"
    }
   },
   "id": "66b2c3c81748b42e",
   "execution_count": 31
  },
  {
   "cell_type": "code",
   "outputs": [],
   "source": [
    "\n",
    "# Patch the OpenAI client\n",
    "client = instructor.from_openai(OpenAI())"
   ],
   "metadata": {
    "collapsed": false,
    "ExecuteTime": {
     "end_time": "2024-05-23T21:19:35.327482Z",
     "start_time": "2024-05-23T21:19:35.313726Z"
    }
   },
   "id": "e3f857ded1a29732",
   "execution_count": 32
  },
  {
   "cell_type": "code",
   "outputs": [
    {
     "data": {
      "text/plain": "  0%|          | 0/17 [00:00<?, ?it/s]",
      "application/vnd.jupyter.widget-view+json": {
       "version_major": 2,
       "version_minor": 0,
       "model_id": "497862b5404a4e3c9d9f7ae1a170b8eb"
      }
     },
     "metadata": {},
     "output_type": "display_data"
    }
   ],
   "source": [
    "all_trivia = []\n",
    "for i in trange(17):\n",
    "    with open(f\"transcripts/{i}.vtt\") as f:\n",
    "        text = f.read()\n",
    "    msg = get_initial_messages(text)\n",
    "    \n",
    "    trivia = client.chat.completions.create(\n",
    "        model=\"gpt-4o\",\n",
    "        response_model=TriviaQuestions,\n",
    "        messages=msg,\n",
    "    )\n",
    "    \n",
    "    all_trivia.append(trivia)"
   ],
   "metadata": {
    "collapsed": false,
    "ExecuteTime": {
     "end_time": "2024-05-23T21:20:28.838601Z",
     "start_time": "2024-05-23T21:19:39.080663Z"
    }
   },
   "id": "d472211cac29e59c",
   "execution_count": 33
  },
  {
   "cell_type": "code",
   "outputs": [],
   "source": [
    "all_questions = sum([[t.question1, t.question2] for t in all_trivia], [])\n",
    "all_answers = sum([[t.answer1, t.answer2] for t in all_trivia], [])"
   ],
   "metadata": {
    "collapsed": false,
    "ExecuteTime": {
     "end_time": "2024-05-23T21:21:13.583657Z",
     "start_time": "2024-05-23T21:21:13.580310Z"
    }
   },
   "id": "fddeb7e20c845467",
   "execution_count": 34
  },
  {
   "cell_type": "code",
   "outputs": [
    {
     "data": {
      "text/plain": "[\"Which magician's documentary adventure series titled 'Beyond Belief' is coming soon to Disney Plus?\",\n 'In Hot Ones Season 19, which magician recounted learning to expel water like a stream from a performer in Liberia?',\n 'Who was the guest on the Hot Ones episode hosted by Sean Evans in which they joked about feeling their whiskers tingling?',\n 'In the Hot Ones episode with Puss in Boots, what common beverage was mentioned multiple times as being essential during the challenge?',\n 'Which charity is partnered with Hot Ones during their holiday fundraising efforts?',\n 'Who won the Hot Ones 2022 Lifetime Achievement Award?',\n \"Who did Emma Chamberlain claim she would marry, according to her 'F, Marry, Kill' choices involving milk types?\",\n \"What was Emma Chamberlain's major cooking success as mentioned on Hot Ones?\",\n 'Who hosted the Hot Ones season 19 Thanksgiving special with Israel Adesanya?',\n 'Which UFC fighter mentioned in the Hot Ones season 19 Thanksgiving special has defended his middleweight championship belt five times?',\n \"What challenges did the production face when shooting epic battle scenes in the film 'The Woman King' starring Viola Davis?\",\n \"Which television character portrayed by Viola Davis underwent significant transformation from the script to on-screen performance, as she discussed during the interview on 'Hot Ones'?\",\n 'In Hot Ones Season 19, which actress overcomes her spicy food fears while promoting the psychological drama \"TÁR\"?',\n 'Which challenging piece did Cate Blanchett preside over during a scene from the movie \"TÁR\" in Hot Ones Season 19?',\n \"What is the name of Kid Cudi's album and animated TV special mentioned on Hot Ones Season 19?\",\n 'Which fashion designer was responsible for the costume design in Kid Cudi\\'s animated TV special \"Intergalactic\"?',\n 'Which Emmy award-winning host appeared on Season 19 of Hot Ones and is known for The Late Late Show?',\n 'In which dark comedy drama series, set to release on Prime Video on November 11th, does James Corden star alongside Sally Hawkins?',\n 'Which actress, known for roles in Avatar, Star Trek, and Guardians of the Galaxy, appeared on Hot Ones in Season 19?',\n 'What upcoming movie directed by James Cameron was promoted by Zoe Saldana on Hot Ones, set for release on December 16th?',\n 'In Hot Ones Season 19, which actress and entrepreneur joined Sean Evans for an episode, discussing everything from her business ventures to her film roles?',\n \"In her Hot Ones appearance, which upcoming Netflix film did Kate Hudson promote, described as the 'long-weighted follow-up film'?\",\n 'What semi-autobiographical film directed by Steven Spielberg features Paul Dano as one of its main characters?',\n \"Which song by Nirvana influenced Paul Dano's portrayal of the Riddler?\",\n 'Who created the media company Lyrical Lemonade that has worked with artists like Eminem and Post Malone?',\n \"Who directed the music video for Eminem's song 'Godzilla' which featured Mike Tyson?\",\n 'What ingredient is mentioned in the transcript as a component of the spicy food being consumed?',\n 'What does one of the participants ask for during the spicy challenge in the transcript?',\n 'Which Hot Ones sauce featuring a sweet heat combination includes maple syrup, apple cider vinegar, and Fresno chilies?',\n 'What is the name of the Hot Ones sauce from Montreal, Canada, that features a blend of fruits like mango, papaya, and banana, combined with Scotch bonnet peppers?',\n 'Who was the guest on the episode of Hot Ones hosted by Sean Evans, where the guest discussed his struggles with faith and being human?',\n 'What comedy-drama series co-created by Rami Youssef was mentioned, which is now streaming on Netflix?',\n 'What is the name of the final sauce used in the Hot Ones challenge, often added as a tradition on the last wing?',\n 'Which hot sauce challenge involves eating a wing with extra sauce added at the end as a final tradition on the show Hot Ones?']"
     },
     "execution_count": 37,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "all_questions"
   ],
   "metadata": {
    "collapsed": false,
    "ExecuteTime": {
     "end_time": "2024-05-23T21:23:30.012900Z",
     "start_time": "2024-05-23T21:23:30.010901Z"
    }
   },
   "id": "6726839ddeb01f30",
   "execution_count": 37
  },
  {
   "cell_type": "code",
   "outputs": [],
   "source": [
    "with open(\"questions.json\", \"w\") as f:\n",
    "    json.dump(all_questions, f)"
   ],
   "metadata": {
    "collapsed": false,
    "ExecuteTime": {
     "end_time": "2024-05-23T21:23:30.594300Z",
     "start_time": "2024-05-23T21:23:30.589748Z"
    }
   },
   "id": "2adde26e15e3bb21",
   "execution_count": 38
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 2
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython2",
   "version": "2.7.6"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}