Update space

app.py

@@ -8,7 +8,7 @@ client = InferenceClient("meta-llama/Llama-3.2-3B-Instruct")
 # Global data store for the table
 data = []
 
-def respond(message, history, system_message, max_tokens=2048, temperature=0.7, top_p=0.9):
+def respond(message, history, system_message, max_tokens, temperature, top_p):
     messages = [{"role": "system", "content": system_message}]
     for val in history:
         if val[0]:
@@ -28,6 +28,7 @@ def respond(message, history, system_message, max_tokens=2048, temperature=0.7,
         response += token
         yield response
 
+
 def extract_table(url):
     global data
     try:
@@ -39,69 +40,48 @@ def extract_table(url):
             return "<p>No table found on page</p>"
 
         # Extract data
-        data.clear()  # Clear existing data
+        data = []
         rows = table.find_all("tr")
         for i, row in enumerate(rows[1:]):  # Skip header
             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
-        table_html = "<table border='1'><tr><th>Date</th><th>Topic</th></tr>"
+                data.append({"Index": i, "Date": cells[0].text.strip()[:10], "Topic": cells[1].text.strip()})
+
+        # Generate HTML table without buttons
+        html_rows = ""
         for row in data:
-            table_html += f"<tr><td>{row['Date']}</td><td>{row['Topic']}</td></tr>"
-        table_html += "</table>"
-        return table_html
+            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
     except Exception as e:
-        return f"<p>Error: {str(e)}</p>"
-
-def handle_prepare(index, history):
-    if 0 <= index < len(data):
-        topic = data[index]["Topic"]
-        prompt = f"Prepare a 10-minute reading on what I should know before the class for the topic: {topic}"
-        history = history or []
-        history.append((prompt, None))
-        return history
-
-def chat(message, history, system_msg):
-    history = history or []
-    history.append((message, None))
-    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
+        return f"<p>Error: {str(e)}</p>" 
+
+
+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}"
+
+
+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:
@@ -111,42 +91,27 @@ with gr.Blocks() as demo:
                 value="https://id2223kth.github.io/schedule/",
                 label="Table URL",
             )
-            table_container = gr.Column()  # Container for table and buttons
+            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):
-            chatbot = gr.Chatbot()
-            msg = gr.Textbox(label="Message")
-            system_message = gr.Textbox(
-                value="Student class preparation companion.",
-                label="System message"
+            chatbot = gr.ChatInterface(
+                respond,
+                additional_inputs=[
+                    gr.Textbox(value="Student class preparation companion.", label="System message"),
+                ],
             )
 
-            with gr.Row():
-                submit = gr.Button("Submit")
-                clear = gr.Button("Clear")
-
-    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])
+            buttons_container = gr.Column(visible=True)
+
+            # 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()
+    demo.launch()