Update app.py

app.py

@@ -11,6 +11,7 @@ import gradio as gr
 import os
 
 from assistants import (
+    get_openai_client,
     set_openai_client,
     get_assistant,
     set_assistant,
@@ -33,19 +34,15 @@ def chat(message, history, openai_api_key):
         raise gr.Error("OpenAI API Key is required (see additional inputs below).")
     if not message:
         raise gr.Error("Message is required.")
+
+    if get_openai_client() == None:
+        set_openai_client(openai_api_key)
     
-    global assistant, thread
-    
-    # On first run, create assistant and update assistant_id,
-    # see https://platform.openai.com/playground/assistants.
-    # On subsequent runs, load assistant.
     if get_assistant() == None:
         #set_assistant(create_assistant())
         set_assistant(load_assistant())
 
-    # TODO: Use Gradio session to support multiple users
     if get_thread == None or len(history) == 0:
-        set_openai_client(openai_api_key)
         set_thread(create_thread())
         
     create_message(get_thread(), message)
@@ -62,8 +59,8 @@ 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]})"
 
-    #return f"{'<hr>'.join(list(reversed(text_values))[1:])}{download_link}"
-    return f"{text_values[0]}{download_link}"
+    return f"{'<hr>'.join(list(reversed(text_values))[1:])}{download_link}"
+    #return f"{text_values[0]}{download_link}"
 
 gr.ChatInterface(
         fn=chat,