Update app.py

app.py

@@ -1,4 +1,5 @@
 import gradio as gr
+import os
 
 from langchain.document_loaders import OnlinePDFLoader
 
@@ -17,18 +18,22 @@ from langchain.chains import ConversationalRetrievalChain
 def loading_pdf():
     return "Loading..."
 
-def pdf_changes(pdf_doc):
-    loader = OnlinePDFLoader(pdf_doc.name)
-    documents = loader.load()
-    texts = text_splitter.split_documents(documents)
-    db = Chroma.from_documents(texts, embeddings)
-    retriever = db.as_retriever()
-    global qa 
-    qa = ConversationalRetrievalChain.from_llm(
-        llm=OpenAI(temperature=0.5), 
-        retriever=retriever, 
-        return_source_documents=False)
-    return "Ready"
+def pdf_changes(pdf_doc, open_ai_key):
+    if openai_key is not None:
+        os.environ['OPENAI_API_KEY'] = open_ai_key
+        loader = OnlinePDFLoader(pdf_doc.name)
+        documents = loader.load()
+        texts = text_splitter.split_documents(documents)
+        db = Chroma.from_documents(texts, embeddings)
+        retriever = db.as_retriever()
+        global qa 
+        qa = ConversationalRetrievalChain.from_llm(
+            llm=OpenAI(temperature=0.5), 
+            retriever=retriever, 
+            return_source_documents=False)
+        return "Ready"
+    else:
+        return "You forgot OpenAI API key"
 
 def add_text(history, text):
     history = history + [(text, None)]
@@ -40,18 +45,17 @@ def bot(history):
     return history
 
 def infer(question, history):
-    print(history[-1])
     
     res = []
     for human, ai in history[:-1]:
-        pair = f"Human:{human}\nAI:{ai}"
+        pair = (human, ai)
         res.append(pair)
     res = ["\n".join(res)]
     chat_history = res
-    print(chat_history)
+    #print(chat_history)
     query = question
     result = qa({"question": query, "chat_history": chat_history})
-    print(result)
+    #print(result)
     return result["answer"]
 
 css="""
@@ -72,6 +76,7 @@ with gr.Blocks(css=css) as demo:
         gr.HTML(title)
         
         with gr.Column():
+            openai_key = gr.Textbox(label="You OpenAI API key", type="password")
             pdf_doc = gr.File(label="Load a pdf", file_types=['.pdf'], type="file")
             with gr.Row():
                 langchain_status = gr.Textbox(label="Status", placeholder="", interactive=False)
@@ -81,7 +86,7 @@ with gr.Blocks(css=css) as demo:
         with gr.Row():
             question = gr.Textbox(label="Question", placeholder="Type your question and hit Enter ")
     load_pdf.click(loading_pdf, None, langchain_status, queue=False)    
-    load_pdf.click(pdf_changes, pdf_doc, langchain_status, queue=False)
+    load_pdf.click(pdf_changes, inputs=[pdf_doc, openai_key], outputs=[langchain_status], queue=False)
     question.submit(add_text, [chatbot, question], [chatbot, question]).then(
         bot, chatbot, chatbot
     )