Update app.py

app.py

@@ -1,7 +1,11 @@
 import gradio as gr
-import json, openai, os, time
+import json, openai, os
 from openai import OpenAI
 
+def show_json(str, obj):
+    print(f"### {str}")
+    print(json.loads(obj.model_dump_json()))
+
 def init_assistant():
     client = OpenAI(api_key=os.environ.get("OPENAI_API_KEY"))
     
@@ -10,27 +14,20 @@ def init_assistant():
         instructions="You are a personal math tutor. Answer questions briefly, in a sentence or less.",
         model="gpt-4-1106-preview",
     )
-    
-    show_json(assistant)
+    show_json("assistant", assistant)
     
     thread = client.beta.threads.create()
-    
-    show_json(thread)
+    show_json("thread", thread)
 
     return client, assistant, thread
 
-def show_json(obj):
-    print("###")
-    print(json.loads(obj.model_dump_json()))
-    print("###")
-
 def wait_on_run(client, run, thread):
     while run.status == "queued" or run.status == "in_progress":
         run = client.beta.threads.runs.retrieve(
             thread_id=thread.id,
             run_id=run.id,
         )
-        time.sleep(0.5)
+        time.sleep(0.25)
     return run
 
 def extract_content_value(data):
@@ -51,37 +48,31 @@ def chat(message, history):
         history_openai_format.append({"role": "assistant", "content":assistant})
     history_openai_format.append({"role": "user", "content": message})
 
-    print("###")
+    print("### history")
     print(history_openai_format)
-    print("###")
     
-    message = client.beta.threads.messages.create(
+    messages = client.beta.threads.messages.create(
         thread_id=thread.id,
         role="user",
         content=message,
     )
     
-    show_json(message)
-
-    print("###")
-    print(thread.id)
-    print("###")
-    print(assistant.id)
+    show_json("messages", messages)
     
-    run = client.beta.threads.runs.create(
+    runs = client.beta.threads.runs.create(
         thread_id=thread.id,
         assistant_id=assistant.id,
     )
     
-    show_json(run)
+    show_json("runs", runs)
 
     run = wait_on_run(client, run, thread)
     
-    show_json(run)
+    show_json("runs", runs)
 
     messages = client.beta.threads.messages.list(thread_id=thread.id)
     
-    show_json(messages)
+    show_json("messages", messages)
 
     return extract_content_value(messages)[0]