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, temperature, top_p):
+def respond(message, history, system_message, max_tokens=2048, temperature=0.7, top_p=0.9):
     messages = [{"role": "system", "content": system_message}]
     for val in history:
         if val[0]:
@@ -28,7 +28,6 @@ def respond(message, history, system_message, max_tokens, temperature, top_p):
         response += token
         yield response
 
-
 def extract_table(url):
     global data
     try:
@@ -40,14 +39,18 @@ def extract_table(url):
             return "<p>No table found on page</p>"
 
         # Extract data
-        data = []
+        data.clear()  # Clear existing 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()})
+                data.append({
+                    "Index": i,
+                    "Date": cells[0].text.strip()[:10],
+                    "Topic": cells[1].text.strip()
+                })
 
-        # Generate HTML table without buttons
+        # Generate HTML table
         html_rows = ""
         for row in data:
             html_rows += f"""
@@ -67,21 +70,15 @@ def extract_table(url):
         """
         return html_table
     except Exception as e:
-        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
+        return f"<p>Error: {str(e)}</p>"
 
+def handle_prepare(index, chatbot):
+    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 = chatbot.value if chatbot.value else []
+        history.append((prompt, None))
+        return history
 
 # Gradio App
 with gr.Blocks() as demo:
@@ -94,24 +91,60 @@ with gr.Blocks() as demo:
             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.ChatInterface(
-                respond,
-                additional_inputs=[
-                    gr.Textbox(value="Student class preparation companion.", label="System message"),
-                ],
+            chatbot = gr.Chatbot()
+            msg = gr.Textbox(label="Message")
+            system_message = gr.Textbox(
+                value="Student class preparation companion.",
+                label="System message"
             )
 
-            buttons_container = gr.Column(visible=True)
+            with gr.Row():
+                submit = gr.Button("Submit")
+                clear = gr.Button("Clear")
 
-            # Dynamically add buttons after table extraction
-            def update_buttons():
-                buttons = create_buttons()
-                return gr.update(components=buttons)
+            # Container for dynamic buttons
+            buttons_container = gr.Column()
 
-            extract_btn.click(fn=update_buttons, inputs=[], outputs=[buttons_container])
+            def clear_chat():
+                return [], []
+
+            def chat(message, history, system_msg):
+                return "", history + [[message, None]]
+
+            submit.click(chat, [msg, chatbot, system_message], [msg, chatbot]).then(
+                respond, [chatbot.value[-1][0], chatbot.value, system_message], chatbot
+            )
+            clear.click(clear_chat, None, [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=handle_prepare,
+                            inputs=[gr.Number(value=i, visible=False), chatbot],
+                            outputs=[chatbot]
+                        ).then(
+                            respond,
+                            inputs=[
+                                lambda: f"Prepare a 10-minute reading for: {row['Topic']}",
+                                lambda: chatbot.value,
+                                system_message
+                            ],
+                            outputs=chatbot
+                        )
+
+            extract_btn.click(
+                fn=extract_table,
+                inputs=[url_input],
+                outputs=[table_output]
+            ).then(
+                fn=update_buttons,
+                inputs=None,
+                outputs=None
+            )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()