return only last bit of message

app.py

@@ -123,12 +123,12 @@ def chat_with_model(messages):
 
         print(f'Step 2: {messages}')
 
-        yield messages
+        yield messages[-1]["content"]
 
     if in_think:
         output_text += "*"
         messages[-1]["content"] = output_text
-        yield messages
+        yield messages[-1]["content"]
 
     # Wait for thread to finish
     thread.join(timeout=1.0)
@@ -137,7 +137,7 @@ def chat_with_model(messages):
 
     print(f'Step 3: {messages}')
 
-    return messages
+    return messages[-1]["content"]