Update app.py

app.py

@@ -337,12 +337,14 @@ def create_ui():
                     gr.Examples(examples, inputs=msg)
 
                     def respond_wrapper(message, chat_history, max_tokens, temperature, top_p):
-                        bot_message = respond(message, chat_history, max_tokens, temperature, top_p)
+                        bot_message = ""
+                        for response in respond(message, chat_history, max_tokens=max_tokens, temperature=temperature, top_p=top_p):
+                            bot_message = response  # 마지막 응답을 저장
+    
                         chat_history.append({"role": "user", "content": message})
                         chat_history.append({"role": "assistant", "content": bot_message})
                         return "", chat_history
 
-                    msg.submit(respond_wrapper, [msg, chatbot, max_tokens, temperature, top_p], [msg, chatbot])
 
             space_id_state = gr.State()
             tree_structure_state = gr.State()