Update app.py

app.py

@@ -60,10 +60,16 @@ with gr.Blocks(title="kFlash 1.3") as demo:
         return "", history + [[user_message, None]]
 
     def bot_reply(history):
-        history[-1][1] = ""
+        if not history:
+            return history
+        
+        # Kerää vastaus kokonaan ennen kuin päivität UI:ta
+        full_response = ""
         for partial in respond(history[-1][0], history[:-1]):
-            history[-1][1] = partial
-            yield history
+            full_response = partial
+        
+        history[-1][1] = full_response
+        return history
 
     msg.submit(user_submit, [msg, chatbot], [msg, chatbot]).then(
         bot_reply, chatbot, chatbot