Update space

app.py

@@ -52,8 +52,18 @@ def extract_table(url):
                     'Topic': cells[1].text.strip(),
                 })
         
-        # Create HTML table with prepare buttons
-        html = '<table class="dataframe">'
+        # Create HTML table with prepare buttons using data attributes
+        html = '''
+        <script>
+        function prepareTopic(index) {
+            const indexInput = document.getElementById("prepare-index");
+            indexInput.value = index;
+            indexInput.dispatchEvent(new Event("input", { bubbles: true }));
+            document.getElementById("prepare-trigger").click();
+        }
+        </script>
+        '''
+        html += '<table class="dataframe">'
         html += '<thead><tr><th>Date</th><th>Topic</th><th>Action</th></tr></thead>'
         html += '<tbody>'
         for i, row in enumerate(data):
@@ -62,15 +72,7 @@ def extract_table(url):
                     <td>{row['Date']}</td>
                     <td>{row['Topic']}</td>
                     <td>
-                        <button onclick='
-                            (function() {{
-                                const indexInput = document.getElementById("prepare-index");
-                                indexInput.value = {i};
-                                const event = new Event("input", {{ bubbles: true }});
-                                indexInput.dispatchEvent(event);
-                                document.getElementById("prepare-trigger").click();
-                            }})();
-                        '>
+                        <button onclick="prepareTopic({i})" data-index="{i}">
                             Prepare
                         </button>
                     </td>
@@ -97,15 +99,10 @@ def generate_response(history, system_message):
 
 def prepare_topic_message(index):
     try:
-        print(f"Received index: '{index}'")  # Debug print
+        print(f"Received index: {index}")  # Debug print
         
-        # Convert index to integer, handling both string and numeric inputs
-        try:
-            idx = int(float(index))
-        except (ValueError, TypeError):
-            print(f"Invalid index value: {index}")
-            return ""
-            
+        # Handle the index as a number
+        idx = int(index)
         if 0 <= idx < len(data):
             topic = data[idx]["Topic"]
             date = data[idx]["Date"]
@@ -159,17 +156,17 @@ with gr.Blocks() as demo:
     # Prepare trigger handler
     prepare_trigger.click(
         fn=prepare_topic_message,
-        inputs=[prepare_index],
-        outputs=[msg],
-        queue=False  # Add this
-    ).success(  # Change .then to .success
+        inputs=prepare_index,
+        outputs=msg,
+        api_name="prepare"  # Add this
+    ).success(
         fn=add_text,
         inputs=[chatbot, msg],
-        outputs=[chatbot]
+        outputs=chatbot
     ).then(
         fn=generate_response,
         inputs=[chatbot, system_message],
-        outputs=[chatbot]
+        outputs=chatbot
     )
     
     # Submit button handler