Update app.py

app.py

@@ -31,9 +31,9 @@ def chat(message, history, openai_api_key, tavily_api_key):
 
     if not openai_api_key:
         raise gr.Error("OpenAI API key is required.")
-    
-    print(openai_api_key)
-    print(tavily_api_key)
+
+    os.environ["OPENAI_API_KEY"] = openai_api_key
+    os.environ["TAVILY_API_KEY"] = tavily_api_key
     
     global assistant, thread     
 
@@ -61,6 +61,9 @@ def chat(message, history, openai_api_key, tavily_api_key):
     # TODO: Handle multiple images and other file types
     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"]
+    del os.environ["TAVILY_API_KEY"]
     
     #return f"{'<hr>'.join(list(reversed(text_values))[1:])}{download_link}"
     return f"{text_values[0]}{download_link}"