Gradio chat working with indexes drop down

app.ipynb

@@ -79,51 +79,56 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 11,
+   "execution_count": 65,
    "metadata": {},
    "outputs": [
     {
      "name": "stdout",
      "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"
+      "Using vector store: Roman History\n",
+      "Using index: rag-demo-1-history-rome\n"
      ]
     }
    ],
    "source": [
     "embeddings = OpenAIEmbeddings()\n",
     "\n",
-    "course = \"Roman History\"\n",
-    "if course == \"Roman History\":\n",
-    "    print(f\"Using course: {course}\")\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",
+    "vector_store = None\n",
+    "\n",
+    "def create_vector_store(indexToUse: str):\n",
     "    \n",
-    "pinecone.init(api_key=os.environ[apiKeyKey], environment=os.environ[apiEnvKey])\n",
+    "    if indexToUse == \"Roman History\":\n",
+    "        print(f\"Using vector store: {indexToUse}\")\n",
+    "        apiKeyKey = \"PINECONE_API_KEY_2\"\n",
+    "        apiEnvKey = \"PINECONE_API_ENV_2\"\n",
+    "        testQuestion = \"When was Ceasar born?\"\n",
+    "    else:\n",
+    "        print(f\"Using vector store: {indexToUse}\")\n",
+    "        apiKeyKey = \"PINECONE_API_KEY\"\n",
+    "        apiEnvKey = \"PINECONE_API_ENV\"\n",
+    "        testQuestion = \"What is Moby Dick?\"\n",
+    "        \n",
+    "    pinecone.init(api_key=os.environ[apiKeyKey], environment=os.environ[apiEnvKey])\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",
-    "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",
+    "    # query = testQuestion\n",
+    "    # print(f'Test questions: {testQuestion}')\n",
+    "    # result = vector_store.similarity_search(query, k=3)\n",
+    "    # print(result[0])\n",
+    "    \n",
+    "    return vector_store\n",
     "\n",
-    "query = testQuestion\n",
-    "print(f'Test questions: {testQuestion}')\n",
-    "result = vector_store.similarity_search(query, k=3)\n",
-    "print(result[0])"
+    "vector_store = create_vector_store(\"Roman History\")"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": 12,
+   "execution_count": 66,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -139,18 +144,9 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 13,
+   "execution_count": 67,
    "metadata": {},
-   "outputs": [
-    {
-     "name": "stderr",
-     "output_type": "stream",
-     "text": [
-      "c:\\sc\\ai\\rag-demo-1\\.venv\\lib\\site-packages\\gradio\\components\\button.py:89: UserWarning: Using the update method is deprecated. Simply return a new object instead, e.g. `return gr.Button(...)` instead of `return gr.Button.update(...)`.\n",
-      "  warnings.warn(\n"
-     ]
-    }
-   ],
+   "outputs": [],
    "source": [
     "from langchain.memory import ConversationBufferMemory\n",
     "from langchain.chains import ConversationalRetrievalChain\n",
@@ -161,7 +157,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 6,
+   "execution_count": 68,
    "metadata": {},
    "outputs": [
     {
@@ -170,7 +166,7 @@
        "' \\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": 6,
+     "execution_count": 68,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -201,7 +197,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 49,
+   "execution_count": 109,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -213,8 +209,11 @@
     "from langchain.memory import ConversationBufferMemory\n",
     "import pinecone\n",
     "\n",
-    "def create_conversation(query: str, chat_history: list) -> tuple:\n",
+    "def create_conversation(query: str, chat_history: list, indexToUse: str) -> tuple: \n",
     "    try:\n",
+    "        #vector_store = create_vector_store(\"Roman History\")\n",
+    "        print(indexToUse)\n",
+    "        vector_store = create_vector_store(\"Literature\")\n",
     "        memory = ConversationBufferMemory(\n",
     "            memory_key='chat_history',\n",
     "            return_messages=False\n",
@@ -228,7 +227,7 @@
     "        )\n",
     "        result = cqa({'question': query, 'chat_history': chat_history})\n",
     "        chat_history.append((query, result['answer']))\n",
-    "        return '', chat_history\n",
+    "        return '', chat_history, indexToUse\n",
     "    except Exception as e:\n",
     "        chat_history.append((query, e))\n",
     "        return '', chat_history"
@@ -236,14 +235,30 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 53,
+   "execution_count": 112,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "\n",
+    "\n",
+    "# vector_store_index_to_use = None \n",
+    "# def set_vector_store_index(indexName:str):\n",
+    "#     vector_store_index_to_use = indexName\n",
+    "    \n",
+    "# with gr.Blocks() as demo:\n",
+    "#     indexToUseDD = gr.Dropdown(choices=[\"Roman History\", \"Literature\"])"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 111,
    "metadata": {},
    "outputs": [
     {
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "Running on local URL:  http://127.0.0.1:7886\n",
+      "Running on local URL:  http://127.0.0.1:7917\n",
       "\n",
       "To create a public link, set `share=True` in `launch()`.\n"
      ]
@@ -251,7 +266,7 @@
     {
      "data": {
       "text/html": [
-       "<div><iframe src=\"http://127.0.0.1:7886/\" width=\"100%\" height=\"500\" allow=\"autoplay; camera; microphone; clipboard-read; clipboard-write;\" frameborder=\"0\" allowfullscreen></iframe></div>"
+       "<div><iframe src=\"http://127.0.0.1:7917/\" width=\"100%\" height=\"500\" allow=\"autoplay; camera; microphone; clipboard-read; clipboard-write;\" frameborder=\"0\" allowfullscreen></iframe></div>"
       ],
       "text/plain": [
        "<IPython.core.display.HTML object>"
@@ -266,14 +281,15 @@
     "\n",
     "with gr.Blocks() as demo:\n",
     "    gr.Markdown(\"Chat to docs\")\n",
+    "    indexToUseDD = gr.Dropdown(choices=[\"Roman History\", \"Literature\"])\n",
     "    chatbot = gr.Chatbot(label='Talk to the Doument')\n",
     "    msg = gr.Textbox()\n",
     "    \n",
     "    submitBtn = gr.Button(value=\"Submit\")    \n",
     "    clear = gr.ClearButton([msg, chatbot])\n",
     "\n",
-    "    msg.submit(create_conversation, [msg, chatbot], [msg, chatbot])\n",
-    "    submitBtn.click(create_conversation, [msg, chatbot], [msg, chatbot])\n",
+    "    msg.submit(create_conversation, [msg, chatbot, indexToUseDD], [msg, chatbot, indexToUseDD])\n",
+    "    submitBtn.click(create_conversation, [msg, chatbot, indexToUseDD], [msg, chatbot, indexToUseDD])\n",
     "\n",
     "if __name__ == '__main__':\n",
     "    demo.launch(show_error=True) #show_error=True, debug=True)"