Update app.py

app.py

@@ -1,5 +1,4 @@
 from langchain.prompts.prompt import PromptTemplate
-# from langchain.llms import OpenAI
 from langchain.llms import OpenAIChat
 from langchain.chains import ChatVectorDBChain
 from langchain.embeddings import HuggingFaceEmbeddings, HuggingFaceInstructEmbeddings
@@ -76,17 +75,9 @@ def get_chain(vectorstore):
         condense_question_prompt=CONDENSE_QUESTION_PROMPT,
     )
     return qa_chain
-
-def set_openai_api_key(api_key: str):
-    """Set the api key and return chain.
-    If no api_key, then None is returned.
-    """
-    if api_key:
-        os.environ["OPENAI_API_KEY"] = api_key
-        chain = get_chain(vectorstore)
-        os.environ["OPENAI_API_KEY"] = ""
-        return chain
-
+    
+chain = get_chain(vectorstore)
+    
 class ChatWrapper:
 
     def __init__(self):
@@ -98,15 +89,9 @@ class ChatWrapper:
         self.lock.acquire()
         try:
             history = history or []
-            # If chain is None, that is because no API key was provided.
-            # if chain is None:
-            #     history.append((inp, "Please paste your OpenAI key to use"))
-            #     return history, history
             # Set OpenAI key
-            import openai
-            openai.api_key = os.environ["OPENAI_API_KEY"]
+            
             # Run chain and append input.
-            chain = get_chain(vectorstore)
             output = chain({"question": inp, "chat_history": history})["answer"]
             history.append((inp, output))
         except Exception as e:
@@ -125,11 +110,13 @@ with block:
 
         openai_api_key_textbox = gr.Textbox(
             placeholder="Paste your OpenAI API key (sk-...)",
+            value=os.environ["OPENAI_API_KEY"],
             show_label=False,
             lines=1,
             type="password",
         )
 
+
     with gr.Row():
         embeddings = gr.Radio(choices=model_options_list,value=model_options_list[0], label='Choose your Embedding Model',
                              interactive=True)
@@ -167,14 +154,8 @@ with block:
     agent_state = gr.State()
 
     
-    submit.click(chat, inputs=[openai_api_key_textbox, message, state, agent_state], outputs=[chatbot, state])
-    message.submit(chat, inputs=[openai_api_key_textbox, message, state, agent_state], outputs=[chatbot, state])
-
-    openai_api_key_textbox.change(
-        set_openai_api_key,
-        inputs=[openai_api_key_textbox],
-        outputs=[agent_state],
-    )
+    submit.click(chat, inputs=[message, state, agent_state], outputs=[chatbot, state])
+    message.submit(chat, inputs=[message, state, agent_state], outputs=[chatbot, state])
 
     gr.Markdown("![visitor badge](https://visitor-badge.glitch.me/badge?page_id=nickmuchi-investor-chatchain)")