Update app.py

app.py

@@ -26,10 +26,13 @@ from assistants import (
 )
 
 def chat(message, history, openai_api_key):
+    if not openai_api_key:
+        raise gr.Error("OpenAI API Key is required.")
     if not message:
         raise gr.Error("Message is required.")
 
     #raise gr.Error("Please clone and bring your own credentials.")
+    os.environ["OPENAI_API_KEY"] = openai_api_key
     
     global assistant, thread
 
@@ -58,6 +61,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]})"
 
+    del os.environ["OPENAI_API_KEY"]
+    
     #return f"{'<hr>'.join(list(reversed(text_values))[1:])}{download_link}"
     return f"{text_values[0]}{download_link}"