Update app.py

app.py

@@ -32,7 +32,15 @@ def create_message(client, thread, msg):
     )
     show_json("message", message)
     return message
-    
+
+def create_run(client, assistant, thread):
+    run = client.beta.threads.runs.create(
+        assistant_id=assistant.id,
+        thread_id=thread.id,
+    )
+    show_json("run", run)
+    return run
+
 def wait_on_run(client, thread, run):
     while run.status == "queued" or run.status == "in_progress":
         run = client.beta.threads.runs.retrieve(
@@ -40,6 +48,7 @@ def wait_on_run(client, thread, run):
             run_id=run.id,
         )
         time.sleep(0.25)
+    show_json("run", run)
     return run
 
 def extract_content_values(data):
@@ -61,14 +70,8 @@ def chat(message, history, openai_api_key):
         
     message = create_message(_client, _thread, message)
     
-    run = _client.beta.threads.runs.create(
-        assistant_id=_assistant.id,
-        thread_id=_thread.id,
-    )
-    show_json("run", run)
-    
+    run = create_run(_client, _assistant, _thread)
     run = wait_on_run(_client, _thread, run)
-    show_json("run", run)
     
     messages = _client.beta.threads.messages.list(
         thread_id=_thread.id