Update space

app.py

@@ -53,30 +53,50 @@ def extract_table(url):
         if not table:
             return "<p>No table found on page</p>"
         
+        # Extract data
         data = []
         rows = table.find_all('tr')
         for i, row in enumerate(rows[1:]):
             cells = row.find_all('td')
             if len(cells) >= 2:
                 data.append({
+                    'Index': i,
                     'Date': cells[0].text.strip()[:10],
                     'Topic': cells[1].text.strip(),
                 })
         
-        df = pd.DataFrame(data)
-        return df.to_html(escape=False, index=False)
+        # Generate HTML table with buttons
+        html_rows = ""
+        for row in data:
+            html_rows += f"""
+                <tr>
+                    <td>{row['Date']}</td>
+                    <td>{row['Topic']}</td>
+                    <td><button onclick="handlePrepare({row['Index']})">Prepare</button></td>
+                </tr>
+            """
+        html_table = f"""
+            <table border="1">
+                <tr>
+                    <th>Date</th>
+                    <th>Topic</th>
+                    <th>Action</th>
+                </tr>
+                {html_rows}
+            </table>
+        """
+        return html_table
     except Exception as e:
         return f"<p>Error: {str(e)}</p>"
 
 
+
 def display_table(url):
     return extract_table(url)
 
-def prepare(index):
-    row = data[index]
-    message = f"Prepare a 10 minutes reading of important part regarding {row['Topic']}"
-    return message
-
+def handle_prepare(index):
+    topic = data[index]["Topic"]
+    return f"Prepare a 10-minute reading on what I should know before the class for the topic: {topic}"
 
 """
 For information on how to customize the ChatInterface, peruse the gradio docs: https://www.gradio.app/docs/chatinterface
@@ -98,7 +118,6 @@ with gr.Blocks() as demo:
             )
         
         with gr.Column(scale=2):
-            
             chatbot = gr.ChatInterface(
                 respond,
                 additional_inputs=[
@@ -106,6 +125,32 @@ with gr.Blocks() as demo:
                 ],
             )
             
+            # Dynamically generate buttons for "Prepare" actions
+            prepare_buttons = gr.Column(visible=False)
+
+            def create_prepare_buttons():
+                # Create individual buttons for each topic
+                components = []
+                for i, row in enumerate(data):
+                    btn = gr.Button(f"Prepare: {row['Topic']}")
+                    btn.click(
+                        fn=handle_prepare,
+                        inputs=[gr.State(i)],  # Pass the index of the row
+                        outputs=[],
+                    ).then(
+                        fn=lambda message: ([message],),  # Feed to chatbot
+                        inputs=[],
+                        outputs=[chatbot],
+                    )
+                    components.append(btn)
+                return components
+
+            extract_btn.click(
+                fn=create_prepare_buttons,
+                inputs=[],
+                outputs=[prepare_buttons],
+            )
+
 
 if __name__ == "__main__":
     demo.launch()