Update app.py

app.py

@@ -266,7 +266,7 @@ def load_chain(tools_list, llm):
 
 
 # async def set_chain_state_api_key(api_key):
-def set_openai_key(api_key):
+# def set_openai_key(api_key):
     # Set the API key for chain_state
 #    chain_state.api_key = api_key
 
@@ -289,7 +289,7 @@ def set_openai_key(api_key):
 # async def set_memory_state_api_key(api_key):
     # Set the API key for memory_state
 #     memory_state.api_key = api_key
-
+def set_openai_api_key(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")
@@ -312,32 +312,6 @@ def set_openai_key(api_key):
 
 PROMPTLAYER_API_BASE = "https://api.promptlayer.com"
 
-
-# async def set_openai_api_key(api_key):
-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_openai_key(api_key)
-#        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):
     output = ""
     hidden_text = None