Chatbot chatGPT with  save & load & continue v1.0.ipynb

{
 "cells": [
  {
   "cell_type": "markdown",
   "id": "1a2381dd",
   "metadata": {},
   "source": [
    "### \"Creating a chatGPT chatbot in Python with Save, Load and Continue\"\n",
    "Author: Sergiu Iatco | 2023.04.14\n",
    "\n",
    "Useful links: <br>\n",
    "https://help.openai.com/en/articles/4936850-where-do-i-find-my-secret-api-key <br>\n",
    "https://towardsdatascience.com/build-your-own-chatGPT-like-app-with-streamlit-20d940417389 <br>\n",
    "https://blogs.sap.com/2023/02/15/chatGPT-for-sap-developments-threat-or-an-opportunity <br>\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 1,
   "id": "e56fce4e",
   "metadata": {},
   "outputs": [],
   "source": [
    "import openai\n",
    "import json"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "id": "3f75d033",
   "metadata": {},
   "outputs": [],
   "source": [
    "# from cryptography.fernet import Fernet # consider encryption of password\n",
    "# Set org ID and API key\n",
    "# openai_organization = \"<YOUR_OPENAI_ORG_ID>\"\n",
    "openai_api_key = \"<YOUR_OPENAI_API_KEY>\" # replace with a valid key"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "id": "96f938c0",
   "metadata": {},
   "outputs": [],
   "source": [
    "class chatGPT:\n",
    "    def __init__(self, api_key, model):\n",
    "        openai.api_key = api_key\n",
    "        self.model = model\n",
    "        \n",
    "        self.messages = list()\n",
    "        self.responses = list()\n",
    "        self.completions = list()        \n",
    "\n",
    "        self._file_default = \"chatbot_data.json\"\n",
    "        \n",
    "    def message_post(self, prompt):\n",
    "        self.messages.append({\"role\": \"user\", \"content\": prompt})\n",
    "        return self.messages\n",
    "\n",
    "    def generate_response(self):\n",
    "        completion = openai.ChatCompletion.create(\n",
    "            model=self.model,\n",
    "            messages=self.messages\n",
    "        )\n",
    "        response = completion.choices[0].message.content\n",
    "        self.messages.append({\"role\": \"assistant\", \"content\": response})\n",
    "        \n",
    "        self.response = response\n",
    "        self.responses.append(response)\n",
    "                \n",
    "        self.completion = completion\n",
    "        self.completions.append(completion)\n",
    "        \n",
    "        return self.response\n",
    "\n",
    "    def reset(self):\n",
    "        self.messages = list()\n",
    "        self.responses = list()\n",
    "        self.completions = list()        \n",
    "        \n",
    "    def save_data(self, file_path = None):\n",
    "        data = {\n",
    "            \"responses\": self.responses,\n",
    "            \"completions\": self.completions,\n",
    "            \"messages\": self.messages\n",
    "        }\n",
    "        \n",
    "        if file_path is None:\n",
    "            file_path = self._file_default\n",
    "        \n",
    "        with open(file_path, \"w\") as f:\n",
    "            json.dump(data, f)\n",
    "            \n",
    "    def load_data(self, file_path = None):\n",
    "        if file_path is None:\n",
    "            file_path = self._file_default\n",
    "        \n",
    "        with open(file_path, \"r\") as f:\n",
    "            data = json.load(f)\n",
    "            responses = data[\"responses\"]\n",
    "            completions = data[\"completions\"]\n",
    "            messages = data[\"messages\"]\n",
    "            return responses, completions, messages\n",
    "    \n",
    "    def load_messages(self, messages):\n",
    "        self.messages = messages \n",
    "\n",
    "    def list_messages(self):\n",
    "        # list all messages\n",
    "        for e in self.messages:\n",
    "            print(f\"role: {e['role']}\")\n",
    "            print(f\"content:\\n{e['content']}\")\n",
    "            print(f\"\\n\")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "id": "70453489",
   "metadata": {},
   "outputs": [],
   "source": [
    "# initialize\n",
    "my_chatbot = chatGPT(api_key=openai_api_key, model=\"gpt-3.5-turbo\")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "id": "72664ed4",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "1. Bucharest\n",
      "2. Cluj-Napoca\n",
      "3. Timișoara\n",
      "4. Iași\n",
      "5. Craiova\n",
      "6. Constanța\n",
      "7. Galați\n",
      "8. Brașov\n",
      "9. Ploiești\n",
      "10. Oradea\n",
      "11. Arad\n",
      "12. Sibiu\n",
      "13. Bacău\n",
      "14. Baia Mare\n",
      "15. Buzău\n",
      "16. Focșani\n",
      "17. Suceava\n",
      "18. Târgu Mureș\n",
      "19. Bistrița\n",
      "20. Satu Mare.\n",
      "CPU times: total: 78.1 ms\n",
      "Wall time: 40.7 s\n"
     ]
    }
   ],
   "source": [
    "%%time\n",
    "# your question\n",
    "my_chatbot.message_post(\"Romania cities\")\n",
    "response = my_chatbot.generate_response()\n",
    "print(response)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "id": "553a428e",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "[{'role': 'user', 'content': 'Romania cities'},\n",
       " {'role': 'assistant',\n",
       "  'content': '1. Bucharest\\n2. Cluj-Napoca\\n3. Timișoara\\n4. Iași\\n5. Craiova\\n6. Constanța\\n7. Galați\\n8. Brașov\\n9. Ploiești\\n10. Oradea\\n11. Arad\\n12. Sibiu\\n13. Bacău\\n14. Baia Mare\\n15. Buzău\\n16. Focșani\\n17. Suceava\\n18. Târgu Mureș\\n19. Bistrița\\n20. Satu Mare.'}]"
      ]
     },
     "execution_count": 6,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# stored messages\n",
    "my_chatbot.messages"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 7,
   "id": "8c6f9500",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "'1. Bucharest\\n2. Cluj-Napoca\\n3. Timișoara\\n4. Iași\\n5. Craiova\\n6. Constanța\\n7. Galați\\n8. Brașov\\n9. Ploiești\\n10. Oradea\\n11. Arad\\n12. Sibiu\\n13. Bacău\\n14. Baia Mare\\n15. Buzău\\n16. Focșani\\n17. Suceava\\n18. Târgu Mureș\\n19. Bistrița\\n20. Satu Mare.'"
      ]
     },
     "execution_count": 7,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# response\n",
    "my_chatbot.response"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 8,
   "id": "c67e8289",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "<OpenAIObject chat.completion id=chatcmpl-74O3WnmLwaz4zqvmTUs29Jvyg3IwT at 0x20060e0ae50> JSON: {\n",
       "  \"choices\": [\n",
       "    {\n",
       "      \"finish_reason\": \"stop\",\n",
       "      \"index\": 0,\n",
       "      \"message\": {\n",
       "        \"content\": \"1. Bucharest\\n2. Cluj-Napoca\\n3. Timi\\u0219oara\\n4. Ia\\u0219i\\n5. Craiova\\n6. Constan\\u021ba\\n7. Gala\\u021bi\\n8. Bra\\u0219ov\\n9. Ploie\\u0219ti\\n10. Oradea\\n11. Arad\\n12. Sibiu\\n13. Bac\\u0103u\\n14. Baia Mare\\n15. Buz\\u0103u\\n16. Foc\\u0219ani\\n17. Suceava\\n18. T\\u00e2rgu Mure\\u0219\\n19. Bistri\\u021ba\\n20. Satu Mare.\",\n",
       "        \"role\": \"assistant\"\n",
       "      }\n",
       "    }\n",
       "  ],\n",
       "  \"created\": 1681280138,\n",
       "  \"id\": \"chatcmpl-74O3WnmLwaz4zqvmTUs29Jvyg3IwT\",\n",
       "  \"model\": \"gpt-3.5-turbo-0301\",\n",
       "  \"object\": \"chat.completion\",\n",
       "  \"usage\": {\n",
       "    \"completion_tokens\": 132,\n",
       "    \"prompt_tokens\": 11,\n",
       "    \"total_tokens\": 143\n",
       "  }\n",
       "}"
      ]
     },
     "execution_count": 8,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# completion\n",
    "my_chatbot.completion"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 9,
   "id": "c7f171c9",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Here are the estimated populations (as of 2021) for the above cities in Romania:\n",
      "\n",
      "1. Bucharest - 1,836,000\n",
      "2. Cluj-Napoca - 335,000\n",
      "3. Timișoara - 317,000\n",
      "4. Iași - 290,000\n",
      "5. Craiova - 269,000\n",
      "6. Constanța - 259,000\n",
      "7. Galați - 249,000\n",
      "8. Brașov - 227,000\n",
      "9. Ploiești - 203,000\n",
      "10. Oradea - 200,000\n",
      "11. Arad - 157,000\n",
      "12. Sibiu - 147,000\n",
      "13. Bacău - 144,000\n",
      "14. Baia Mare - 115,000\n",
      "15. Buzău - 111,000\n",
      "16. Focșani - 93,000\n",
      "17. Suceava - 86,000\n",
      "18. Târgu Mureș - 82,000\n",
      "19. Bistrița - 75,000\n",
      "20. Satu Mare - 68,000\n",
      "CPU times: total: 15.6 ms\n",
      "Wall time: 1min 2s\n"
     ]
    }
   ],
   "source": [
    "%%time\n",
    "# you question\n",
    "my_chatbot.message_post(\"population in above cities\")\n",
    "response = my_chatbot.generate_response()\n",
    "print(response)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 10,
   "id": "9dde543f",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "['1. Bucharest\\n2. Cluj-Napoca\\n3. Timișoara\\n4. Iași\\n5. Craiova\\n6. Constanța\\n7. Galați\\n8. Brașov\\n9. Ploiești\\n10. Oradea\\n11. Arad\\n12. Sibiu\\n13. Bacău\\n14. Baia Mare\\n15. Buzău\\n16. Focșani\\n17. Suceava\\n18. Târgu Mureș\\n19. Bistrița\\n20. Satu Mare.',\n",
       " 'Here are the estimated populations (as of 2021) for the above cities in Romania:\\n\\n1. Bucharest - 1,836,000\\n2. Cluj-Napoca - 335,000\\n3. Timișoara - 317,000\\n4. Iași - 290,000\\n5. Craiova - 269,000\\n6. Constanța - 259,000\\n7. Galați - 249,000\\n8. Brașov - 227,000\\n9. Ploiești - 203,000\\n10. Oradea - 200,000\\n11. Arad - 157,000\\n12. Sibiu - 147,000\\n13. Bacău - 144,000\\n14. Baia Mare - 115,000\\n15. Buzău - 111,000\\n16. Focșani - 93,000\\n17. Suceava - 86,000\\n18. Târgu Mureș - 82,000\\n19. Bistrița - 75,000\\n20. Satu Mare - 68,000']"
      ]
     },
     "execution_count": 10,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# stored responses\n",
    "my_chatbot.responses"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 11,
   "id": "6f997df0",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "[<OpenAIObject chat.completion id=chatcmpl-74O3WnmLwaz4zqvmTUs29Jvyg3IwT at 0x20060e0ae50> JSON: {\n",
       "   \"choices\": [\n",
       "     {\n",
       "       \"finish_reason\": \"stop\",\n",
       "       \"index\": 0,\n",
       "       \"message\": {\n",
       "         \"content\": \"1. Bucharest\\n2. Cluj-Napoca\\n3. Timi\\u0219oara\\n4. Ia\\u0219i\\n5. Craiova\\n6. Constan\\u021ba\\n7. Gala\\u021bi\\n8. Bra\\u0219ov\\n9. Ploie\\u0219ti\\n10. Oradea\\n11. Arad\\n12. Sibiu\\n13. Bac\\u0103u\\n14. Baia Mare\\n15. Buz\\u0103u\\n16. Foc\\u0219ani\\n17. Suceava\\n18. T\\u00e2rgu Mure\\u0219\\n19. Bistri\\u021ba\\n20. Satu Mare.\",\n",
       "         \"role\": \"assistant\"\n",
       "       }\n",
       "     }\n",
       "   ],\n",
       "   \"created\": 1681280138,\n",
       "   \"id\": \"chatcmpl-74O3WnmLwaz4zqvmTUs29Jvyg3IwT\",\n",
       "   \"model\": \"gpt-3.5-turbo-0301\",\n",
       "   \"object\": \"chat.completion\",\n",
       "   \"usage\": {\n",
       "     \"completion_tokens\": 132,\n",
       "     \"prompt_tokens\": 11,\n",
       "     \"total_tokens\": 143\n",
       "   }\n",
       " },\n",
       " <OpenAIObject chat.completion id=chatcmpl-74O4BvwdjSIpCkIyCXjRKy1y86CbY at 0x20060de5040> JSON: {\n",
       "   \"choices\": [\n",
       "     {\n",
       "       \"finish_reason\": \"stop\",\n",
       "       \"index\": 0,\n",
       "       \"message\": {\n",
       "         \"content\": \"Here are the estimated populations (as of 2021) for the above cities in Romania:\\n\\n1. Bucharest - 1,836,000\\n2. Cluj-Napoca - 335,000\\n3. Timi\\u0219oara - 317,000\\n4. Ia\\u0219i - 290,000\\n5. Craiova - 269,000\\n6. Constan\\u021ba - 259,000\\n7. Gala\\u021bi - 249,000\\n8. Bra\\u0219ov - 227,000\\n9. Ploie\\u0219ti - 203,000\\n10. Oradea - 200,000\\n11. Arad - 157,000\\n12. Sibiu - 147,000\\n13. Bac\\u0103u - 144,000\\n14. Baia Mare - 115,000\\n15. Buz\\u0103u - 111,000\\n16. Foc\\u0219ani - 93,000\\n17. Suceava - 86,000\\n18. T\\u00e2rgu Mure\\u0219 - 82,000\\n19. Bistri\\u021ba - 75,000\\n20. Satu Mare - 68,000\",\n",
       "         \"role\": \"assistant\"\n",
       "       }\n",
       "     }\n",
       "   ],\n",
       "   \"created\": 1681280179,\n",
       "   \"id\": \"chatcmpl-74O4BvwdjSIpCkIyCXjRKy1y86CbY\",\n",
       "   \"model\": \"gpt-3.5-turbo-0301\",\n",
       "   \"object\": \"chat.completion\",\n",
       "   \"usage\": {\n",
       "     \"completion_tokens\": 252,\n",
       "     \"prompt_tokens\": 157,\n",
       "     \"total_tokens\": 409\n",
       "   }\n",
       " }]"
      ]
     },
     "execution_count": 11,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# stored completions\n",
    "my_chatbot.completions"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 12,
   "id": "c5bae36f",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "[{'role': 'user', 'content': 'Romania cities'},\n",
       " {'role': 'assistant',\n",
       "  'content': '1. Bucharest\\n2. Cluj-Napoca\\n3. Timișoara\\n4. Iași\\n5. Craiova\\n6. Constanța\\n7. Galați\\n8. Brașov\\n9. Ploiești\\n10. Oradea\\n11. Arad\\n12. Sibiu\\n13. Bacău\\n14. Baia Mare\\n15. Buzău\\n16. Focșani\\n17. Suceava\\n18. Târgu Mureș\\n19. Bistrița\\n20. Satu Mare.'},\n",
       " {'role': 'user', 'content': 'population in above cities'},\n",
       " {'role': 'assistant',\n",
       "  'content': 'Here are the estimated populations (as of 2021) for the above cities in Romania:\\n\\n1. Bucharest - 1,836,000\\n2. Cluj-Napoca - 335,000\\n3. Timișoara - 317,000\\n4. Iași - 290,000\\n5. Craiova - 269,000\\n6. Constanța - 259,000\\n7. Galați - 249,000\\n8. Brașov - 227,000\\n9. Ploiești - 203,000\\n10. Oradea - 200,000\\n11. Arad - 157,000\\n12. Sibiu - 147,000\\n13. Bacău - 144,000\\n14. Baia Mare - 115,000\\n15. Buzău - 111,000\\n16. Focșani - 93,000\\n17. Suceava - 86,000\\n18. Târgu Mureș - 82,000\\n19. Bistrița - 75,000\\n20. Satu Mare - 68,000'}]"
      ]
     },
     "execution_count": 12,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# stored messages\n",
    "my_chatbot.messages"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 13,
   "id": "53f4733b",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "'chatbot_data.json'"
      ]
     },
     "execution_count": 13,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# default file name\n",
    "my_chatbot._file_default"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 14,
   "id": "8683dbf4",
   "metadata": {},
   "outputs": [],
   "source": [
    "# save in a file responses, completions, messages\n",
    "my_chatbot.save_data()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 15,
   "id": "39eb1bcb",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "([], [], [])"
      ]
     },
     "execution_count": 15,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# reset - responses, completions, messages\n",
    "my_chatbot.reset()\n",
    "my_chatbot.responses, my_chatbot.completions, my_chatbot.messages"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 16,
   "id": "0bfbeee2",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "(['1. Bucharest\\n2. Cluj-Napoca\\n3. Timișoara\\n4. Iași\\n5. Craiova\\n6. Constanța\\n7. Galați\\n8. Brașov\\n9. Ploiești\\n10. Oradea\\n11. Arad\\n12. Sibiu\\n13. Bacău\\n14. Baia Mare\\n15. Buzău\\n16. Focșani\\n17. Suceava\\n18. Târgu Mureș\\n19. Bistrița\\n20. Satu Mare.',\n",
       "  'Here are the estimated populations (as of 2021) for the above cities in Romania:\\n\\n1. Bucharest - 1,836,000\\n2. Cluj-Napoca - 335,000\\n3. Timișoara - 317,000\\n4. Iași - 290,000\\n5. Craiova - 269,000\\n6. Constanța - 259,000\\n7. Galați - 249,000\\n8. Brașov - 227,000\\n9. Ploiești - 203,000\\n10. Oradea - 200,000\\n11. Arad - 157,000\\n12. Sibiu - 147,000\\n13. Bacău - 144,000\\n14. Baia Mare - 115,000\\n15. Buzău - 111,000\\n16. Focșani - 93,000\\n17. Suceava - 86,000\\n18. Târgu Mureș - 82,000\\n19. Bistrița - 75,000\\n20. Satu Mare - 68,000'],\n",
       " [{'id': 'chatcmpl-74O3WnmLwaz4zqvmTUs29Jvyg3IwT',\n",
       "   'object': 'chat.completion',\n",
       "   'created': 1681280138,\n",
       "   'model': 'gpt-3.5-turbo-0301',\n",
       "   'usage': {'prompt_tokens': 11,\n",
       "    'completion_tokens': 132,\n",
       "    'total_tokens': 143},\n",
       "   'choices': [{'message': {'role': 'assistant',\n",
       "      'content': '1. Bucharest\\n2. Cluj-Napoca\\n3. Timișoara\\n4. Iași\\n5. Craiova\\n6. Constanța\\n7. Galați\\n8. Brașov\\n9. Ploiești\\n10. Oradea\\n11. Arad\\n12. Sibiu\\n13. Bacău\\n14. Baia Mare\\n15. Buzău\\n16. Focșani\\n17. Suceava\\n18. Târgu Mureș\\n19. Bistrița\\n20. Satu Mare.'},\n",
       "     'finish_reason': 'stop',\n",
       "     'index': 0}]},\n",
       "  {'id': 'chatcmpl-74O4BvwdjSIpCkIyCXjRKy1y86CbY',\n",
       "   'object': 'chat.completion',\n",
       "   'created': 1681280179,\n",
       "   'model': 'gpt-3.5-turbo-0301',\n",
       "   'usage': {'prompt_tokens': 157,\n",
       "    'completion_tokens': 252,\n",
       "    'total_tokens': 409},\n",
       "   'choices': [{'message': {'role': 'assistant',\n",
       "      'content': 'Here are the estimated populations (as of 2021) for the above cities in Romania:\\n\\n1. Bucharest - 1,836,000\\n2. Cluj-Napoca - 335,000\\n3. Timișoara - 317,000\\n4. Iași - 290,000\\n5. Craiova - 269,000\\n6. Constanța - 259,000\\n7. Galați - 249,000\\n8. Brașov - 227,000\\n9. Ploiești - 203,000\\n10. Oradea - 200,000\\n11. Arad - 157,000\\n12. Sibiu - 147,000\\n13. Bacău - 144,000\\n14. Baia Mare - 115,000\\n15. Buzău - 111,000\\n16. Focșani - 93,000\\n17. Suceava - 86,000\\n18. Târgu Mureș - 82,000\\n19. Bistrița - 75,000\\n20. Satu Mare - 68,000'},\n",
       "     'finish_reason': 'stop',\n",
       "     'index': 0}]}],\n",
       " [{'role': 'user', 'content': 'Romania cities'},\n",
       "  {'role': 'assistant',\n",
       "   'content': '1. Bucharest\\n2. Cluj-Napoca\\n3. Timișoara\\n4. Iași\\n5. Craiova\\n6. Constanța\\n7. Galați\\n8. Brașov\\n9. Ploiești\\n10. Oradea\\n11. Arad\\n12. Sibiu\\n13. Bacău\\n14. Baia Mare\\n15. Buzău\\n16. Focșani\\n17. Suceava\\n18. Târgu Mureș\\n19. Bistrița\\n20. Satu Mare.'},\n",
       "  {'role': 'user', 'content': 'population in above cities'},\n",
       "  {'role': 'assistant',\n",
       "   'content': 'Here are the estimated populations (as of 2021) for the above cities in Romania:\\n\\n1. Bucharest - 1,836,000\\n2. Cluj-Napoca - 335,000\\n3. Timișoara - 317,000\\n4. Iași - 290,000\\n5. Craiova - 269,000\\n6. Constanța - 259,000\\n7. Galați - 249,000\\n8. Brașov - 227,000\\n9. Ploiești - 203,000\\n10. Oradea - 200,000\\n11. Arad - 157,000\\n12. Sibiu - 147,000\\n13. Bacău - 144,000\\n14. Baia Mare - 115,000\\n15. Buzău - 111,000\\n16. Focșani - 93,000\\n17. Suceava - 86,000\\n18. Târgu Mureș - 82,000\\n19. Bistrița - 75,000\\n20. Satu Mare - 68,000'}])"
      ]
     },
     "execution_count": 16,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# load from a file responses, completions, messages\n",
    "responses, completions, messages = my_chatbot.load_data()\n",
    "responses, completions, messages"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 17,
   "id": "abe8470f",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "[{'role': 'user', 'content': 'Romania cities'},\n",
       " {'role': 'assistant',\n",
       "  'content': '1. Bucharest\\n2. Cluj-Napoca\\n3. Timișoara\\n4. Iași\\n5. Craiova\\n6. Constanța\\n7. Galați\\n8. Brașov\\n9. Ploiești\\n10. Oradea\\n11. Arad\\n12. Sibiu\\n13. Bacău\\n14. Baia Mare\\n15. Buzău\\n16. Focșani\\n17. Suceava\\n18. Târgu Mureș\\n19. Bistrița\\n20. Satu Mare.'},\n",
       " {'role': 'user', 'content': 'population in above cities'},\n",
       " {'role': 'assistant',\n",
       "  'content': 'Here are the estimated populations (as of 2021) for the above cities in Romania:\\n\\n1. Bucharest - 1,836,000\\n2. Cluj-Napoca - 335,000\\n3. Timișoara - 317,000\\n4. Iași - 290,000\\n5. Craiova - 269,000\\n6. Constanța - 259,000\\n7. Galați - 249,000\\n8. Brașov - 227,000\\n9. Ploiești - 203,000\\n10. Oradea - 200,000\\n11. Arad - 157,000\\n12. Sibiu - 147,000\\n13. Bacău - 144,000\\n14. Baia Mare - 115,000\\n15. Buzău - 111,000\\n16. Focșani - 93,000\\n17. Suceava - 86,000\\n18. Târgu Mureș - 82,000\\n19. Bistrița - 75,000\\n20. Satu Mare - 68,000'}]"
      ]
     },
     "execution_count": 17,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "messages"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 18,
   "id": "ed15a9cc",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "'chatbot_data.json'"
      ]
     },
     "execution_count": 18,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# default file name\n",
    "my_chatbot._file_default"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 19,
   "id": "97eb5ee3",
   "metadata": {},
   "outputs": [],
   "source": [
    "# load messages from default file\n",
    "my_chatbot.load_messages(messages)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 20,
   "id": "7d1c0548",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "[{'role': 'user', 'content': 'Romania cities'},\n",
       " {'role': 'assistant',\n",
       "  'content': '1. Bucharest\\n2. Cluj-Napoca\\n3. Timișoara\\n4. Iași\\n5. Craiova\\n6. Constanța\\n7. Galați\\n8. Brașov\\n9. Ploiești\\n10. Oradea\\n11. Arad\\n12. Sibiu\\n13. Bacău\\n14. Baia Mare\\n15. Buzău\\n16. Focșani\\n17. Suceava\\n18. Târgu Mureș\\n19. Bistrița\\n20. Satu Mare.'},\n",
       " {'role': 'user', 'content': 'population in above cities'},\n",
       " {'role': 'assistant',\n",
       "  'content': 'Here are the estimated populations (as of 2021) for the above cities in Romania:\\n\\n1. Bucharest - 1,836,000\\n2. Cluj-Napoca - 335,000\\n3. Timișoara - 317,000\\n4. Iași - 290,000\\n5. Craiova - 269,000\\n6. Constanța - 259,000\\n7. Galați - 249,000\\n8. Brașov - 227,000\\n9. Ploiești - 203,000\\n10. Oradea - 200,000\\n11. Arad - 157,000\\n12. Sibiu - 147,000\\n13. Bacău - 144,000\\n14. Baia Mare - 115,000\\n15. Buzău - 111,000\\n16. Focșani - 93,000\\n17. Suceava - 86,000\\n18. Târgu Mureș - 82,000\\n19. Bistrița - 75,000\\n20. Satu Mare - 68,000'}]"
      ]
     },
     "execution_count": 20,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# show messsages\n",
    "my_chatbot.messages"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 21,
   "id": "db58d22e",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Here are some of the notable monuments in the above mentioned cities in Romania:\n",
      "\n",
      "1. Bucharest: Palace of the Parliament, Romanian Athenaeum, Triumphal Arch, Stavropoleos Monastery\n",
      "2. Cluj-Napoca: St. Michael's Church, Matthias Corvinus Monument, National History Museum of Transylvania, Banffy Palace\n",
      "3. Timișoara: Huniade Castle, Millennium Church, Victory Square, Timișoara Orthodox Cathedral\n",
      "4. Iași: Palace of Culture, Golia Monastery, Three Hierarchs Monastery, Union Square\n",
      "5. Craiova: Nicolae Romanescu Park, Saint Demetrius Cathedral, Craiova Art Museum, Alexandru Ioan Cuza Park\n",
      "6. Constanța: The Roman Mosaic, Carol I Mosque, The Casino, Genoese Lighthouse.\n",
      "7. Galați: Natural Sciences Museum, Mihai Eminescu Park, Galati Public Garden, The Sulina channel lighthouse \n",
      "8. Brașov: Black Church, Mount Tâmpa, The White Tower, The Rope Street\n",
      "9. Ploiești: Oil Museum, Clock Museum, Memorial House of Nichita Stănescu, Art Museum Ploiești.\n",
      "10. Oradea: Unirii Square, The Black Eagle Palace, Oradea Fortress, The Moon Church\n",
      "11. Arad: Holy Trinity Cathedral, Ioan Slavici Theatre, Palace of Culture, Reconciliation Park\n",
      "12. Sibiu: The Council Tower, The Brukenthal Palace, The Bridge of Lies, The Lutheran Evangelical Cathedral\n",
      "13. Bacău: Stephen the Great Monument, Bacau Natural Sciences Museum, Saint John the Baptist Church, Mihail Sadoveanu Memorial House\n",
      "14. Baia Mare: Merry Cemetery of Saschiz, Baia Mare Ethnographic Museum, The Baia Mare Art Museum, Baia Mare City Hall.\n",
      "15. Buzău: The Communal Palace, Saint Eligius Church, The Municipal Park, The Waterfall of Lopata\n",
      "16. Focșani: Focșani History Museum, Saint Nicholas Church, The Heroes Memorial, Zoological Park Focșani\n",
      "17. Suceava: The Fortress of Suceava, The Painted Monasteries of Bucovina, Stefan cel Mare Statue, The History Museum of Bucovina\n",
      "18. Târgu Mureș: Culture Palace, Teleki-Bolyai Library, Piarist Church, Reformed Church \n",
      "19. Bistrița: The Lutheran Church, Bistrița Museum Complex, The Main Square, The Mill.\n",
      "20. Satu Mare: Satu Mare Roman-Catholic Cathedral, The Decebal Victory Monument, The Firemen’s Tower, The Terry Guard Tower.\n",
      "CPU times: total: 15.6 ms\n",
      "Wall time: 2min 10s\n"
     ]
    }
   ],
   "source": [
    "%%time\n",
    "# continue with a follow up question\n",
    "my_chatbot.message_post(\"monuments in above cities\")\n",
    "response = my_chatbot.generate_response()\n",
    "print(response)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 22,
   "id": "65328836",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "role: user\n",
      "content:\n",
      "Romania cities\n",
      "\n",
      "\n",
      "role: assistant\n",
      "content:\n",
      "1. Bucharest\n",
      "2. Cluj-Napoca\n",
      "3. Timișoara\n",
      "4. Iași\n",
      "5. Craiova\n",
      "6. Constanța\n",
      "7. Galați\n",
      "8. Brașov\n",
      "9. Ploiești\n",
      "10. Oradea\n",
      "11. Arad\n",
      "12. Sibiu\n",
      "13. Bacău\n",
      "14. Baia Mare\n",
      "15. Buzău\n",
      "16. Focșani\n",
      "17. Suceava\n",
      "18. Târgu Mureș\n",
      "19. Bistrița\n",
      "20. Satu Mare.\n",
      "\n",
      "\n",
      "role: user\n",
      "content:\n",
      "population in above cities\n",
      "\n",
      "\n",
      "role: assistant\n",
      "content:\n",
      "Here are the estimated populations (as of 2021) for the above cities in Romania:\n",
      "\n",
      "1. Bucharest - 1,836,000\n",
      "2. Cluj-Napoca - 335,000\n",
      "3. Timișoara - 317,000\n",
      "4. Iași - 290,000\n",
      "5. Craiova - 269,000\n",
      "6. Constanța - 259,000\n",
      "7. Galați - 249,000\n",
      "8. Brașov - 227,000\n",
      "9. Ploiești - 203,000\n",
      "10. Oradea - 200,000\n",
      "11. Arad - 157,000\n",
      "12. Sibiu - 147,000\n",
      "13. Bacău - 144,000\n",
      "14. Baia Mare - 115,000\n",
      "15. Buzău - 111,000\n",
      "16. Focșani - 93,000\n",
      "17. Suceava - 86,000\n",
      "18. Târgu Mureș - 82,000\n",
      "19. Bistrița - 75,000\n",
      "20. Satu Mare - 68,000\n",
      "\n",
      "\n",
      "role: user\n",
      "content:\n",
      "monuments in above cities\n",
      "\n",
      "\n",
      "role: assistant\n",
      "content:\n",
      "Here are some of the notable monuments in the above mentioned cities in Romania:\n",
      "\n",
      "1. Bucharest: Palace of the Parliament, Romanian Athenaeum, Triumphal Arch, Stavropoleos Monastery\n",
      "2. Cluj-Napoca: St. Michael's Church, Matthias Corvinus Monument, National History Museum of Transylvania, Banffy Palace\n",
      "3. Timișoara: Huniade Castle, Millennium Church, Victory Square, Timișoara Orthodox Cathedral\n",
      "4. Iași: Palace of Culture, Golia Monastery, Three Hierarchs Monastery, Union Square\n",
      "5. Craiova: Nicolae Romanescu Park, Saint Demetrius Cathedral, Craiova Art Museum, Alexandru Ioan Cuza Park\n",
      "6. Constanța: The Roman Mosaic, Carol I Mosque, The Casino, Genoese Lighthouse.\n",
      "7. Galați: Natural Sciences Museum, Mihai Eminescu Park, Galati Public Garden, The Sulina channel lighthouse \n",
      "8. Brașov: Black Church, Mount Tâmpa, The White Tower, The Rope Street\n",
      "9. Ploiești: Oil Museum, Clock Museum, Memorial House of Nichita Stănescu, Art Museum Ploiești.\n",
      "10. Oradea: Unirii Square, The Black Eagle Palace, Oradea Fortress, The Moon Church\n",
      "11. Arad: Holy Trinity Cathedral, Ioan Slavici Theatre, Palace of Culture, Reconciliation Park\n",
      "12. Sibiu: The Council Tower, The Brukenthal Palace, The Bridge of Lies, The Lutheran Evangelical Cathedral\n",
      "13. Bacău: Stephen the Great Monument, Bacau Natural Sciences Museum, Saint John the Baptist Church, Mihail Sadoveanu Memorial House\n",
      "14. Baia Mare: Merry Cemetery of Saschiz, Baia Mare Ethnographic Museum, The Baia Mare Art Museum, Baia Mare City Hall.\n",
      "15. Buzău: The Communal Palace, Saint Eligius Church, The Municipal Park, The Waterfall of Lopata\n",
      "16. Focșani: Focșani History Museum, Saint Nicholas Church, The Heroes Memorial, Zoological Park Focșani\n",
      "17. Suceava: The Fortress of Suceava, The Painted Monasteries of Bucovina, Stefan cel Mare Statue, The History Museum of Bucovina\n",
      "18. Târgu Mureș: Culture Palace, Teleki-Bolyai Library, Piarist Church, Reformed Church \n",
      "19. Bistrița: The Lutheran Church, Bistrița Museum Complex, The Main Square, The Mill.\n",
      "20. Satu Mare: Satu Mare Roman-Catholic Cathedral, The Decebal Victory Monument, The Firemen’s Tower, The Terry Guard Tower.\n",
      "\n",
      "\n"
     ]
    }
   ],
   "source": [
    "# list all messages\n",
    "my_chatbot.list_messages()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 23,
   "id": "3d09a689",
   "metadata": {},
   "outputs": [],
   "source": [
    "# save in a new file - responses, completions, messages\n",
    "my_chatbot.save_data(\"chat2.json\")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "9f807ca7",
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3 (ipykernel)",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.8.16"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}