Update space

app.py

@@ -2,7 +2,6 @@ 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")
 
@@ -76,6 +75,14 @@ def handle_prepare(index):
     return f"Prepare a 10-minute reading on what I should know before the class for the topic: {topic}"
 
 
+def create_buttons():
+    # Dynamically create buttons for each topic
+    button_list = []
+    for i, row in enumerate(data):
+        button_list.append(gr.Button(f"Prepare: {row['Topic']}").style(full_width=False))
+    return button_list
+
+
 # Gradio App
 with gr.Blocks() as demo:
     with gr.Row():
@@ -97,23 +104,14 @@ with gr.Blocks() as demo:
                 ],
             )
 
-            # Container for dynamically generated buttons
             buttons_container = gr.Column(visible=True)
 
-            def create_buttons():
-                components = []
-                for i, row in enumerate(data):
-                    btn = gr.Button(f"Prepare: {row['Topic']}")
-                    btn.click(
-                        fn=handle_prepare,
-                        inputs=[gr.State(i)],
-                        outputs=[chatbot],
-                    )
-                    components.append(btn)
-                return components
-
-            # Create buttons when table is extracted
-            extract_btn.click(fn=create_buttons, inputs=[], outputs=[buttons_container])
+            # Dynamically add buttons after table extraction
+            def update_buttons():
+                buttons = create_buttons()
+                return gr.update(components=buttons)
+
+            extract_btn.click(fn=update_buttons, inputs=[], outputs=[buttons_container])
 
 if __name__ == "__main__":
     demo.launch()