Update app.py

app.py

@@ -3,7 +3,6 @@
 import gradio as gr
 from bs4 import BeautifulSoup
 import requests
-from transformers import pipeline
 from sentence_transformers import SentenceTransformer
 import faiss
 import numpy as np
@@ -100,37 +99,43 @@ def vectorize_and_index(bookmarks):
     return faiss_idx, embeddings
 
 def display_bookmarks():
-    data = []
-    classes = []
+    cards = ''
     for i, bookmark in enumerate(bookmarks):
+        index = i + 1  # Start index at 1
         status = "Dead Link" if bookmark.get('dead_link') else "Active"
-        row_class = "dead-link" if bookmark.get('dead_link') else ""
-        data.append({
-            'Index': i,
-            'Title': bookmark['title'],
-            'URL': f"<a href='{bookmark['url']}' target='_blank'>{bookmark['url']}</a>",
-            'Status': status,
-            'ETag': bookmark.get('etag', 'N/A'),
-            'Summary': bookmark.get('summary', ''),
-        })
-        classes.append(row_class)
-    df = pd.DataFrame(data)
-    # Return HTML with styles
-    return df.to_html(escape=False, index=False)
+        card_class = "card dead-link" if bookmark.get('dead_link') else "card"
+        title = bookmark['title']
+        url = bookmark['url']
+        etag = bookmark.get('etag', 'N/A')
+        summary = bookmark.get('summary', '')
+
+        card_html = f'''
+        <div class="{card_class}">
+            <div class="card-content">
+                <h3>{index}. {title}</h3>
+                <p><strong>URL:</strong> <a href="{url}" target="_blank">{url}</a></p>
+                <p><strong>Status:</strong> {status}</p>
+                <p><strong>ETag:</strong> {etag}</p>
+                <p><strong>Summary:</strong> {summary}</p>
+            </div>
+        </div>
+        '''
+        cards += card_html
+    return cards
 
 def process_uploaded_file(file):
     global bookmarks, faiss_index
     if file is None:
-        return "Please upload a bookmarks HTML file.", pd.DataFrame()
+        return "Please upload a bookmarks HTML file.", ''
     try:
         file_content = file.decode('utf-8')
     except UnicodeDecodeError:
-        return "Error decoding the file. Please ensure it's a valid HTML file.", pd.DataFrame()
+        return "Error decoding the file. Please ensure it's a valid HTML file.", ''
 
     bookmarks = parse_bookmarks(file_content)
 
     if not bookmarks:
-        return "No bookmarks found in the uploaded file.", pd.DataFrame()
+        return "No bookmarks found in the uploaded file.", ''
 
     # Asynchronously fetch bookmark info
     asyncio.run(process_bookmarks_async(bookmarks))
@@ -157,13 +162,14 @@ def chatbot_response(user_query):
     for idx in I[0]:
         if idx < len(bookmarks):
             bookmark = bookmarks[idx]
-            response += f"Title: {bookmark['title']}\nURL: {bookmark['url']}\nSummary: {bookmark['summary']}\n\n"
+            index = bookmarks.index(bookmark) + 1  # Start index at 1
+            response += f"{index}. Title: {bookmark['title']}\nURL: {bookmark['url']}\nSummary: {bookmark['summary']}\n\n"
     return response.strip()
 
 def edit_bookmark(bookmark_idx, new_title, new_url):
     global faiss_index
     try:
-        bookmark_idx = int(bookmark_idx)
+        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[bookmark_idx]['title'] = new_title
@@ -182,7 +188,7 @@ def edit_bookmark(bookmark_idx, new_title, new_url):
 def delete_bookmark(bookmark_idx):
     global faiss_index
     try:
-        bookmark_idx = int(bookmark_idx)
+        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)
@@ -199,22 +205,22 @@ def delete_bookmark(bookmark_idx):
 
 def build_app():
     with gr.Blocks(css="app.css") as demo:
-        gr.Markdown("<h1 style='text-align: center;'>Bookmark Manager App</h1>")
+        gr.Markdown("<h1>Bookmark Manager App</h1>")
 
         with gr.Tab("Upload and Process Bookmarks"):
             upload = gr.File(label="Upload Bookmarks HTML File", type='binary')
             process_button = gr.Button("Process Bookmarks")
             output_text = gr.Textbox(label="Output")
-            bookmark_table = gr.HTML(label="Bookmarks")
+            bookmark_display = gr.HTML(label="Bookmarks")
 
-            def update_bookmark_table(file):
-                message, html_table = process_uploaded_file(file)
-                return message, html_table
+            def update_bookmark_display(file):
+                message, html_content = process_uploaded_file(file)
+                return message, html_content
 
             process_button.click(
-                update_bookmark_table,
+                update_bookmark_display,
                 inputs=upload,
-                outputs=[output_text, bookmark_table]
+                outputs=[output_text, bookmark_display]
             )
 
         with gr.Tab("Chat with Bookmarks"):
@@ -230,41 +236,41 @@ def build_app():
 
         with gr.Tab("Manage Bookmarks"):
             manage_output = gr.Textbox(label="Manage Output")
-            bookmark_table_manage = gr.HTML(label="Bookmarks")
+            bookmark_display_manage = gr.HTML(label="Bookmarks")
             refresh_button = gr.Button("Refresh Bookmark List")
 
             with gr.Row():
-                index_input = gr.Number(label="Bookmark Index")
+                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")
 
             edit_button = gr.Button("Edit Bookmark")
             delete_button = gr.Button("Delete Bookmark")
 
-            def update_manage_table():
-                html_table = display_bookmarks()
-                return html_table
+            def update_manage_display():
+                html_content = display_bookmarks()
+                return html_content
 
             refresh_button.click(
-                update_manage_table,
+                update_manage_display,
                 inputs=None,
-                outputs=bookmark_table_manage
+                outputs=bookmark_display_manage
             )
 
             edit_button.click(
                 edit_bookmark,
                 inputs=[index_input, new_title_input, new_url_input],
-                outputs=[manage_output, bookmark_table_manage]
+                outputs=[manage_output, bookmark_display_manage]
             )
 
             delete_button.click(
                 delete_bookmark,
                 inputs=index_input,
-                outputs=[manage_output, bookmark_table_manage]
+                outputs=[manage_output, bookmark_display_manage]
             )
 
-            # Initial load of the bookmarks table
-            bookmark_table_manage.value = update_manage_table()
+            # Initial load of the bookmarks display
+            bookmark_display_manage.value = update_manage_display()
 
     demo.launch()