Update app.py

app.py

@@ -22,13 +22,35 @@ from langchain.embeddings.openai import OpenAIEmbeddings
 
 embeddings = OpenAIEmbeddings()
 
+from langchain.vectorstores import Chroma
 
+persist_directory = "vector_db"
 
+vectordb = Chroma.from_documents(documents=documents, embedding=embeddings, persist_directory=persist_directory)
 
-import gradio as gr
+vectordb = Chroma(persist_directory=persist_directory, embedding_function=embeddings)
 
-def greet(name):
-    return "Hello " + name + "!!"
+from langchain.chat_models import ChatOpenAI
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
-iface.launch()
+#llm = ChatOpenAI(temperature=0, model="gpt-4")
+llm = ChatOpenAI(temperature=0, model="gpt-3.5-turbo")
+
+doc_retriever = vectordb.as_retriever()
+
+from langchain.chains import RetrievalQA
+
+shakespeare_qa = RetrievalQA.from_chain_type(llm=llm, chain_type="stuff", retriever=doc_retriever)
+
+
+if __name__ == "__main__":
+    # make a gradio interface
+    import gradio as gr
+
+    gr.Interface(
+        shakespeare_qa,
+        [
+            gr.inputs.Textbox(lines=2, label="Question"),
+        ],
+        gr.outputs.Textbox(label="Response"),
+        title="ShakesQA",
+        description="ShakesQA",    ).launch()