Update space

app.py

@@ -19,7 +19,7 @@ def respond(message, history, system_message):
             
     messages.append({"role": "user", "content": message})
     
-    full_response = ""
+    response = ""
     for message in client.chat_completion(
         messages,
         max_tokens=2048,
@@ -28,9 +28,8 @@ def respond(message, history, system_message):
         top_p=0.9,
     ):
         if message.choices[0].delta.content is not None:
-            full_response += message.choices[0].delta.content
-    
-    return full_response
+            response += message.choices[0].delta.content
+            yield response
 
 def extract_table(url):
     global data
@@ -64,8 +63,8 @@ def extract_table(url):
                     <td>{row['Topic']}</td>
                     <td>
                         <button onclick='
-                            document.getElementById("prepare-topic").value = "{i}";
-                            document.getElementById("prepare-topic").dispatchEvent(new Event("input"));
+                            document.getElementById("prepare-index").value = "{i}";
+                            document.getElementById("prepare-trigger").click();
                         '>
                             Prepare
                         </button>
@@ -85,32 +84,21 @@ 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 = respond(history[-1][0], history[:-1], system_message)
-    history[-1] = (history[-1][0], response)
-    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, system_message):
+def prepare_topic_message(index):
     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}"
-            
-            # Generate response for the prepare request
-            response = respond(message, history, system_message)
-            
-            # Add both the request and response to history
-            new_history = history + [(message, response)]
-            
-            return new_history
-    except Exception as e:
-        print(f"Error in handle_prepare: {e}")
-        return history
+            return f"Please prepare a 10-minute reading guide for the topic '{topic}' scheduled for {date}"
+    except:
+        return ""
 
 def clear_chat():
     return [], ""
@@ -127,7 +115,8 @@ with gr.Blocks() as demo:
             extract_btn = gr.Button("Extract Table")
             
             # Hidden components for prepare functionality
-            prepare_topic = gr.Textbox(value="", visible=False, elem_id="prepare-topic")
+            prepare_index = gr.Textbox(value="", visible=False, elem_id="prepare-index")
+            prepare_trigger = gr.Button("Prepare", visible=False, elem_id="prepare-trigger")
             
         with gr.Column(scale=2):
             chatbot = gr.Chatbot()
@@ -148,10 +137,24 @@ with gr.Blocks() as demo:
         outputs=[table_output]
     )
     
-    # Prepare topic handler
-    prepare_topic.change(
-        fn=handle_prepare,
-        inputs=[prepare_topic, chatbot, system_message],
+    # Prepare trigger handler
+    prepare_trigger.click(
+        fn=prepare_topic_message,
+        inputs=[prepare_index],
+        outputs=[msg]
+    )
+    
+    # Submit button handler for prepare trigger
+    prepare_trigger.click(
+        fn=add_text,
+        inputs=[chatbot, msg],
+        outputs=[chatbot]
+    ).success(
+        fn=lambda: "",
+        outputs=[msg]
+    ).then(
+        fn=generate_response,
+        inputs=[chatbot, system_message],
         outputs=[chatbot]
     )
     
@@ -159,27 +162,27 @@ with gr.Blocks() as demo:
     msg.submit(
         fn=add_text,
         inputs=[chatbot, msg],
-        outputs=chatbot
+        outputs=[chatbot]
     ).success(
         fn=lambda: "",
-        outputs=msg
+        outputs=[msg]
     ).then(
         fn=generate_response,
         inputs=[chatbot, system_message],
-        outputs=chatbot
+        outputs=[chatbot]
     )
     
     submit.click(
         fn=add_text,
         inputs=[chatbot, msg],
-        outputs=chatbot
+        outputs=[chatbot]
     ).success(
         fn=lambda: "",
-        outputs=msg
+        outputs=[msg]
     ).then(
         fn=generate_response,
         inputs=[chatbot, system_message],
-        outputs=chatbot
+        outputs=[chatbot]
     )
     
     # Clear button handler