Update space

app.py

@@ -3,7 +3,6 @@ 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")
 
@@ -34,7 +33,6 @@ def respond(message, history, system_message):
 
 def extract_table(url):
     global data
-    data = []  # Clear existing data
     try:
         response = requests.get(url)
         response.raise_for_status()
@@ -43,8 +41,10 @@ def extract_table(url):
         if not table:
             return "<p>No table found on page</p>"
         
+        # Clear existing data
+        data = []
         rows = table.find_all('tr')
-        for row in rows[1:]:  # Skip header row
+        for i, row in enumerate(rows[1:]):
             cells = row.find_all('td')
             if len(cells) >= 2:
                 data.append({
@@ -52,33 +52,65 @@ def extract_table(url):
                     'Topic': cells[1].text.strip(),
                 })
         
-        print(f"Extracted {len(data)} rows")  # Debug print
-        return create_table_html()  # Your existing HTML creation code
+        # Create HTML table with prepare buttons
+        html = '''
+        <style>
+            .dataframe {
+                border-collapse: collapse;
+                width: 100%;
+                margin: 10px 0;
+            }
+            .dataframe th, .dataframe td {
+                border: 1px solid #ddd;
+                padding: 8px;
+                text-align: left;
+            }
+            .dataframe th {
+                background-color: #f6f8fa;
+            }
+            .dataframe tr:nth-child(even) {
+                background-color: #f9f9f9;
+            }
+        </style>
+        '''
+        
+        html += '<table class="dataframe">'
+        html += '<thead><tr><th>Date</th><th>Topic</th></tr></thead>'
+        html += '<tbody>'
+        
+        for row in data:
+            html += f'''
+                <tr>
+                    <td>{row['Date']}</td>
+                    <td>{row['Topic']}</td>
+                </tr>
+            '''
+        html += '</tbody></table>'
+        return html
+        
     except Exception as e:
-        print(f"Error in extract_table: {e}")
+        print(f"Error in extract_table: {e}")  # Debug print
         return f"<p>Error: {str(e)}</p>"
 
-def prepare_topic(selected_topic):
-    print(f"Preparing topic: {selected_topic}")  # Debug print
+def prepare_topic(index):
+    print(f"Preparing topic with index: {index}")  # Debug print
     try:
-        # Handle potential list or string input
-        if isinstance(selected_topic, list):
-            # If it's a list, take the first item
-            selected_topic = selected_topic[0] if selected_topic else ""
-        
-        # Find the index of the selected topic
-        for index, row in enumerate(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
+        # Convert index to integer
+        index = int(index)
         
-        print(f"Topic not found: {selected_topic}")
-        return "Error: Topic not found"
-    
+        # 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"
     except Exception as e:
         print(f"Unexpected error in prepare_topic: {e}")
         return "Error: Could not prepare topic"
@@ -97,15 +129,6 @@ def generate_response(history, system_message):
         history[-1] = (history[-1][0], response)
         yield history
 
-def update_dropdown():
-    try:
-        choices = [f"{row['Topic']} ({row['Date']})" for row in data]
-        print(f"Generated choices: {choices}")  # Debug print
-        return gr.Dropdown(choices=choices)
-    except Exception as e:
-        print(f"Error updating dropdown: {e}")
-        return gr.Dropdown(choices=[])
-    
 def clear_chat():
     return [], ""
 
@@ -124,8 +147,7 @@ with gr.Blocks() as demo:
             topic_dropdown = gr.Dropdown(
                 label="Select Topic",
                 choices=[],
-                interactive=True,
-                allow_custom_value=True
+                interactive=True
             )
             prepare_btn = gr.Button("Prepare Topic")
             
@@ -143,12 +165,12 @@ with gr.Blocks() as demo:
     
     # Event handlers
     extract_btn.click(
-    fn=extract_table,
-    inputs=[url_input],
-    outputs=[table_output]
+        fn=extract_table,
+        inputs=[url_input],
+        outputs=[table_output]
     ).success(
-    fn=lambda: [f"{row['Topic']} ({row['Date']})" for row in data],
-    outputs=[topic_dropdown]
+        fn=lambda: [f"{row['Topic']} ({row['Date']})" for row in data],
+        outputs=[topic_dropdown]
     )
     
     # Prepare topic handler