Update app.py

app.py

@@ -18,6 +18,14 @@ def wait_on_run(run, thread):
         time.sleep(0.5)
     return run
 
+def extract_content_value(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):    
     assistant = client.beta.assistants.create(
     name="Math Tutor",
@@ -53,12 +61,8 @@ def chat(message, history):
     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]
-    return "Done"
+
+    return extract_content_value(messages)[0]
 
 gr.ChatInterface(
     chat,