Update app.py

app.py

@@ -13,7 +13,9 @@ import os
 from assistants import (
     set_openai_client,
     get_assistant,
+    set_assistant,
     get_thread,
+    set_thread,
     create_assistant,
     load_assistant,
     create_thread,
@@ -37,22 +39,21 @@ def chat(message, history, openai_api_key):
     # 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:
-        #assistant = create_assistant()
-        assistant = 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)
-
-        thread = create_thread()
+        set_thread(create_thread())
         
-    create_message(thread, message)
-    run = create_run(assistant, thread)
-    run = wait_on_run(thread, run)
-    run_steps = get_run_steps(thread, run)
-    recurse_execute_tool_calls(thread, run, run_steps, 0)
-    messages = get_messages(thread)
+    create_message(get_thread(), message)
+    run = create_run(get_assistant(), get_thread())
+    run = wait_on_run(get_thread(), run)
+    run_steps = get_run_steps(get_thread(), run)
+    recurse_execute_tool_calls(get_thread(), run, run_steps, 0)
+    messages = get_messages(get_thread())
     text_values, image_values = extract_content_values(messages)
 
     download_link = ""