Update app.py

app.py

@@ -44,6 +44,7 @@ from langchain.chains.question_answering import load_qa_chain
 # os.environ["TMDB_BEARER_TOKEN"] = ""
 
 news_api_key = os.environ["NEWS_API_KEY"]
+openai_api_key = os.environ["OPENAI_API_KEY"]
 
 TOOLS_LIST = ['serpapi', 'wolfram-alpha', 'pal-math', 'pal-colored-objects', 'news-api']  #'google-search','news-api','tmdb-api','open-meteo-api'
 TOOLS_DEFAULT_LIST = ['serpapi', 'wolfram-alpha', 'pal-math', 'pal-colored-objects', 'news-api']
@@ -261,12 +262,12 @@ def load_chain(tools_list, llm):
 
 # The rest of the code remains the same.
 
-def set_openai_api_key(api_key):
+def set_openai_api_key():
     """Set the api key and return chain.
     If no api_key, then None is returned.
     """
+    api_key = os.environ.get("OPENAI_API_KEY")
     if api_key and api_key.startswith("sk-") and len(api_key) > 50:
-        os.environ["OPENAI_API_KEY"] = api_key
         print("\n\n ++++++++++++++ Setting OpenAI API key ++++++++++++++ \n\n")
         print(str(datetime.datetime.now()) + ": Before OpenAI, OPENAI_API_KEY length: " + str(
             len(os.environ["OPENAI_API_KEY"])))
@@ -281,11 +282,11 @@ def set_openai_api_key(api_key):
 
         print(str(datetime.datetime.now()) + ": After load_chain, OPENAI_API_KEY length: " + str(
             len(os.environ["OPENAI_API_KEY"])))
-        os.environ["OPENAI_API_KEY"] = ""
         return chain, express_chain, llm, embeddings, qa_chain, memory
     return None, None, None, None, None, None
 
 
+
 def run_chain(chain, inp, capture_hidden_text):
     output = ""
     hidden_text = None
@@ -855,33 +856,22 @@ with gr.Blocks(css=".gradio-container {background-color: lightgray}") as block:
         Powered by <a href='https://github.com/hwchase17/langchain'>LangChain 🦜️🔗</a>
         </center>""")
 
-    message.submit(chat, inputs=[openai_api_key_textbox, message, history_state, chain_state, trace_chain_state,
+    message.submit(chat, inputs=[message, history_state, chain_state, trace_chain_state,
                              speak_text_state, talking_head_state, monologue_state,
                              express_chain_state, num_words_state, formality_state,
                              anticipation_level_state, joy_level_state, trust_level_state, fear_level_state,
                              surprise_level_state, sadness_level_state, disgust_level_state, anger_level_state,
                              lang_level_state, translate_to_state, literary_style_state,
                              qa_chain_state, docsearch_state, use_embeddings_state],
-               outputs=[chatbot, history_state, video_html, my_file, audio_html, tmp_aud_file, message, openai_api_key_textbox])
+               outputs=[chatbot, history_state, video_html, my_file, audio_html, tmp_aud_file, message])
 
-    submit.click(chat, inputs=[openai_api_key_textbox, message, history_state, chain_state, trace_chain_state,
+    submit.click(chat, inputs=[message, history_state, chain_state, trace_chain_state,
                            speak_text_state, talking_head_state, monologue_state,
                            express_chain_state, num_words_state, formality_state,
                            anticipation_level_state, joy_level_state, trust_level_state, fear_level_state,
                            surprise_level_state, sadness_level_state, disgust_level_state, anger_level_state,
                            lang_level_state, translate_to_state, literary_style_state,
                            qa_chain_state, docsearch_state, use_embeddings_state],
-             outputs=[chatbot, history_state, video_html, my_file, audio_html, tmp_aud_file, message, openai_api_key_textbox])
-
-    # outputs=[chatbot, history_state, audio_html, tmp_aud_file, message])
-
-    
-
-    openai_api_key_textbox.change(set_openai_api_key,
-                                  inputs=[openai_api_key_textbox],
-                                  outputs=[chain_state, express_chain_state, llm_state, embeddings_state,
-                                           qa_chain_state, memory_state])
-    
-   
+             outputs=[chatbot, history_state, video_html, my_file, audio_html, tmp_aud_file, message])
 
 block.launch(debug=True)