Update space

app.py

@@ -71,44 +71,18 @@ def extract_table(url):
             .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 += '<thead><tr><th>Date</th><th>Topic</th></tr></thead>'
         html += '<tbody>'
         
-        for i, row in enumerate(data):
-            safe_topic = row['Topic'].replace("'", "\\'")  # Escape single quotes
+        for row in data:
             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>'
@@ -118,26 +92,28 @@ def extract_table(url):
         print(f"Error in extract_table: {e}")  # Debug print
         return f"<p>Error: {str(e)}</p>"
 
-def prepare_topic_message(index):
+def prepare_topic(index):
+    print(f"Preparing topic with index: {index}")  # 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"]
+        # Convert index to integer
+        index = int(index)
+        
+        # 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 message: {message}")  # Debug print
+            print(f"Generated preparation message: {message}")  # Debug print
             return message
         else:
-            print(f"Index {idx} out of range")
-            return ""
+            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"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,11 +129,9 @@ def generate_response(history, system_message):
         history[-1] = (history[-1][0], response)
         yield history
 
-
 def clear_chat():
     return [], ""
 
-# Gradio app
 # Gradio app
 with gr.Blocks() as demo:
     with gr.Row():
@@ -169,10 +143,13 @@ with gr.Blocks() as demo:
             table_output = gr.HTML(label="Extracted Table")
             extract_btn = gr.Button("Extract Table")
             
-            # Replace this section
-            with gr.Row(visible=False):  # Hide these components
-                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
+            )
+            prepare_btn = gr.Button("Prepare Topic")
             
         with gr.Column(scale=2):
             chatbot = gr.Chatbot()
@@ -191,14 +168,16 @@ with gr.Blocks() as demo:
         fn=extract_table,
         inputs=[url_input],
         outputs=[table_output]
+    ).success(
+        fn=lambda: [f"{row['Topic']} ({row['Date']})" for row in data],
+        outputs=[topic_dropdown]
     )
     
-    # Replace the existing prepare trigger handler with this one
-    prepare_trigger.click(
-        fn=prepare_topic_message,
-        inputs=[prepare_index],
-        outputs=[msg],
-        queue=False
+    # Prepare topic handler
+    prepare_btn.click(
+        fn=prepare_topic,
+        inputs=[topic_dropdown],
+        outputs=[msg]
     ).success(
         fn=add_text,
         inputs=[chatbot, msg],
@@ -240,6 +219,5 @@ with gr.Blocks() as demo:
     # Clear button handler
     clear.click(fn=clear_chat, outputs=[chatbot, msg])
 
-    
 if __name__ == "__main__":
     demo.launch()