Update space

app.py

@@ -3,6 +3,7 @@ from huggingface_hub import InferenceClient
 import requests
 from bs4 import BeautifulSoup
 import pandas as pd
+import ast
 
 client = InferenceClient("meta-llama/Llama-3.2-3B-Instruct")
 
@@ -39,7 +40,7 @@ def extract_table(url):
         soup = BeautifulSoup(response.text, 'html.parser')
         table = soup.find('table')
         if not table:
-            return "<p>No table found on page</p>"
+            return "<p>No table found on page</p>", []
         
         # Clear existing data
         data = []
@@ -52,7 +53,7 @@ def extract_table(url):
                     'Topic': cells[1].text.strip(),
                 })
         
-        # Create HTML table with prepare buttons
+        # Create HTML table
         html = '''
         <style>
             .dataframe {
@@ -86,31 +87,38 @@ def extract_table(url):
                 </tr>
             '''
         html += '</tbody></table>'
-        return html
+
+        # Generate choices for dropdown
+        choices = [f"{row['Topic']} ({row['Date']})" for row in data]
+        return html, choices
         
     except Exception as e:
-        print(f"Error in extract_table: {e}")  # Debug print
-        return f"<p>Error: {str(e)}</p>"
+        print(f"Error in extract_table: {e}")
+        return f"<p>Error: {str(e)}</p>", []
 
-def prepare_topic(index):
-    print(f"Preparing topic with index: {index}")  # Debug print
+def prepare_topic(selected_topic):
+    print(f"Preparing topic: {selected_topic}")  # Debug print
     try:
-        # Convert index to integer
-        index = int(index)
+        if not selected_topic:
+            return "Please select a topic first"
+            
+        # Handle potential list or string input
+        if isinstance(selected_topic, list):
+            selected_topic = selected_topic[0] if selected_topic else ""
+        
+        # Find the index of the selected topic
+        for row in data:
+            full_topic = f"{row['Topic']} ({row['Date']})"
+            if full_topic == selected_topic:
+                topic = row["Topic"]
+                date = row["Date"]
+                message = f"Please prepare a 10-minute reading guide for the topic '{topic}' scheduled for {date}"
+                print(f"Generated preparation message: {message}")  # Debug print
+                return message
         
-        # Check if index is valid
-        if 0 <= index < len(data):
-            topic = data[index]["Topic"]
-            date = data[index]["Date"]
-            message = f"Please prepare a 10-minute reading guide for the topic '{topic}' scheduled for {date}"
-            print(f"Generated preparation message: {message}")  # Debug print
-            return message
-        else:
-            print(f"Invalid index: {index}. Total data length: {len(data)}")
-            return "Error: Invalid topic selection"
-    except ValueError:
-        print(f"Could not convert {index} to integer")
-        return "Error: Invalid topic selection"
+        print(f"Topic not found: {selected_topic}")
+        return "Error: Topic not found"
+    
     except Exception as e:
         print(f"Unexpected error in prepare_topic: {e}")
         return "Error: Could not prepare topic"
@@ -147,7 +155,8 @@ with gr.Blocks() as demo:
             topic_dropdown = gr.Dropdown(
                 label="Select Topic",
                 choices=[],
-                interactive=True
+                interactive=True,
+                value=None
             )
             prepare_btn = gr.Button("Prepare Topic")
             
@@ -164,13 +173,15 @@ with gr.Blocks() as demo:
                 clear = gr.Button("Clear")
     
     # Event handlers
+    # Extract table and update dropdown
+    def update_interface(url):
+        html, choices = extract_table(url)
+        return html, gr.Dropdown(choices=choices)
+
     extract_btn.click(
-        fn=extract_table,
+        fn=update_interface,
         inputs=[url_input],
-        outputs=[table_output]
-    ).success(
-        fn=lambda: [f"{row['Topic']} ({row['Date']})" for row in data],
-        outputs=[topic_dropdown]
+        outputs=[table_output, topic_dropdown]
     )
     
     # Prepare topic handler
@@ -189,7 +200,7 @@ with gr.Blocks() as demo:
         outputs=[chatbot]
     )
     
-    # Rest of your event handlers remain the same
+    # Message submit handlers
     msg.submit(
         fn=add_text,
         inputs=[chatbot, msg],