Update space

app.py

@@ -9,26 +9,15 @@ client = InferenceClient("meta-llama/Llama-3.2-3B-Instruct")
 # Global data store for the table
 data = []
 
-def respond(
-    message,
-    history: list[tuple[str, str]],
-    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]:
             messages.append({"role": "user", "content": val[0]})
         if val[1]:
             messages.append({"role": "assistant", "content": val[1]})
-
     messages.append({"role": "user", "content": message})
-
     response = ""
-
     for message in client.chat_completion(
         messages,
         max_tokens=max_tokens,
@@ -63,9 +52,7 @@ def extract_table(url):
         
         # Create HTML table with prepare buttons
         html = '<table class="dataframe">'
-        # Add header
         html += '<thead><tr><th>Date</th><th>Topic</th><th>Action</th></tr></thead>'
-        # Add rows with prepare buttons
         html += '<tbody>'
         for i, row in enumerate(data):
             html += f'''
@@ -75,6 +62,7 @@ def extract_table(url):
                     <td>
                         <button onclick='
                             document.getElementById("prepare-topic").value = "{i}";
+                            document.getElementById("prepare-topic").dispatchEvent(new Event("input"));
                             document.getElementById("prepare-button").click();
                         '>
                             Prepare
@@ -87,11 +75,27 @@ def extract_table(url):
     except Exception as e:
         return f"<p>Error: {str(e)}</p>"
 
+def user_message(message, history):
+    history = history or []
+    history.append((message, None))
+    return "", history
+
+def handle_prepare(index, history):
+    try:
+        index = int(index)
+        if 0 <= index < len(data):
+            topic = data[index]["Topic"]
+            message = f"Prepare a 10-minute reading for the topic: {topic}"
+            history = history or []
+            history.append((message, None))
+            return history
+    except:
+        return history or []
+
 def clear_chat():
-    # This function will be called when prepare button is clicked
-    # For now, it just returns an empty chat history
-    return []
+    return [], ""
 
+# Gradio app
 with gr.Blocks() as demo:
     with gr.Row():
         with gr.Column(scale=1):
@@ -107,12 +111,16 @@ with gr.Blocks() as demo:
             prepare_button = gr.Button("Prepare", visible=False, elem_id="prepare-button")
             
         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"
             )
+
+            with gr.Row():
+                submit = gr.Button("Submit")
+                clear = gr.Button("Clear")
     
     # Event handlers
     extract_btn.click(
@@ -121,12 +129,30 @@ with gr.Blocks() as demo:
         outputs=[table_output]
     )
     
-    # When prepare button is clicked, clear the chat
+    # Submit button handler
+    submit.click(
+        fn=user_message,
+        inputs=[msg, chatbot],
+        outputs=[msg, chatbot]
+    ).then(
+        fn=respond,
+        inputs=[msg, chatbot, system_message],
+        outputs=[chatbot]
+    )
+    
+    # Prepare button handler
     prepare_button.click(
-        fn=clear_chat,
-        inputs=None,
+        fn=handle_prepare,
+        inputs=[prepare_topic, chatbot],
+        outputs=[chatbot]
+    ).then(
+        fn=respond,
+        inputs=[prepare_topic, chatbot, system_message],
         outputs=[chatbot]
     )
+    
+    # Clear button handler
+    clear.click(fn=clear_chat, outputs=[chatbot, msg])
 
 if __name__ == "__main__":
     demo.launch()