Update app.py

app.py

@@ -3,6 +3,7 @@ import ai  # your custom model backend
 
 # --- Response function ---
 def respond(message, max_tokens, temperature, top_p):
+    start = f"{message}"
     output_so_far = ""
     for chunk in ai.stream_response(
         message=message,
@@ -10,7 +11,7 @@ def respond(message, max_tokens, temperature, top_p):
         temperature=temperature,
         top_p=top_p,
     ):
-        output_so_far = chunk
+        output_so_far = start + chunk
         yield output_so_far
 
 # --- UI ---