Update space

app.py

@@ -52,80 +52,79 @@ def extract_table(url):
                     'Topic': cells[1].text.strip(),
                 })
         
-        # Create HTML table with prepare buttons using inline JavaScript
-        html = '<table class="dataframe">'
+        # Create HTML table with prepare buttons
+        html = '''
+        <style>
+            .dataframe {
+                border-collapse: collapse;
+                width: 100%;
+                margin: 10px 0;
+            }
+            .dataframe th, .dataframe td {
+                border: 1px solid #ddd;
+                padding: 8px;
+                text-align: left;
+            }
+            .dataframe th {
+                background-color: #f6f8fa;
+            }
+            .dataframe tr:nth-child(even) {
+                background-color: #f9f9f9;
+            }
+            .prepare-btn {
+                padding: 5px 10px;
+                cursor: pointer;
+                background-color: #0366d6;
+                color: white;
+                border: none;
+                border-radius: 4px;
+            }
+            .prepare-btn:hover {
+                background-color: #0056b3;
+            }
+        </style>
+        '''
+        
+        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):
+            safe_topic = row['Topic'].replace("'", "\\'")  # Escape single quotes
             html += f'''
                 <tr>
                     <td>{row['Date']}</td>
                     <td>{row['Topic']}</td>
                     <td>
-                        <button 
-                            onclick="(() => {{
-                                const index = document.getElementById('prepare-index');
-                                index.value = '{i}';
-                                index.dispatchEvent(new Event('input'));
+                        <button class="prepare-btn" onclick="(function() {{
+                            const index = '{i}';
+                            const input = document.getElementById('prepare-index');
+                            input.value = index;
+                            const event = new Event('input');
+                            input.dispatchEvent(event);
+                            setTimeout(() => {{
                                 document.getElementById('prepare-trigger').click();
-                            }})();"
-                            style="padding: 5px 10px; cursor: pointer; background-color: #0366d6; color: white; border: none; border-radius: 4px;"
-                        >
+                            }}, 100);
+                        }})()">
                             Prepare
                         </button>
                     </td>
                 </tr>
             '''
         html += '</tbody></table>'
-        
-        # Add some CSS to make the table look better
-        html = f'''
-            <style>
-                .dataframe {{
-                    border-collapse: collapse;
-                    width: 100%;
-                    margin: 10px 0;
-                }}
-                .dataframe th, .dataframe td {{
-                    border: 1px solid #ddd;
-                    padding: 8px;
-                    text-align: left;
-                }}
-                .dataframe th {{
-                    background-color: #f6f8fa;
-                }}
-                .dataframe tr:nth-child(even) {{
-                    background-color: #f9f9f9;
-                }}
-                .dataframe button:hover {{
-                    background-color: #0056b3 !important;
-                }}
-            </style>
-            {html}
-        '''
         return html
+        
     except Exception as e:
+        print(f"Error in extract_table: {e}")  # Debug print
         return f"<p>Error: {str(e)}</p>"
 
-def add_text(history, text):
-    history = history + [(text, None)]
-    return history
-
-def generate_response(history, system_message):
-    if not history:
-        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 prepare_topic_message(index):
     try:
         print(f"Received index: {index}")  # Debug print
-        
-        # Handle the index as a number
+        if not index:
+            print("Empty index received")
+            return ""
+            
         idx = int(index)
         if 0 <= idx < len(data):
             topic = data[idx]["Topic"]
@@ -140,9 +139,25 @@ def prepare_topic_message(index):
         print(f"Error in prepare_topic_message: {e}")
         return ""
 
+def add_text(history, text):
+    history = history + [(text, None)]
+    return history
+
+def generate_response(history, system_message):
+    if not history:
+        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 clear_chat():
     return [], ""
 
+# Gradio app
 # Gradio app
 with gr.Blocks() as demo:
     with gr.Row():
@@ -154,9 +169,10 @@ with gr.Blocks() as demo:
             table_output = gr.HTML(label="Extracted Table")
             extract_btn = gr.Button("Extract Table")
             
-            # Hidden components for prepare functionality
-            prepare_index = gr.Textbox(value="", visible=False, elem_id="prepare-index")
-            prepare_trigger = gr.Button("Prepare", visible=False, elem_id="prepare-trigger")
+            # Replace this section
+            with gr.Row(visible=False):  # Hide these components
+                prepare_index = gr.Textbox(value="", elem_id="prepare-index")
+                prepare_trigger = gr.Button("Prepare", elem_id="prepare-trigger")
             
         with gr.Column(scale=2):
             chatbot = gr.Chatbot()
@@ -177,25 +193,24 @@ with gr.Blocks() as demo:
         outputs=[table_output]
     )
     
-    # Prepare trigger handler
-    # Prepare trigger handler
+    # Replace the existing prepare trigger handler with this one
     prepare_trigger.click(
         fn=prepare_topic_message,
-        inputs=prepare_index,
-        outputs=msg,
-        queue=False  # Changed from api_name to queue=False
+        inputs=[prepare_index],
+        outputs=[msg],
+        queue=False
     ).success(
         fn=add_text,
         inputs=[chatbot, msg],
-        outputs=chatbot,
+        outputs=[chatbot],
         queue=False
     ).then(
         fn=generate_response,
         inputs=[chatbot, system_message],
-        outputs=chatbot
+        outputs=[chatbot]
     )
     
-    # Submit button handler
+    # Rest of your event handlers remain the same
     msg.submit(
         fn=add_text,
         inputs=[chatbot, msg],
@@ -223,7 +238,4 @@ with gr.Blocks() as demo:
     )
     
     # Clear button handler
-    clear.click(fn=clear_chat, outputs=[chatbot, msg])
-
-if __name__ == "__main__":
-    demo.launch()
+    clear.click(fn=clear_chat, outputs=[chatbot, msg])