Update app.py

app.py

@@ -9,6 +9,7 @@ import numpy as np
 import asyncio
 import aiohttp
 import re
+import base64
 
 # Initialize models and variables
 embedding_model = SentenceTransformer('all-MiniLM-L6-v2')
@@ -115,6 +116,10 @@ def generate_summary(bookmark):
     return bookmark
 
 def assign_category(bookmark):
+    if bookmark.get('dead_link'):
+        bookmark['category'] = 'Dead Link'
+        return bookmark
+
     summary = bookmark.get('summary', '').lower()
     assigned_category = 'Uncategorized'
 
@@ -183,6 +188,7 @@ def display_bookmarks():
         card_html = f'''
         <div class="card" style="{card_style}">
             <div class="card-content">
+                <input type="checkbox" class="bookmark-checkbox" data-index="{index-1}">
                 <h3 style="{text_style}">{index}. {title}</h3>
                 <p style="{text_style}"><strong>Category:</strong> {category}</p>
                 <p style="{text_style}"><strong>URL:</strong> <a href="{url}" target="_blank" style="{text_style}">{url}</a></p>
@@ -239,7 +245,7 @@ def chatbot_response(user_query):
             response += f"{index}. Title: {bookmark['title']}\nURL: {bookmark['url']}\nCategory: {bookmark.get('category', 'Uncategorized')}\nSummary: {bookmark['summary']}\n\n"
     return response.strip()
 
-def edit_bookmark(bookmark_idx, new_title, new_url):
+def edit_bookmark(bookmark_idx, new_title, new_url, new_category):
     global faiss_index
     try:
         bookmark_idx = int(bookmark_idx) - 1  # Adjust index to match list (starting at 0)
@@ -247,10 +253,10 @@ def edit_bookmark(bookmark_idx, new_title, new_url):
             return "Invalid bookmark index.", display_bookmarks()
         bookmarks[bookmark_idx]['title'] = new_title
         bookmarks[bookmark_idx]['url'] = new_url
+        bookmarks[bookmark_idx]['category'] = new_category
         # Re-fetch bookmark info
         asyncio.run(process_bookmarks_async([bookmarks[bookmark_idx]]))
         generate_summary(bookmarks[bookmark_idx])
-        assign_category(bookmarks[bookmark_idx])
         # Rebuild the FAISS index
         faiss_index, embeddings = vectorize_and_index(bookmarks)
         message = "Bookmark updated successfully."
@@ -259,24 +265,43 @@ def edit_bookmark(bookmark_idx, new_title, new_url):
     except Exception as e:
         return f"Error: {str(e)}", display_bookmarks()
 
-def delete_bookmark(bookmark_idx):
+def delete_bookmarks(indices):
     global faiss_index
     try:
-        bookmark_idx = int(bookmark_idx) - 1  # Adjust index to match list (starting at 0)
-        if bookmark_idx < 0 or bookmark_idx >= len(bookmarks):
-            return "Invalid bookmark index.", display_bookmarks()
-        bookmarks.pop(bookmark_idx)
+        indices = sorted([int(idx) for idx in indices], reverse=True)
+        for idx in indices:
+            if 0 <= idx < len(bookmarks):
+                bookmarks.pop(idx)
         # Rebuild the FAISS index
         if bookmarks:
             faiss_index, embeddings = vectorize_and_index(bookmarks)
         else:
             faiss_index = None
-        message = "Bookmark deleted successfully."
+        message = "Selected bookmarks deleted successfully."
         updated_html = display_bookmarks()
         return message, updated_html
     except Exception as e:
         return f"Error: {str(e)}", display_bookmarks()
 
+def export_bookmarks():
+    if not bookmarks:
+        return None
+    # Create an HTML content similar to the imported bookmarks file
+    soup = BeautifulSoup("<!DOCTYPE NETSCAPE-Bookmark-file-1><Title>Bookmarks</Title><H1>Bookmarks</H1>", 'html.parser')
+    dl = soup.new_tag('DL')
+    for bookmark in bookmarks:
+        dt = soup.new_tag('DT')
+        a = soup.new_tag('A', href=bookmark['url'])
+        a.string = bookmark['title']
+        dt.append(a)
+        dl.append(dt)
+    soup.append(dl)
+    html_content = str(soup)
+    # Encode the HTML content to base64 for download
+    b64 = base64.b64encode(html_content.encode()).decode()
+    href = f'data:text/html;base64,{b64}'
+    return href
+
 def build_app():
     with gr.Blocks(css="app.css") as demo:
         gr.Markdown("<h1>Bookmark Manager App</h1>")
@@ -313,13 +338,19 @@ def build_app():
             bookmark_display_manage = gr.HTML(label="Bookmarks")
             refresh_button = gr.Button("Refresh Bookmark List")
 
+            select_all_checkbox = gr.Checkbox(label="Select All")
+            selected_indices = gr.Textbox(label="Selected Indices (comma-separated)", visible=False)
+
             with gr.Row():
                 index_input = gr.Number(label="Bookmark Index (Starting from 1)", precision=0)
                 new_title_input = gr.Textbox(label="New Title")
                 new_url_input = gr.Textbox(label="New URL")
+                new_category_input = gr.Dropdown(label="New Category", choices=CATEGORIES)
 
             edit_button = gr.Button("Edit Bookmark")
-            delete_button = gr.Button("Delete Bookmark")
+            delete_button = gr.Button("Delete Selected Bookmarks")
+            export_button = gr.Button("Export Bookmarks")
+            download_link = gr.HTML(label="Download Exported Bookmarks")
 
             def update_manage_display():
                 html_content = display_bookmarks()
@@ -333,19 +364,63 @@ def build_app():
 
             edit_button.click(
                 edit_bookmark,
-                inputs=[index_input, new_title_input, new_url_input],
+                inputs=[index_input, new_title_input, new_url_input, new_category_input],
                 outputs=[manage_output, bookmark_display_manage]
             )
 
             delete_button.click(
-                delete_bookmark,
-                inputs=index_input,
+                delete_bookmarks,
+                inputs=selected_indices,
                 outputs=[manage_output, bookmark_display_manage]
             )
 
+            def provide_download_link():
+                href = export_bookmarks()
+                if href:
+                    return f'<a href="{href}" download="bookmarks.html">Download Exported Bookmarks</a>'
+                else:
+                    return "No bookmarks to export."
+
+            export_button.click(
+                provide_download_link,
+                inputs=None,
+                outputs=download_link
+            )
+
             # Initial load of the bookmarks display
             bookmark_display_manage.value = update_manage_display()
 
+    # Include JavaScript to handle checkbox selection and Select All functionality
+    demo.load(None, None, None, _js="""
+    function() {
+        // Handle Select All checkbox
+        document.querySelector('input[type="checkbox"][label="Select All"]').addEventListener('change', function() {
+            var checkboxes = document.querySelectorAll('.bookmark-checkbox');
+            for (var i = 0; i < checkboxes.length; i++) {
+                checkboxes[i].checked = this.checked;
+            }
+        });
+
+        // Update selected indices
+        function updateSelectedIndices() {
+            var checkboxes = document.querySelectorAll('.bookmark-checkbox');
+            var indices = [];
+            for (var i = 0; i < checkboxes.length; i++) {
+                if (checkboxes[i].checked) {
+                    indices.push(checkboxes[i].getAttribute('data-index'));
+                }
+            }
+            document.querySelector('textarea[label="Selected Indices (comma-separated)"]').value = indices.join(',');
+        }
+
+        document.addEventListener('change', function(e) {
+            if (e.target && e.target.classList.contains('bookmark-checkbox')) {
+                updateSelectedIndices();
+            }
+        });
+    }
+    """)
+
     demo.launch()
 
 if __name__ == "__main__":