Update space

app.py

@@ -51,24 +51,11 @@ def extract_table(url):
                 })
 
         # Generate HTML table
-        html_rows = ""
+        table_html = "<table border='1'><tr><th>Date</th><th>Topic</th></tr>"
         for row in data:
-            html_rows += f"""
-                <tr>
-                    <td>{row['Date']}</td>
-                    <td>{row['Topic']}</td>
-                </tr>
-            """
-        html_table = f"""
-            <table border="1">
-                <tr>
-                    <th>Date</th>
-                    <th>Topic</th>
-                </tr>
-                {html_rows}
-            </table>
-        """
-        return html_table
+            table_html += f"<tr><td>{row['Date']}</td><td>{row['Topic']}</td></tr>"
+        table_html += "</table>"
+        return table_html
     except Exception as e:
         return f"<p>Error: {str(e)}</p>"
 
@@ -83,13 +70,39 @@ def handle_prepare(index, history):
 def chat(message, history, system_msg):
     history = history or []
     history.append((message, None))
-    response = respond(message, history, system_msg)
-    history[-1] = (message, response)
     return "", history
 
 def clear_chat():
     return [], []
 
+def create_table_with_buttons():
+    with gr.Column() as container:
+        # Create the table display
+        table_html = "<table border='1'><tr><th>Date</th><th>Topic</th></tr>"
+        for row in data:
+            table_html += f"<tr><td>{row['Date']}</td><td>{row['Topic']}</td></tr>"
+        table_html += "</table>"
+        gr.HTML(table_html)
+        
+        # Create buttons below the table
+        with gr.Row():
+            for i, row in enumerate(data):
+                btn = gr.Button(f"Prepare: {row['Topic']}")
+                btn.click(
+                    fn=handle_prepare,
+                    inputs=[gr.Number(value=i, visible=False), chatbot],
+                    outputs=[chatbot]
+                ).then(
+                    fn=respond,
+                    inputs=[
+                        gr.Textbox(value=f"Prepare a 10-minute reading for: {row['Topic']}", visible=False),
+                        chatbot,
+                        system_message
+                    ],
+                    outputs=[chatbot]
+                )
+    return container
+
 # Gradio App
 with gr.Blocks() as demo:
     with gr.Row():
@@ -98,7 +111,7 @@ with gr.Blocks() as demo:
                 value="https://id2223kth.github.io/schedule/",
                 label="Table URL",
             )
-            table_output = gr.HTML(label="Extracted Table")
+            table_container = gr.Column()  # Container for table and buttons
             extract_btn = gr.Button("Extract Table")
 
         with gr.Column(scale=2):
@@ -113,56 +126,27 @@ with gr.Blocks() as demo:
                 submit = gr.Button("Submit")
                 clear = gr.Button("Clear")
 
-            # Container for dynamic buttons
-            buttons_container = gr.Column()
-
-            def prepare_topic(index, history):
-                if 0 <= index < len(data):
-                    topic = data[index]["Topic"]
-                    prompt = f"Prepare a 10-minute reading for: {topic}"
-                    history = history or []
-                    history.append((prompt, None))
-                    return history
-
-            submit.click(
-                fn=chat,
-                inputs=[msg, chatbot, system_message],
-                outputs=[msg, chatbot]
-            )
-            
-            clear.click(fn=clear_chat, outputs=[chatbot, msg])
-
-            def update_buttons():
-                buttons_container.clear()
-                with buttons_container:
-                    for i, row in enumerate(data):
-                        btn = gr.Button(f"Prepare: {row['Topic']}")
-                        btn.click(
-                            fn=prepare_topic,
-                            inputs=[
-                                gr.Number(value=i, visible=False),
-                                chatbot
-                            ],
-                            outputs=[chatbot]
-                        ).then(
-                            fn=respond,
-                            inputs=[
-                                gr.Textbox(value=f"Prepare a 10-minute reading for: {row['Topic']}", visible=False),
-                                chatbot,
-                                system_message
-                            ],
-                            outputs=[chatbot]
-                        )
-
-            extract_btn.click(
-                fn=extract_table,
-                inputs=[url_input],
-                outputs=[table_output]
-            ).then(
-                fn=update_buttons,
-                inputs=None,
-                outputs=None
-            )
+    extract_btn.click(
+        fn=extract_table,
+        inputs=[url_input],
+        outputs=[table_container]
+    ).then(
+        fn=create_table_with_buttons,
+        inputs=None,
+        outputs=table_container
+    )
+
+    submit.click(
+        fn=chat,
+        inputs=[msg, chatbot, system_message],
+        outputs=[msg, chatbot]
+    ).then(
+        fn=respond,
+        inputs=[msg, chatbot, system_message],
+        outputs=[chatbot]
+    )
+    
+    clear.click(fn=clear_chat, outputs=[chatbot, msg])
 
 if __name__ == "__main__":
     demo.launch()