Update app.py

app.py

@@ -7,6 +7,15 @@ def show_json(obj):
     print(json.loads(obj.model_dump_json()))
     print("###")
 
+def wait_on_run(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)
+    return run
+
 def chat(message, history):
     client = OpenAI(api_key=os.environ.get("OPENAI_API_KEY"))
     
@@ -36,11 +45,19 @@ def chat(message, history):
     )
     
     show_json(run)
+
+    run = wait_on_run(run, thread)
     
+    show_json(run)
+
+    messages = client.beta.threads.messages.list(thread_id=thread.id)
+    
+    show_json(messages)
     ###
     
-    for i in range(len(message)):
-        time.sleep(0.5)
-        yield "You typed: " + message[: i+1]
+    #for i in range(len(message)):
+    #    time.sleep(0.5)
+    #    yield "You typed: " + message[: i+1]
+    return "Done"
 
 gr.ChatInterface(chat).launch()