Update app.py

app.py

@@ -44,7 +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"]
+tmdb_bearer_token = os.environ["TMDB_BEARER_TOKEN"]
 
 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']
@@ -252,7 +252,7 @@ def load_chain(tools_list, llm):
     if llm:
         print("\ntools_list", tools_list)
         tool_names = tools_list
-        tools = load_tools(tool_names, llm=llm, news_api_key=news_api_key)
+        tools = load_tools(tool_names, llm=llm, news_api_key=news_api_key, tmdb_bearer_token=tmdb_bearer_token)
 
         memory = ConversationBufferMemory(memory_key="chat_history")
 
@@ -260,31 +260,52 @@ def load_chain(tools_list, llm):
         express_chain = LLMChain(llm=llm, prompt=PROMPT_TEMPLATE, verbose=True)
     return chain, express_chain, memory
 
-# The rest of the code remains the same.
 
-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:
-        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"])))
-        llm = OpenAI(temperature=0, max_tokens=MAX_TOKENS)
-        print(str(datetime.datetime.now()) + ": After OpenAI, OPENAI_API_KEY length: " + str(
-            len(os.environ["OPENAI_API_KEY"])))
-        chain, express_chain, memory = load_chain(TOOLS_DEFAULT_LIST, llm)
+async def set_chain_state_api_key(api_key):
+    # Set the API key for chain_state
+    chain_state.api_key = api_key
+
+async def set_express_chain_state_api_key(api_key):
+    # Set the API key for express_chain_state
+    express_chain_state.api_key = api_key
+
+async def set_llm_state_api_key(api_key):
+    # Set the API key for llm_state
+    llm_state.api_key = api_key
+
+async def set_embeddings_state_api_key(api_key):
+    # Set the API key for embeddings_state
+    embeddings_state.api_key = api_key
 
-        # Pertains to question answering functionality
-        embeddings = OpenAIEmbeddings()
-        qa_chain = load_qa_chain(OpenAI(temperature=0), chain_type="stuff")
+async def set_qa_chain_state_api_key(api_key):
+    # Set the API key for qa_chain_state
+    qa_chain_state.api_key = api_key
 
-        print(str(datetime.datetime.now()) + ": After load_chain, OPENAI_API_KEY length: " + str(
-            len(os.environ["OPENAI_API_KEY"])))
-        return chain, express_chain, llm, embeddings, qa_chain, memory
-    return None, None, None, None, None, None
+async def set_memory_state_api_key(api_key):
+    # Set the API key for memory_state
+    memory_state.api_key = api_key
 
+async def set_openai_api_key(api_key):
+    """
+    Sets the OpenAI API key for various components in the application asynchronously.
+
+    Args:
+        api_key (str): The OpenAI API key.
+
+    Side effects:
+        Updates the API key for chain_state, express_chain_state, llm_state,
+        embeddings_state, qa_chain_state, and memory_state.
+    """
+    try:
+        await set_chain_state_api_key(api_key)
+        await set_express_chain_state_api_key(api_key)
+        await set_llm_state_api_key(api_key)
+        await set_embeddings_state_api_key(api_key)
+        await set_qa_chain_state_api_key(api_key)
+        await set_memory_state_api_key(api_key)
+    except Exception as e:
+        # Handle the error, e.g., log the error message or show an alert to the user
+        print(f"Error setting OpenAI API key: {e}")
 
 
 def run_chain(chain, inp, capture_hidden_text):
@@ -373,7 +394,7 @@ class ChatWrapper:
         self.lock.acquire()
         try:
             print("\n==== date/time: " + str(datetime.datetime.now()) + " ====")
-            print("inp: " + str(inp))
+            print("inp: " + inp)
             print("trace_chain: ", trace_chain)
             print("speak_text: ", speak_text)
             print("talking_head: ", talking_head)
@@ -386,7 +407,7 @@ class ChatWrapper:
             if chain:
                 # Set OpenAI key
                 import openai
-                openai.api_key = openai_api_key
+                openai.api_key = api_key
                 if not monologue:
                     if use_embeddings:
                         if inp and inp.strip() != "":
@@ -616,6 +637,9 @@ with gr.Blocks(css=".gradio-container {background-color: lightgray}") as block:
                     <p><center>New features: <b>API key save. 2048 Input Tokens. News-api enabled
             </b></center></p>""")
 
+            openai_api_key_textbox = gr.Textbox(placeholder="Paste your OpenAI API key (sk-...)",
+                                                show_label=False, lines=1, type='password', elem_id="openai_api_key_textbox")
+
         with gr.Row():
             with gr.Column(scale=1, min_width=TALKING_HEAD_WIDTH, visible=True):
                 speak_text_cb = gr.Checkbox(label="Enable speech", value=False)
@@ -844,7 +868,29 @@ with gr.Blocks(css=".gradio-container {background-color: lightgray}") as block:
 <input type="hidden" name="currency_code" value="USD" />
 <input type="image" src="https://www.paypalobjects.com/en_US/i/btn/btn_donate_LG.gif" border="0" name="submit" title="PayPal - The safer, easier way to pay online!" alt="Donate with PayPal button" />
 <img alt="" border="0" src="https://www.paypal.com/en_US/i/scr/pixel.gif" width="1" height="1" />
-</form>
+</form># The OpenAI API key is stored in the browser's local storage and retrieved when the application is loaded.
+# This is done using the change() and load() methods of the openai_api_key_textbox object.
+
+# When the user inputs the OpenAI API key, it is saved to the local storage:
+openai_api_key_textbox.change(None,
+                              inputs=[openai_api_key_textbox],
+                              outputs=None, _js="(api_key) => localStorage.setItem('open_api_key', api_key)")
+
+# When the application is loaded, the OpenAI API key is retrieved from the local storage and set to the openai_api_key_textbox:
+block.load(None, inputs=None, outputs=openai_api_key_textbox, _js="()=> localStorage.getItem('open_api_key')")
+
+# The OpenAI API key is then used to set the API key for various components in the application:
+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])
+
+# The algorithmic timeline for using the OpenAI API key is as follows:
+# 1. The user inputs the OpenAI API key, which is saved to the local storage.
+# 2. The application retrieves the OpenAI API key from the local storage when it is loaded.
+# 3. The OpenAI API key is used to set the API key for various components in the application.
+# 4. The application can now use the OpenAI API key to make requests to the OpenAI API.
+
     """)
 
     gr.HTML("""<center>
@@ -853,22 +899,36 @@ 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=[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])
-
-    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])
-
-block.launch(debug=True)
+    message.submit(chat, inputs=[openai_api_key_textbox, 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])
+    # outputs=[chatbot, history_state, audio_html, tmp_aud_file, message])
+
+    submit.click(chat, inputs=[openai_api_key_textbox, 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])
+    # outputs=[chatbot, history_state, audio_html, tmp_aud_file, message])
+
+    openai_api_key_textbox.change(None,
+                                  inputs=[openai_api_key_textbox],
+                                  outputs=None, _js="(api_key) => localStorage.setItem('open_api_key', api_key)")
+
+    openai_api_key_textbox.change(lambda api_key: asyncio.run(set_openai_api_key(api_key)),
+                              inputs=[openai_api_key_textbox],
+                              outputs=[chain_state, express_chain_state, llm_state, embeddings_state,
+                                       qa_chain_state, memory_state])
+    
+    block.load(None, inputs=None, outputs=openai_api_key_textbox, _js="()=> localStorage.getItem('open_api_key')")
+   
+
+block.launch(debug=True)