alpha commit

app.py

@@ -1,6 +1,22 @@
+'''
+Om Sri Sai Ram
+
+Swami's Chatbot Alpha Version
+'''
+
+from langchain.vectorstores import FAISS
+from langchain.chains.question_answering import load_qa_chain
+from langchain.embeddings.openai import OpenAIEmbeddings
+from langchain.llms import OpenAI
 import gradio as gr
-import random
 import time
+import os
+
+os.environ['OPENAI_API_KEY'] = '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'
+
+vectordb = FAISS.load_local("faiss_index OPENAI", OpenAIEmbeddings())
+
+chain = load_qa_chain(OpenAI(), chain_type = 'stuff')
 
 with gr.Blocks() as demo:
     chatbot = gr.Chatbot()
@@ -8,9 +24,8 @@ with gr.Blocks() as demo:
     clear = gr.ClearButton([msg, chatbot])
 
     def respond(message, chat_history):
-        bot_message = random.choice(["How are you?", "I love you", "I'm very hungry"])
-        chat_history.append((message, bot_message))
-        time.sleep(2)
+        docs = vectordb.similarity_search(message)
+        chat_history.append((message, chain.run(input_documents = docs, question = message)))
         return "", chat_history
 
     msg.submit(respond, [msg, chatbot], [msg, chatbot])

requirements.txt

@@ -0,0 +1,3 @@
+langchain
+gradio
+openai