Update space

app.py

@@ -2,11 +2,11 @@ import gradio as gr
 from huggingface_hub import InferenceClient
 import requests
 from bs4 import BeautifulSoup
+import pandas as pd
 
 client = InferenceClient("meta-llama/Llama-3.2-3B-Instruct")
 
-# Global data store for the table
-data = []
+data = []  # Global variable to store table data
 
 def respond(message, history, system_message, max_tokens, temperature, top_p):
     messages = [{"role": "system", "content": system_message}]
@@ -40,23 +40,21 @@ def extract_table(url):
             return "<p>No table found on page</p>"
 
         # Extract data
-        data.clear()
+        data = []
         rows = table.find_all("tr")
         for i, row in enumerate(rows[1:]):  # Skip header row
             cells = row.find_all("td")
             if len(cells) >= 2:
                 data.append({"Index": i, "Date": cells[0].text.strip()[:10], "Topic": cells[1].text.strip()})
 
-        # Generate HTML table with "Prepare" buttons in the third column
+        # Generate HTML table with Prepare buttons
         html_rows = ""
         for row in data:
             html_rows += f"""
                 <tr>
                     <td>{row['Date']}</td>
                     <td>{row['Topic']}</td>
-                    <td>
-                        <button onclick="gradioApp().submit('{row['Index']}', 'prepare_button')">Prepare</button>
-                    </td>
+                    <td><button onclick="prepareTopic({row['Index']})">Prepare</button></td>
                 </tr>
             """
         html_table = f"""
@@ -68,29 +66,33 @@ def extract_table(url):
                 </tr>
                 {html_rows}
             </table>
+            <script>
+                function prepareTopic(index) {{
+                    document.getElementById('prepareInput').value = index;
+                    document.getElementById('prepareSubmit').click();
+                }}
+            </script>
         """
         return html_table
     except Exception as e:
-        return f"<p>Error: {str(e)}</p>" 
+        return f"<p>Error: {str(e)}</p>"
 
 
 def handle_prepare(index):
-    index = int(index)
-    topic = data[index]["Topic"]
-    return f"Prepare a 10-minute reading on what I should know before the class for the topic: {topic}"
-
+    try:
+        index = int(index)
+        topic = data[index]["Topic"]
+        return f"Prepare a 10-minute reading on what I should know before the class for the topic: {topic}"
+    except Exception as e:
+        return f"<p>Error: {str(e)}</p>"
 
-# Gradio App
+# Gradio app
 with gr.Blocks() as demo:
     with gr.Row():
         with gr.Column(scale=1):
-            url_input = gr.Textbox(
-                value="https://id2223kth.github.io/schedule/",
-                label="Table URL",
-            )
+            url_input = gr.Textbox(value="https://id2223kth.github.io/schedule/", label="Table URL")
             table_output = gr.HTML(label="Extracted Table")
             extract_btn = gr.Button("Extract Table")
-
             extract_btn.click(fn=extract_table, inputs=[url_input], outputs=[table_output])
 
         with gr.Column(scale=2):
@@ -100,15 +102,10 @@ with gr.Blocks() as demo:
                     gr.Textbox(value="Student class preparation companion.", label="System message"),
                 ],
             )
-
-    # Button interaction logic
-    prepare_button = gr.Textbox(label="Prepare Input", visible=False)
-
-    prepare_button.change(
-        fn=handle_prepare,
-        inputs=[prepare_button],
-        outputs=chatbot,
-    )
+            # Hidden input to handle prepare actions
+            prepare_input = gr.Textbox(visible=False, elem_id="prepareInput")
+            prepare_submit = gr.Button("Prepare Submit", visible=False, elem_id="prepareSubmit")
+            prepare_submit.click(fn=handle_prepare, inputs=[prepare_input], outputs=[chatbot])
 
 if __name__ == "__main__":
     demo.launch()