Update app.py

app.py

@@ -13,6 +13,9 @@ import initialize
 from langchain_openai import ChatOpenAI
 from langchain.memory import ConversationBufferMemory
 from langchain.chains import ConversationalRetrievalChain
+from langchain.chains import VectorDBQA
+from langchain.llms import OpenAI
+
 
 import gradio as gr
 import os
@@ -49,16 +52,21 @@ def chat_query(question):
     
     llm = ChatOpenAI(model=llm_name, temperature=0.1, api_key = OPENAI_API_KEY)
 
-    # Memory
+    # Conversation Retrival Chain with Memory
     memory = ConversationBufferMemory(memory_key="chat_history", return_messages=True)
-
-    # Conversation Retrival Chain
     retriever=vectordb.as_retriever()
-    qa = ConversationalRetrievalChain.from_llm(llm, retriever=retriever, memory=memory)
+    #qa = ConversationalRetrievalChain.from_llm(llm, retriever=retriever, memory=memory)
 
     # Replace input() with question variable for Gradio
-    result = qa({"question": question})
-    return result['answer']
+    # result = qa({"question": question})
+    # return result['answer']
+
+    # Chatbot only answers based on Documents
+    qa = VectorDBQA.from_chain_type(llm=OpenAI(openai_api_key = OPENAI_API_KEY, ), chain_type="stuff", vectorstore=vectordb)
+    result = qa.query(question)
+    return result
+
+    
 
 
 # logo_path = os.path.join(os.getcwd(), "Logo.png")