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")
 
@@ -33,6 +34,7 @@ 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()
@@ -41,10 +43,8 @@ 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 i, row in enumerate(rows[1:]):
+        for row in rows[1:]:  # Skip header row
             cells = row.find_all('td')
             if len(cells) >= 2:
                 data.append({
@@ -52,92 +52,36 @@ def extract_table(url):
                     'Topic': cells[1].text.strip(),
                 })
         
-        # 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;
-            }
-            .prepare-btn {
-                padding: 5px 10px;
-                cursor: pointer;
-                background-color: #0366d6;
-                color: white;
-                border: none;
-                border-radius: 4px;
-            }
-            .prepare-btn:hover {
-                background-color: #0056b3;
-            }
-        </style>
-        '''
-        
-        html += '<table class="dataframe">'
-        html += '<thead><tr><th>Date</th><th>Topic</th><th>Action</th></tr></thead>'
-        html += '<tbody>'
-        
-        for i, row in enumerate(data):
-            safe_topic = row['Topic'].replace("'", "\\'")  # Escape single quotes
-            html += f'''
-                <tr>
-                    <td>{row['Date']}</td>
-                    <td>{row['Topic']}</td>
-                    <td>
-                        <button class="prepare-btn" onclick="(function() {{
-                            const index = {i};
-                            const input = document.getElementById('prepare-index');
-                            input.value = index;
-                            const event = new Event('input');
-                            input.dispatchEvent(event);
-                            setTimeout(() => {{
-                                document.getElementById('prepare-trigger').click();
-                            }}, 100);
-                        }})()">
-                            Prepare
-                        </button>
-                    </td>
-                </tr>
-            '''
-        html += '</tbody></table>'
-        return html
-        
+        print(f"Extracted {len(data)} rows")  # Debug print
+        return create_table_html()  # Your existing HTML creation code
     except Exception as e:
-        print(f"Error in extract_table: {e}")  # Debug print
+        print(f"Error in extract_table: {e}")
         return f"<p>Error: {str(e)}</p>"
 
-def prepare_topic_message(index):
+def prepare_topic(selected_topic):
+    print(f"Preparing topic: {selected_topic}")  # Debug print
     try:
-        print(f"Received index: {index}")  # Debug print
-        if not index:
-            print("Empty index received")
-            return ""
-            
-        idx = int(index)
-        if 0 <= idx < len(data):
-            topic = data[idx]["Topic"]
-            date = data[idx]["Date"]
-            message = f"Please prepare a 10-minute reading guide for the topic '{topic}' scheduled for {date}"
-            print(f"Generated message: {message}")  # Debug print
-            return message
-        else:
-            print(f"Index {idx} out of range")
-            return ""
+        # 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
+        
+        print(f"Topic not found: {selected_topic}")
+        return "Error: Topic not found"
+    
     except Exception as e:
-        print(f"Error in prepare_topic_message: {e}")
-        return ""
+        print(f"Unexpected error in prepare_topic: {e}")
+        return "Error: Could not prepare topic"
 
 def add_text(history, text):
     history = history + [(text, None)]
@@ -153,6 +97,15 @@ 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 [], ""
 
@@ -167,10 +120,14 @@ with gr.Blocks() as demo:
             table_output = gr.HTML(label="Extracted Table")
             extract_btn = gr.Button("Extract Table")
             
-            # Hidden components for prepare functionality
-            with gr.Row(visible=False):
-                prepare_index = gr.Textbox(value="", elem_id="prepare-index")
-                prepare_trigger = gr.Button("Prepare", elem_id="prepare-trigger")
+            # Dropdown for selecting topic
+            topic_dropdown = gr.Dropdown(
+                label="Select Topic",
+                choices=[],
+                interactive=True,
+                allow_custom_value=True
+            )
+            prepare_btn = gr.Button("Prepare Topic")
             
         with gr.Column(scale=2):
             chatbot = gr.Chatbot()
@@ -186,17 +143,19 @@ 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]
     )
     
     # Prepare topic handler
-    prepare_trigger.click(
-        fn=prepare_topic_message,
-        inputs=[prepare_index],
-        outputs=[msg],
-        queue=False
+    prepare_btn.click(
+        fn=prepare_topic,
+        inputs=[topic_dropdown],
+        outputs=[msg]
     ).success(
         fn=add_text,
         inputs=[chatbot, msg],