Add prompt for own openai api key

app.py

@@ -11,13 +11,19 @@ with open("guide1.txt") as f:
 text_splitter = CharacterTextSplitter(chunk_size=1000, chunk_overlap=0, separator = "\n")
 texts = text_splitter.split_text(hitchhikersguide)
 
-embeddings = OpenAIEmbeddings()
+chain = None
+embeddings = None
+docsearch = None
 
-docsearch = Chroma.from_texts(texts, embeddings, metadatas=[{"source": str(i)} for i in range(len(texts))]).as_retriever()
 
-chain = load_qa_chain(OpenAI(temperature=0), chain_type="stuff")
+def make_inference(openai_api_key, query):
+    global chain, embeddings, docsearch
+
+    if chain is None:
+        chain = load_qa_chain(OpenAI(temperature=0, openai_api_key=openai_api_key.strip()), chain_type="stuff")
+        embeddings = OpenAIEmbeddings(openai_api_key=openai_api_key)
+        docsearch = Chroma.from_texts(texts, embeddings, metadatas=[{"source": str(i)} for i in range(len(texts))]).as_retriever()
 
-def make_inference(query):
     docs = docsearch.get_relevant_documents(query)
     return(chain.run(input_documents=docs, question=query))
 
@@ -28,9 +34,10 @@ if __name__ == "__main__":
     gr.Interface(
         make_inference,
         [
+            gr.inputs.Textbox(lines=1, label="OpenAI API Key"),
             gr.inputs.Textbox(lines=2, label="Query"),
         ],
         gr.outputs.Textbox(label="Response"),
         title="🗣️TalkToMyDoc📄",
         description="🗣️TalkToMyDoc📄 is a tool that allows you to ask questions about a document. In this case - Hitch Hitchhiker's Guide to the Galaxy.",
-    ).launch()
+    ).launch()