Update app.py

app.py

@@ -80,4 +80,136 @@ def vectorize_and_index(bookmarks):
     dimension = embeddings.shape[1]
     faiss_idx = faiss.IndexFlatL2(dimension)
     faiss_idx.add(np.array(embeddings))
-    return f
+    return faiss_idx, embeddings
+
+def process_uploaded_file(file):
+    global bookmarks, faiss_index
+    if file is None:
+        return "Please upload a bookmarks HTML file."
+
+    # Since 'file' is now bytes, decode it directly
+    file_content = file.decode('utf-8')
+    bookmarks = parse_bookmarks(file_content)
+
+    for bookmark in bookmarks:
+        fetch_url_info(bookmark)
+        generate_summary(bookmark)
+
+    faiss_index, embeddings = vectorize_and_index(bookmarks)
+    return f"Successfully processed {len(bookmarks)} bookmarks."
+
+def chatbot_response(user_query):
+    if faiss_index is None or not bookmarks:
+        return "No bookmarks available. Please upload and process your bookmarks first."
+
+    # Vectorize user query
+    user_embedding = embedding_model.encode([user_query])
+    D, I = faiss_index.search(np.array(user_embedding), k=5)  # Retrieve top 5 matches
+
+    # Generate response
+    response = ""
+    for idx in I[0]:
+        bookmark = bookmarks[idx]
+        response += f"Title: {bookmark['title']}\nURL: {bookmark['url']}\nSummary: {bookmark['summary']}\n\n"
+    return response.strip()
+
+def display_bookmarks():
+    bookmark_list = []
+    for i, bookmark in enumerate(bookmarks):
+        status = "Dead Link" if bookmark.get('dead_link') else "Active"
+        bookmark_list.append([i, bookmark['title'], bookmark['url'], status])
+    return bookmark_list
+
+def edit_bookmark(bookmark_idx, new_title, new_url):
+    global faiss_index  # Reference the global faiss_index variable
+    try:
+        bookmark_idx = int(bookmark_idx)
+        bookmarks[bookmark_idx]['title'] = new_title
+        bookmarks[bookmark_idx]['url'] = new_url
+        fetch_url_info(bookmarks[bookmark_idx])
+        generate_summary(bookmarks[bookmark_idx])
+        # Rebuild the FAISS index
+        faiss_index, embeddings = vectorize_and_index(bookmarks)
+        return "Bookmark updated successfully."
+    except Exception as e:
+        return f"Error: {str(e)}"
+
+def delete_bookmark(bookmark_idx):
+    global faiss_index  # Reference the global faiss_index variable
+    try:
+        bookmark_idx = int(bookmark_idx)
+        bookmarks.pop(bookmark_idx)
+        # Rebuild the FAISS index
+        if bookmarks:
+            faiss_index, embeddings = vectorize_and_index(bookmarks)
+        else:
+            faiss_index = None  # No bookmarks left
+        return "Bookmark deleted successfully."
+    except Exception as e:
+        return f"Error: {str(e)}"
+
+def build_app():
+    with gr.Blocks() as demo:
+        gr.Markdown("# Bookmark Manager App")
+
+        with gr.Tab("Upload and Process Bookmarks"):
+            upload = gr.File(label="Upload Bookmarks HTML File", type='bytes')  # Updated here
+            process_button = gr.Button("Process Bookmarks")
+            output_text = gr.Textbox(label="Output")
+
+            process_button.click(
+                process_uploaded_file,
+                inputs=upload,
+                outputs=output_text
+            )
+
+        with gr.Tab("Chat with Bookmarks"):
+            user_input = gr.Textbox(label="Ask about your bookmarks")
+            chat_output = gr.Textbox(label="Chatbot Response")
+            chat_button = gr.Button("Send")
+
+            chat_button.click(
+                chatbot_response,
+                inputs=user_input,
+                outputs=chat_output
+            )
+
+        with gr.Tab("Manage Bookmarks"):
+            bookmark_table = gr.Dataframe(
+                headers=["Index", "Title", "URL", "Status"],
+                datatype=["number", "str", "str", "str"],
+                interactive=False
+            )
+            refresh_button = gr.Button("Refresh Bookmark List")
+
+            with gr.Row():
+                index_input = gr.Number(label="Bookmark Index")
+                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")
+            manage_output = gr.Textbox(label="Manage Output")
+
+            refresh_button.click(
+                display_bookmarks,
+                inputs=None,
+                outputs=bookmark_table
+            )
+
+            edit_button.click(
+                edit_bookmark,
+                inputs=[index_input, new_title_input, new_url_input],
+                outputs=manage_output
+            )
+
+            delete_button.click(
+                delete_bookmark,
+                inputs=index_input,
+                outputs=manage_output
+            )
+
+    demo.launch()
+
+if __name__ == "__main__":
+    build_app()