Update app.py

app.py

@@ -10,10 +10,9 @@ import gradio as gr
 
 import os
 
-from openai import OpenAI
-
 from assistants import (
-    openai_client,
+    get_openai_client,
+    set_openai_client,
     assistant,
     thread,
     create_assistant,
@@ -36,14 +35,10 @@ def chat(message, history, openai_api_key):
 
     #raise gr.Error("Please clone and bring your own credentials.")
     
-    global openai_client, assistant, thread
+    global assistant, thread
 
-    print("openai_client=")
-    print(openai_client)
-    if openai_client == None:
-        openai_client = OpenAI(api_key=openai_api_key)
-    print("openai_client=")
-    print(openai_client)
+    if get_openai_client() == None:
+        set_openai_client(openai_api_key)
     
     # On first run, create assistant and update assistant_id,
     # see https://platform.openai.com/playground/assistants.
@@ -70,8 +65,6 @@ def chat(message, history, openai_api_key):
     if len(image_values) > 0:
         download_link = f"<hr>[Download](https://platform.openai.com/storage/files/{image_values[0]})"
 
-    del os.environ["OPENAI_API_KEY"]
-    
     #return f"{'<hr>'.join(list(reversed(text_values))[1:])}{download_link}"
     return f"{text_values[0]}{download_link}"