Update space

app.py

@@ -111,6 +111,25 @@ def prepare_topic_action(topic_index, chatbot, system_message):
         print(f"Error in prepare_topic_action: {e}")
         return chatbot
 
+def add_user_message(chatbot, message):
+    if message:
+        return chatbot + [(message, None)]
+    return chatbot
+
+def generate_response(chatbot, system_message):
+    # Generate response for the last user message
+    if chatbot and chatbot[-1][1] is None:
+        last_user_message = chatbot[-1][0]
+        response_generator = respond(last_user_message, chatbot[:-1], system_message)
+        full_response = ""
+        for chunk in response_generator:
+            full_response = chunk
+        
+        # Update the last message with the full response
+        chatbot[-1] = (chatbot[-1][0], full_response)
+    
+    return chatbot
+
 # Gradio app
 with gr.Blocks() as demo:
     with gr.Row():
@@ -153,16 +172,16 @@ with gr.Blocks() as demo:
     
     # Submit button handler
     submit.click(
-        fn=lambda x: (x + [(msg.value, None)] if msg.value else x),
-        inputs=[chatbot],
-        outputs=[chatbot],
+        fn=add_user_message,
+        inputs=[chatbot, msg],
+        outputs=[chatbot]
     ).then(
         fn=lambda: "",  # Clear input textbox
-        outputs=[msg],
+        outputs=[msg]
     ).then(
-        fn=lambda history: list(respond(history[-1][0], history[:-1], system_message.value)),
+        fn=generate_response,
         inputs=[chatbot, system_message],
-        outputs=[chatbot],
+        outputs=[chatbot]
     )
     
     # Clear button handler