Update app.py

app.py

@@ -16,27 +16,26 @@ from langchain.vectorstores import Chroma
 
 from langchain.chains import ConversationalRetrievalChain
 
+os.environ['OPENAI_API_KEY'] = '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'
+
 def loading_pdf():
     return "Loading..."
 
-def pdf_changes(open_ai_key):
-    if open_ai_key is not None:
-        os.environ['OPENAI_API_KEY'] = open_ai_key
-        loader = PyPDFLoader("He Yingxu_2806.pdf")
-        documents = loader.load()
-        text_splitter = CharacterTextSplitter(chunk_size=1000, chunk_overlap=0)
-        texts = text_splitter.split_documents(documents)
-        embeddings = OpenAIEmbeddings()
-        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 pdf_changes():
+        
+    loader = PyPDFLoader("He Yingxu_2806.pdf")
+    documents = loader.load()
+    text_splitter = CharacterTextSplitter(chunk_size=1000, chunk_overlap=0)
+    texts = text_splitter.split_documents(documents)
+    embeddings = OpenAIEmbeddings()
+    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 add_text(history, text):
     history = history + [(text, None)]
@@ -85,7 +84,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")
+            # 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)
@@ -95,7 +94,7 @@ with gr.Blocks(css=css) as demo:
         question = gr.Textbox(label="Question", placeholder="Type your question and hit Enter ")
         submit_btn = gr.Button("Send Message")
     load_pdf.click(loading_pdf, None, langchain_status, queue=False)    
-    load_pdf.click(pdf_changes, inputs=[openai_key], outputs=[langchain_status], queue=False)
+    load_pdf.click(pdf_changes, inputs=[], outputs=[langchain_status], queue=False)
     question.submit(add_text, [chatbot, question], [chatbot, question]).then(
         bot, chatbot, chatbot
     )