Update space

app.py

@@ -65,7 +65,6 @@ def extract_table(url):
                         <button onclick='
                             document.getElementById("prepare-topic").value = "{i}";
                             document.getElementById("prepare-topic").dispatchEvent(new Event("input"));
-                            document.getElementById("prepare-button").click();
                         '>
                             Prepare
                         </button>
@@ -85,21 +84,39 @@ def generate_response(history, system_message):
     if not history:
         return history
     
+    # Ensure the last entry is a full user message and not partially processed
+    if history[-1][1] is not None:
+        return history
+    
     response = ""
     for chunk in respond(history[-1][0], history[:-1], system_message):
         response = chunk
         history[-1] = (history[-1][0], response)
         yield history
 
-def handle_prepare(index, history):
+def handle_prepare(index, history, system_message):
     try:
         index = int(index)
         if 0 <= index < len(data):
             topic = data[index]["Topic"]
             date = data[index]["Date"]
             message = f"Please prepare a 10-minute reading guide for the topic '{topic}' scheduled for {date}"
-            return history + [(message, None)]
-    except:
+            
+            # Add the prepare message to history
+            new_history = history + [(message, None)]
+            
+            # Generate response specifically for this prepare request
+            response_generator = respond(message, history, system_message)
+            full_response = ""
+            for chunk in response_generator:
+                full_response = chunk
+            
+            # Update history with the full response
+            new_history[-1] = (message, full_response)
+            
+            return new_history
+    except Exception as e:
+        print(f"Error in handle_prepare: {e}")
         return history
 
 def clear_chat():
@@ -118,7 +135,6 @@ with gr.Blocks() as demo:
             
             # Hidden components for prepare functionality
             prepare_topic = gr.Textbox(value="", visible=False, elem_id="prepare-topic")
-            prepare_button = gr.Button("Prepare", visible=False, elem_id="prepare-button")
             
         with gr.Column(scale=2):
             chatbot = gr.Chatbot()
@@ -139,6 +155,13 @@ with gr.Blocks() as demo:
         outputs=[table_output]
     )
     
+    # Prepare topic handler
+    prepare_topic.change(
+        fn=handle_prepare,
+        inputs=[prepare_topic, chatbot, system_message],
+        outputs=[chatbot]
+    )
+    
     # Submit button handler
     msg.submit(
         fn=add_text,
@@ -166,17 +189,6 @@ with gr.Blocks() as demo:
         outputs=chatbot
     )
     
-    # Prepare button handler
-    prepare_button.click(
-        fn=handle_prepare,
-        inputs=[prepare_topic, chatbot],
-        outputs=chatbot
-    ).then(
-        fn=generate_response,
-        inputs=[chatbot, system_message],
-        outputs=chatbot
-    )
-    
     # Clear button handler
     clear.click(fn=clear_chat, outputs=[chatbot, msg])