Setting up UI for chat

app.ipynb

@@ -78,7 +78,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 9,
+   "execution_count": 30,
    "metadata": {},
    "outputs": [
     {
@@ -86,6 +86,8 @@
      "output_type": "stream",
      "text": [
       "Using course: Roman History\n",
+      "Using index: rag-demo-1-history-rome\n",
+      "Test questions: When was Ceasar born?\n",
       "page_content='The situation in Rome. While Pompey was adding to his\\nmilitary reputation in the East he was regarded with jealous\\nand anxious eyes not only by the Senate but also by the other\\nchampions of the popular party, Crassus who found his wealth\\nno match for Pompeys military achievements, and Caius Julius\\nCaesar who was rapidly coming to be one of the leading figures in\\nRoman public life. Caesar was born in 100 B. C., of the patrician [162]\\ngens of the Julii, but since his aunt was the wife of Marius,\\nand he himself had married the daughter of Cinna, his lot was\\ncast with the Populares. As a young man he had distinguished\\nhimself by refusing to divorce his wife at Sullas behest, whereat\\nSulla was with difficulty induced to spare his life, saying that he\\nsaw in him many a Marius. For the time being Caesar judged it\\nprudent to withdraw from Rome to Rhodes. While in the East\\nhe was captured by pirates, and after being ransomed, fulfilled' metadata={'source': '..\\\\rag-demo-1-data\\\\history-roman\\\\3. A History of Rome to 565 A. D. author Arthur Edward Romilly Boak.pdf.txt'}\n"
      ]
     }
@@ -93,20 +95,29 @@
    "source": [
     "embeddings = OpenAIEmbeddings()\n",
     "\n",
+    "\n",
     "course = \"Roman History\"\n",
     "if course == \"Roman History\":\n",
     "    print(f\"Using course: {course}\")\n",
-    "    pinecone.init(api_key=os.environ[\"PINECONE_API_KEY_2\"], environment=os.environ[\"PINECONE_API_ENV_2\"])\n",
-    "    index = pinecone.Index(\"rag-demo-1-history-rome\")\n",
-    "    vector_store = Pinecone(index, embeddings, \"text\")\n",
+    "    apiKeyKey = \"PINECONE_API_KEY_2\"\n",
+    "    apiEnvKey = \"PINECONE_API_ENV_2\"\n",
+    "    testQuestion = \"When was Ceasar born?\"\n",
+    "else:\n",
+    "    print(f\"Using course: {course}\")\n",
+    "    apiKeyKey = \"PINECONE_API_KEY\"\n",
+    "    apiEnvKey = \"PINECONE_API_ENV\"\n",
+    "    testQuestion = \"What is Moby Dick?\"\n",
     "    \n",
-    "query = \"When was Ceasar born?\"\n",
+    "pinecone.init(api_key=os.environ[apiKeyKey], environment=os.environ[apiEnvKey])\n",
     "\n",
-    "result = vector_store.similarity_search(\n",
-    "    query,  # our search query\n",
-    "    k=1  # return 3 most relevant docs\n",
-    ")\n",
+    "index_name = pinecone.list_indexes()[0]\n",
+    "print(f\"Using index: {index_name}\")\n",
+    "index = pinecone.Index(index_name)\n",
+    "vector_store = Pinecone(index, embeddings, \"text\")    \n",
     "\n",
+    "query = testQuestion\n",
+    "print(f'Test questions: {testQuestion}')\n",
+    "result = vector_store.similarity_search(query, k=3)\n",
     "print(result[0])"
    ]
   },
@@ -127,18 +138,19 @@
     }
    ],
    "source": [
-    "llm = OpenAI(temperature=0)\n",
-    "llm.predict(\"Hello there\")"
+    "llm = OpenAI(temperature=0)"
    ]
   },
   {
    "cell_type": "markdown",
    "metadata": {},
-   "source": []
+   "source": [
+    "# Conversational agent"
+   ]
   },
   {
    "cell_type": "code",
-   "execution_count": 21,
+   "execution_count": 35,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -151,16 +163,16 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 26,
+   "execution_count": 36,
    "metadata": {},
    "outputs": [
     {
      "data": {
       "text/plain": [
-       "\" I don't know.\""
+       "' \\nCrassus was a Roman general who was killed in battle by the Parthians. He was killed while trying to extend the Roman Empire into the Middle East. Ceasar was avenging Crassus by trying to defeat the Parthians and expand the Roman Empire. He was also trying to avenge the death of his friend and mentor.'"
       ]
      },
-     "execution_count": 26,
+     "execution_count": 36,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -174,67 +186,12 @@
     "    The answers should be at least 40 words or longer\n",
     "    \n",
     "    QUESTION:\n",
-    "    What is moby dick?\n",
+    "    Why was he avenging Crassus, what happened to him that Ceasar needed to avenge him? \n",
     "    \n",
     "\"\"\"\n",
     "chain.run({'question': query})"
    ]
   },
-  {
-   "cell_type": "code",
-   "execution_count": 5,
-   "metadata": {},
-   "outputs": [
-    {
-     "ename": "KeyboardInterrupt",
-     "evalue": "Interrupted by user",
-     "output_type": "error",
-     "traceback": [
-      "\u001b[1;31m---------------------------------------------------------------------------\u001b[0m",
-      "\u001b[1;31mKeyboardInterrupt\u001b[0m                         Traceback (most recent call last)",
-      "\u001b[1;32mc:\\sc\\ai\\rag-demo-1\\app.ipynb Cell 5\u001b[0m line \u001b[0;36m4\n\u001b[0;32m      <a href='vscode-notebook-cell:/c%3A/sc/ai/rag-demo-1/app.ipynb#W4sZmlsZQ%3D%3D?line=0'>1</a>\u001b[0m llm \u001b[39m=\u001b[39m OpenAI(temperature\u001b[39m=\u001b[39m\u001b[39m0\u001b[39m)\n\u001b[0;32m      <a href='vscode-notebook-cell:/c%3A/sc/ai/rag-demo-1/app.ipynb#W4sZmlsZQ%3D%3D?line=1'>2</a>\u001b[0m qa \u001b[39m=\u001b[39m RetrievalQA\u001b[39m.\u001b[39mfrom_chain_type(llm\u001b[39m=\u001b[39mllm, chain_type\u001b[39m=\u001b[39m\u001b[39m\"\u001b[39m\u001b[39mstuff\u001b[39m\u001b[39m\"\u001b[39m, retriever\u001b[39m=\u001b[39mvector_store\u001b[39m.\u001b[39mas_retriever())\n\u001b[1;32m----> <a href='vscode-notebook-cell:/c%3A/sc/ai/rag-demo-1/app.ipynb#W4sZmlsZQ%3D%3D?line=3'>4</a>\u001b[0m query \u001b[39m=\u001b[39m \u001b[39minput\u001b[39;49m(\u001b[39m\"\u001b[39;49m\u001b[39mWho was ceasar and why whas he noteable?\u001b[39;49m\u001b[39m\"\u001b[39;49m)\n\u001b[0;32m      <a href='vscode-notebook-cell:/c%3A/sc/ai/rag-demo-1/app.ipynb#W4sZmlsZQ%3D%3D?line=4'>5</a>\u001b[0m answer \u001b[39m=\u001b[39m qa\u001b[39m.\u001b[39mrun(query)\n\u001b[0;32m      <a href='vscode-notebook-cell:/c%3A/sc/ai/rag-demo-1/app.ipynb#W4sZmlsZQ%3D%3D?line=5'>6</a>\u001b[0m \u001b[39m# print(answer)\u001b[39;00m\n\u001b[0;32m      <a href='vscode-notebook-cell:/c%3A/sc/ai/rag-demo-1/app.ipynb#W4sZmlsZQ%3D%3D?line=6'>7</a>\u001b[0m \u001b[39m# print(\"\\nWhat else can I help you with:\")\u001b[39;00m\n\u001b[0;32m      <a href='vscode-notebook-cell:/c%3A/sc/ai/rag-demo-1/app.ipynb#W4sZmlsZQ%3D%3D?line=7'>8</a>\u001b[0m \n\u001b[1;32m   (...)\u001b[0m\n\u001b[0;32m     <a href='vscode-notebook-cell:/c%3A/sc/ai/rag-demo-1/app.ipynb#W4sZmlsZQ%3D%3D?line=11'>12</a>\u001b[0m \u001b[39m#     print(answer)\u001b[39;00m\n\u001b[0;32m     <a href='vscode-notebook-cell:/c%3A/sc/ai/rag-demo-1/app.ipynb#W4sZmlsZQ%3D%3D?line=12'>13</a>\u001b[0m \u001b[39m#     print(\"\\nWhat else can I help you with:\")\u001b[39;00m\n",
-      "File \u001b[1;32mc:\\sc\\ai\\rag-demo-1\\.venv\\lib\\site-packages\\ipykernel\\kernelbase.py:1202\u001b[0m, in \u001b[0;36mKernel.raw_input\u001b[1;34m(self, prompt)\u001b[0m\n\u001b[0;32m   1200\u001b[0m     msg \u001b[39m=\u001b[39m \u001b[39m\"\u001b[39m\u001b[39mraw_input was called, but this frontend does not support input requests.\u001b[39m\u001b[39m\"\u001b[39m\n\u001b[0;32m   1201\u001b[0m     \u001b[39mraise\u001b[39;00m StdinNotImplementedError(msg)\n\u001b[1;32m-> 1202\u001b[0m \u001b[39mreturn\u001b[39;00m \u001b[39mself\u001b[39;49m\u001b[39m.\u001b[39;49m_input_request(\n\u001b[0;32m   1203\u001b[0m     \u001b[39mstr\u001b[39;49m(prompt),\n\u001b[0;32m   1204\u001b[0m     \u001b[39mself\u001b[39;49m\u001b[39m.\u001b[39;49m_parent_ident[\u001b[39m\"\u001b[39;49m\u001b[39mshell\u001b[39;49m\u001b[39m\"\u001b[39;49m],\n\u001b[0;32m   1205\u001b[0m     \u001b[39mself\u001b[39;49m\u001b[39m.\u001b[39;49mget_parent(\u001b[39m\"\u001b[39;49m\u001b[39mshell\u001b[39;49m\u001b[39m\"\u001b[39;49m),\n\u001b[0;32m   1206\u001b[0m     password\u001b[39m=\u001b[39;49m\u001b[39mFalse\u001b[39;49;00m,\n\u001b[0;32m   1207\u001b[0m )\n",
-      "File \u001b[1;32mc:\\sc\\ai\\rag-demo-1\\.venv\\lib\\site-packages\\ipykernel\\kernelbase.py:1245\u001b[0m, in \u001b[0;36mKernel._input_request\u001b[1;34m(self, prompt, ident, parent, password)\u001b[0m\n\u001b[0;32m   1242\u001b[0m \u001b[39mexcept\u001b[39;00m \u001b[39mKeyboardInterrupt\u001b[39;00m:\n\u001b[0;32m   1243\u001b[0m     \u001b[39m# re-raise KeyboardInterrupt, to truncate traceback\u001b[39;00m\n\u001b[0;32m   1244\u001b[0m     msg \u001b[39m=\u001b[39m \u001b[39m\"\u001b[39m\u001b[39mInterrupted by user\u001b[39m\u001b[39m\"\u001b[39m\n\u001b[1;32m-> 1245\u001b[0m     \u001b[39mraise\u001b[39;00m \u001b[39mKeyboardInterrupt\u001b[39;00m(msg) \u001b[39mfrom\u001b[39;00m \u001b[39mNone\u001b[39;00m\n\u001b[0;32m   1246\u001b[0m \u001b[39mexcept\u001b[39;00m \u001b[39mException\u001b[39;00m:\n\u001b[0;32m   1247\u001b[0m     \u001b[39mself\u001b[39m\u001b[39m.\u001b[39mlog\u001b[39m.\u001b[39mwarning(\u001b[39m\"\u001b[39m\u001b[39mInvalid Message:\u001b[39m\u001b[39m\"\u001b[39m, exc_info\u001b[39m=\u001b[39m\u001b[39mTrue\u001b[39;00m)\n",
-      "\u001b[1;31mKeyboardInterrupt\u001b[0m: Interrupted by user"
-     ]
-    }
-   ],
-   "source": [
-    "qa = RetrievalQA.from_chain_type(llm=llm, chain_type=\"stuff\", retriever=vector_store.as_retriever())\n",
-    "\n",
-    "query = input(\"Who was ceasar and why whas he noteable?\")\n",
-    "answer = qa.run(query)\n",
-    "\n",
-    "\n",
-    "qa = RetrievalQA.from_chain_type(llm, chain_type=\"stuff\", retriever=retriever)\n",
-    "response = qa.run(query)\n",
-    "\n",
-    "# print(answer)\n",
-    "# print(\"\\nWhat else can I help you with:\")\n",
-    "\n",
-    "# while True:\n",
-    "#     query = input(\"Who was ceasar and why whas he notable?\")\n",
-    "#     answer = qa.run(query)\n",
-    "#     print(answer)\n",
-    "#     print(\"\\nWhat else can I help you with:\")"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "qa = RetrievalQA.from_chain_type(llm=OpenAI(temperature=0), chain_type=\"stuff\", retriever=vector_store.as_retriever())\n",
-    "\n",
-    "print(\"Connector development help bot. What do you want to know?\")\n",
-    "while True:\n",
-    "    query = input(\"\")\n",
-    "    answer = qa.run(query)\n",
-    "    print(answer)\n",
-    "    print(\"\\nWhat else can I help you with:\")"
-   ]
-  },
   {
    "cell_type": "code",
    "execution_count": 3,