Update app.py

app.py

@@ -261,7 +261,7 @@ def generate_contextual_summary(context):
     Type: {page_type}
     Description: {context['description']}
     Keywords: {context['keywords']}
-    
+
     Additional Content:
     {context['content'][:3000]}
 
@@ -574,6 +574,60 @@ def display_bookmarks():
     logger.info("HTML display generated")
     return display_html
 
+def assign_category(bookmark):
+    """
+    Assign a category to a bookmark based on its title or summary.
+    This is a simple implementation and can be enhanced with more sophisticated methods.
+    """
+    title = bookmark.get('title', '').lower()
+    summary = bookmark.get('summary', '').lower()
+    
+    # Simple keyword-based categorization
+    if any(keyword in title or keyword in summary for keyword in ['facebook', 'twitter', 'instagram']):
+        bookmark['category'] = 'Social Media'
+    elif any(keyword in title or keyword in summary for keyword in ['news', 'media', 'huffpost', 'times']):
+        bookmark['category'] = 'News and Media'
+    elif any(keyword in title or keyword in summary for keyword in ['course', 'learning', 'education']):
+        bookmark['category'] = 'Education and Learning'
+    elif any(keyword in title or keyword in summary for keyword in ['movie', 'music', 'audio', 'video']):
+        bookmark['category'] = 'Entertainment'
+    elif any(keyword in title or keyword in summary for keyword in ['shop', 'e-commerce', 'buy', 'purchase']):
+        bookmark['category'] = 'Shopping and E-commerce'
+    elif any(keyword in title or keyword in summary for keyword in ['finance', 'banking', 'investment']):
+        bookmark['category'] = 'Finance and Banking'
+    elif any(keyword in title or keyword in summary for keyword in ['tech', 'technology', 'software']):
+        bookmark['category'] = 'Technology'
+    elif any(keyword in title or keyword in summary for keyword in ['health', 'fitness', 'wellness']):
+        bookmark['category'] = 'Health and Fitness'
+    elif any(keyword in title or keyword in summary for keyword in ['travel', 'tourism', 'flight', 'hotel']):
+        bookmark['category'] = 'Travel and Tourism'
+    elif any(keyword in title or keyword in summary for keyword in ['recipe', 'food', 'cooking']):
+        bookmark['category'] = 'Food and Recipes'
+    elif any(keyword in title or keyword in summary for keyword in ['sport', 'game', 'fitness']):
+        bookmark['category'] = 'Sports'
+    elif any(keyword in title or keyword in summary for keyword in ['art', 'culture', 'museum']):
+        bookmark['category'] = 'Arts and Culture'
+    elif any(keyword in title or keyword in summary for keyword in ['gov', 'government', 'politics']):
+        bookmark['category'] = 'Government and Politics'
+    elif any(keyword in title or keyword in summary for keyword in ['business', 'economy', 'market']):
+        bookmark['category'] = 'Business and Economy'
+    elif any(keyword in title or keyword in summary for keyword in ['science', 'research', 'study']):
+        bookmark['category'] = 'Science and Research'
+    elif any(keyword in title or keyword in summary for keyword in ['blog', 'journal']):
+        bookmark['category'] = 'Personal Blogs and Journals'
+    elif any(keyword in title or keyword in summary for keyword in ['job', 'career', 'employment']):
+        bookmark['category'] = 'Job Search and Careers'
+    elif any(keyword in title or keyword in summary for keyword in ['audio', 'music']):
+        bookmark['category'] = 'Music and Audio'
+    elif any(keyword in title or keyword in summary for keyword in ['video', 'movie']):
+        bookmark['category'] = 'Videos and Movies'
+    elif any(keyword in title or keyword in summary for keyword in ['reference', 'knowledge', 'wiki']):
+        bookmark['category'] = 'Reference and Knowledge Bases'
+    elif bookmark.get('dead_link'):
+        bookmark['category'] = 'Dead Link'
+    else:
+        bookmark['category'] = 'Uncategorized'
+
 def process_uploaded_file(file):
     """
     Process the uploaded bookmarks file with enhanced error handling and user feedback.
@@ -749,7 +803,21 @@ def export_bookmarks():
         href = f'data:text/html;base64,{b64}'
         
         logger.info("Bookmarks exported successfully")
-        return f'<a href="{href}" download="bookmarks.html">💾 Download Exported Bookmarks</a>'
+        return f'''
+        <div style="text-align: center;">
+            <a href="{href}" 
+               download="bookmarks.html" 
+               style="display: inline-block; 
+                      padding: 10px 20px; 
+                      background-color: #4CAF50; 
+                      color: white; 
+                      text-decoration: none; 
+                      border-radius: 5px; 
+                      margin: 10px;">
+                💾 Download Exported Bookmarks
+            </a>
+        </div>
+        '''
     except Exception as e:
         logger.error(f"Error exporting bookmarks: {e}")
         return "⚠️ Error exporting bookmarks."
@@ -793,7 +861,7 @@ def chatbot_response(user_query):
         User Query: {user_query}
 
         Relevant Bookmarks:
-        {'\n\n'.join(bookmark_descriptions)}
+        {'\\n\\n'.join(bookmark_descriptions)}
 
         Please provide a helpful response that:
         1. Identifies the most relevant bookmarks for the query
@@ -829,7 +897,64 @@ def build_app():
     try:
         logger.info("Building Gradio app")
         with gr.Blocks(css="app.css") as demo:
-            # ... [Rest of the UI code remains the same as before] ...
+            gr.Markdown("# 📚 Bookmark Manager")
+
+            with gr.Row():
+                with gr.Column():
+                    file_input = gr.File(label="Upload Bookmarks HTML File", file_types=["file"])
+                    process_button = gr.Button("Process Bookmarks")
+                    process_message = gr.Markdown("")
+                    
+                    category_dropdown = gr.Dropdown(choices=CATEGORIES, label="New Category")
+                    edit_button = gr.Button("Edit Selected Bookmarks Category")
+                    
+                    delete_button = gr.Button("Delete Selected Bookmarks")
+                    export_button = gr.Button("Export Bookmarks")
+                    
+                with gr.Column():
+                    bookmarks_display = gr.HTML(label="Bookmarks")
+            
+            with gr.Row():
+                chatbot_input = gr.Textbox(label="Ask about your bookmarks", placeholder="Enter your query here...")
+                chatbot_output = gr.Textbox(label="Chatbot Response", interactive=False)
+            
+            # Processing File
+            process_button.click(
+                fn=process_uploaded_file,
+                inputs=file_input,
+                outputs=[process_message, bookmarks_display, gr.Dropdown.update(), bookmarks_display]
+            )
+            
+            # Deleting Bookmarks
+            delete_button.click(
+                fn=delete_selected_bookmarks,
+                inputs=gr.CheckboxGroup(label="Select Bookmarks to Delete", choices=[]),
+                outputs=[process_message, gr.Dropdown.update(), bookmarks_display]
+            )
+            
+            # Editing Categories
+            edit_button.click(
+                fn=edit_selected_bookmarks_category,
+                inputs=[
+                    gr.CheckboxGroup(label="Select Bookmarks to Edit", choices=[]),
+                    category_dropdown
+                ],
+                outputs=[process_message, gr.Dropdown.update(), bookmarks_display]
+            )
+            
+            # Exporting Bookmarks
+            export_button.click(
+                fn=export_bookmarks,
+                inputs=None,
+                outputs=gr.HTML(label="Export")
+            )
+            
+            # Chatbot
+            chatbot_input.submit(
+                fn=chatbot_response,
+                inputs=chatbot_input,
+                outputs=chatbot_output
+            )
 
         logger.info("Launching Gradio app")
         demo.launch(debug=True)
@@ -838,4 +963,4 @@ def build_app():
         print(f"Error building the app: {e}")
 
 if __name__ == "__main__":
-    build_app()
+    build_app()