Update app.py

app.py

@@ -3,55 +3,56 @@ import json, openai, os, time
 
 from openai import OpenAI
 
-_client, _assistant, _thread = None, None, None
+_client = OpenAI(api_key=os.environ.get("OPENAI_API_KEY"))
+
+_assistant, _thread = None, None
 
 def show_json(str, obj):
     print(f"=> {str}\n{json.loads(obj.model_dump_json())}")
 
-def init_client():
-    global _client, _assistant, _thread
-    
-    _client = OpenAI(api_key=os.environ.get("OPENAI_API_KEY"))
-    
-    _assistant = _client.beta.assistants.create(
+def init_assistant(client):
+    assistant = client.beta.assistants.create(
         name="Math Tutor",
         instructions="You are a personal math tutor. Answer questions briefly, in a sentence or less.",
         model="gpt-4-1106-preview",
     )
-    #show_json("assistant", _assistant)
+    #show_json("assistant", assistant)
+    return assistant
     
-    _thread = _client.beta.threads.create()
-    #show_json("thread", _thread)
-
-def wait_on_run(run):
-    global _client, _thread
+def init_thread(client):
+    thread = client.beta.threads.create()
+    #show_json("thread", thread)
+    return thread
     
+def wait_on_run(client, thread, run):
     while run.status == "queued" or run.status == "in_progress":
-        run = _client.beta.threads.runs.retrieve(
-            thread_id=_thread.id,
+        run = client.beta.threads.runs.retrieve(
+            thread_id=thread.id,
             run_id=run.id,
         )
-        
         time.sleep(0.25)
-    
     return run
 
 def extract_content_values(data):
     content_values = []
-    
     for item in data.data:
         for content in item.content:
             if content.type == 'text':
                 content_values.append(content.text.value)
-    
     return content_values
 
 def chat(message, history, openai_api_key):
     global _client, _assistant, _thread     
        
     if _client == None:
-        init_client()
-            
+        _client = init_client()
+
+    if _assistant == None:
+        _assitant = init_assistant(_client)
+
+    if _thread == None:
+        _thread = init_thread(_client)
+        
     message = _client.beta.threads.messages.create(
         role="user",
         thread_id=_thread.id,
@@ -65,7 +66,7 @@ def chat(message, history, openai_api_key):
     )
     #show_json("run", run)
     
-    run = wait_on_run(run)
+    run = wait_on_run(_client, _thread, run)
     show_json("run", run)
     
     messages = _client.beta.threads.messages.list(