Update space

app.py

@@ -8,7 +8,7 @@ client = InferenceClient("meta-llama/Llama-3.2-3B-Instruct")
 
 data = []  # Global variable to store 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:
@@ -47,14 +46,14 @@ def extract_table(url):
             if len(cells) >= 2:
                 data.append({"Index": i, "Date": cells[0].text.strip()[:10], "Topic": cells[1].text.strip()})
 
-        # Generate HTML table with Prepare buttons
+        # Generate HTML table with buttons that trigger Gradio events
         html_rows = ""
         for row in data:
             html_rows += f"""
                 <tr>
                     <td>{row['Date']}</td>
                     <td>{row['Topic']}</td>
-                    <td><button onclick="prepareTopic({row['Index']})">Prepare</button></td>
+                    <td><button class="gr-button gr-button-lg" onclick="handle_topic_click({row['Index']})">Prepare</button></td>
                 </tr>
             """
         html_table = f"""
@@ -67,9 +66,13 @@ def extract_table(url):
                 {html_rows}
             </table>
             <script>
-                function prepareTopic(index) {{
-                    document.getElementById('prepareInput').value = index;
-                    document.getElementById('prepareSubmit').click();
+                function handle_topic_click(index) {{
+                    const prepareInput = document.querySelector('#prepareInput textarea');
+                    if (prepareInput) {{
+                        prepareInput.value = index;
+                        prepareInput.dispatchEvent(new Event('input'));
+                        document.querySelector('#prepareSubmit').click();
+                    }}
                 }}
             </script>
         """
@@ -77,14 +80,18 @@ def extract_table(url):
     except Exception as e:
         return f"<p>Error: {str(e)}</p>"
 
-
-def handle_prepare(index):
+def handle_prepare(index, history):
     try:
         index = int(index)
-        topic = data[index]["Topic"]
-        return f"Prepare a 10-minute reading on what I should know before the class for the topic: {topic}"
+        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
     except Exception as e:
-        return f"<p>Error: {str(e)}</p>"
+        print(f"Error in handle_prepare: {e}")
+        return history or []
 
 # Gradio app
 with gr.Blocks() as demo:
@@ -93,19 +100,64 @@ with gr.Blocks() as demo:
             url_input = gr.Textbox(value="https://id2223kth.github.io/schedule/", label="Table URL")
             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])
+            
+            # Hidden components for prepare functionality
+            prepare_input = gr.Textbox(visible=False, elem_id="prepareInput")
+            prepare_submit = gr.Button("Prepare Submit", visible=False, elem_id="prepareSubmit")
 
         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"
             )
-            # Hidden input to handle prepare actions
-            prepare_input = gr.Textbox(visible=False, elem_id="prepareInput")
-            prepare_submit = gr.Button("Prepare Submit", visible=False, elem_id="prepareSubmit")
-            prepare_submit.click(fn=handle_prepare, inputs=[prepare_input], outputs=[chatbot])
+
+            with gr.Row():
+                submit = gr.Button("Submit")
+                clear = gr.Button("Clear")
+
+    # Event handlers
+    extract_btn.click(
+        fn=extract_table,
+        inputs=[url_input],
+        outputs=[table_output]
+    )
+
+    def user_message(message, history):
+        history = history or []
+        history.append((message, None))
+        return "", history
+
+    submit.click(
+        user_message,
+        inputs=[msg, chatbot],
+        outputs=[msg, chatbot]
+    ).then(
+        respond,
+        inputs=[msg, chatbot, system_message],
+        outputs=[chatbot]
+    )
+
+    # Handle prepare button clicks
+    prepare_submit.click(
+        handle_prepare,
+        inputs=[prepare_input, chatbot],
+        outputs=[chatbot]
+    ).then(
+        respond,
+        inputs=[
+            lambda x: f"Prepare a 10-minute reading for topic: {data[int(x)]['Topic'] if x.isdigit() and 0 <= int(x) < len(data) else ''}", 
+            chatbot,
+            system_message
+        ],
+        outputs=[chatbot]
+    )
+
+    def clear_chat():
+        return [], []
+    
+    clear.click(fn=clear_chat, outputs=[chatbot, msg])
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()