app.py

from PyPDF2 import PdfReader
from langchain.embeddings.openai import OpenAIEmbeddings
from langchain.text_splitter import CharacterTextSplitter
from langchain.vectorstores import ElasticVectorSearch, Pinecone, Weaviate, FAISS
from langchain.chains.question_answering import load_qa_chain
from langchain.llms import OpenAI
import gradio as gr

import os



embeddings = OpenAIEmbeddings(openai_api_key=os.environ['OPENAI_API_KEY'])
docsearch = FAISS.load_local(folder_path = 'chanakyaNeeti',embeddings=embeddings)

chain = load_qa_chain(OpenAI(openai_api_key=os.environ['OPENAI_API_KEY']), chain_type="stuff")

def learn_from_chanakya_neeti(chat_history,query):
  docs = docsearch.similarity_search(query)
  result = chain.run(input_documents=docs, question=query)
  return chat_history + [(query,result)]

with gr.Blocks() as demo:
  chatbot = gr.Chatbot()
  textbox = gr.Textbox(label="Your query")

  textbox.submit(fn=learn_from_chanakya_neeti,inputs=[chatbot,textbox],outputs=[chatbot])


demo.launch(debug=True)