Update app.py

app.py

@@ -281,30 +281,60 @@ def process_uploaded_file(file, state_bookmarks):
     logger.info("Processing uploaded file")
     if file is None:
         logger.warning("No file uploaded")
-        return "⚠️ Please upload a bookmarks HTML file.", "", [], "", []
+        return (
+            "⚠️ Please upload a bookmarks HTML file.",
+            "",
+            [],
+            "",
+            state_bookmarks  # Return the unchanged state
+        )
 
     try:
         file_content = file.decode('utf-8')
     except UnicodeDecodeError as e:
         logger.error(f"Error decoding the file: {e}")
-        return "⚠️ Error decoding the file. Please ensure it's a valid HTML file.", "", [], "", state_bookmarks
+        return (
+            "⚠️ Error decoding the file. Please ensure it's a valid HTML file.",
+            "",
+            [],
+            "",
+            state_bookmarks  # Return the unchanged state
+        )
 
     try:
         bookmarks = parse_bookmarks(file_content)
     except Exception as e:
         logger.error(f"Error parsing bookmarks: {e}")
-        return "⚠️ Error parsing the bookmarks HTML file.", "", [], "", state_bookmarks
+        return (
+            "⚠️ Error parsing the bookmarks HTML file.",
+            "",
+            [],
+            "",
+            state_bookmarks  # Return the unchanged state
+        )
 
     if not bookmarks:
         logger.warning("No bookmarks found in the uploaded file")
-        return "⚠️ No bookmarks found in the uploaded file.", "", [], "", state_bookmarks
+        return (
+            "⚠️ No bookmarks found in the uploaded file.",
+            "",
+            [],
+            "",
+            state_bookmarks  # Return the unchanged state
+        )
 
     # Asynchronously fetch bookmark info
     try:
         asyncio.run(process_bookmarks_async(bookmarks))
     except Exception as e:
         logger.error(f"Error processing bookmarks asynchronously: {e}")
-        return "⚠️ Error processing bookmarks.", "", [], "", state_bookmarks
+        return (
+            "⚠️ Error processing bookmarks.",
+            "",
+            [],
+            "",
+            state_bookmarks  # Return the unchanged state
+        )
 
     # Generate summaries and assign categories
     for bookmark in bookmarks:
@@ -315,18 +345,32 @@ def process_uploaded_file(file, state_bookmarks):
         faiss_index, embeddings = vectorize_and_index(bookmarks)
     except Exception as e:
         logger.error(f"Error building FAISS index: {e}")
-        return "⚠️ Error building search index.", "", [], "", state_bookmarks
+        return (
+            "⚠️ Error building search index.",
+            "",
+            [],
+            "",
+            state_bookmarks  # Return the unchanged state
+        )
 
     message = f"✅ Successfully processed {len(bookmarks)} bookmarks."
     logger.info(message)
     bookmark_html = display_bookmarks(bookmarks)
 
+    # Update the shared state
+    updated_state = bookmarks.copy()
+
     # Prepare Manage Bookmarks tab outputs
     choices = [f"{i+1}. {bookmark['title']} (Category: {bookmark['category']})" for i, bookmark in enumerate(bookmarks)]
     bookmarks_html_manage = display_bookmarks(bookmarks)
 
-    # Return the updated state as the last output
-    return message, bookmark_html, choices, bookmarks_html_manage, bookmarks
+    return (
+        message,
+        bookmark_html,
+        choices,
+        bookmarks_html_manage,
+        updated_state  # Return the updated state
+    )
 
 # Delete selected bookmarks
 def delete_selected_bookmarks(selected_indices, state_bookmarks):
@@ -362,7 +406,7 @@ def delete_selected_bookmarks(selected_indices, state_bookmarks):
     bookmarks_html = display_bookmarks(bookmarks)
 
     # Update the shared state
-    state_bookmarks = bookmarks.copy()
+    updated_state = bookmarks.copy()
 
     # Update choices for selection
     choices = [f"{i+1}. {bookmark['title']} (Category: {bookmark['category']})" for i, bookmark in enumerate(bookmarks)]
@@ -372,10 +416,18 @@ def delete_selected_bookmarks(selected_indices, state_bookmarks):
 # Edit category of selected bookmarks
 def edit_selected_bookmarks_category(selected_indices, new_category, state_bookmarks):
     if not selected_indices:
-        return "⚠️ No bookmarks selected.", gr.update(choices=[f"{i+1}. {bookmark['title']} (Category: {bookmark['category']})" for i, bookmark in enumerate(state_bookmarks)]), display_bookmarks(state_bookmarks)
+        return (
+            "⚠️ No bookmarks selected.",
+            gr.update(choices=[f"{i+1}. {bookmark['title']} (Category: {bookmark['category']})" for i, bookmark in enumerate(state_bookmarks)]),
+            display_bookmarks(state_bookmarks)
+        )
 
     if not new_category:
-        return "⚠️ No new category selected.", gr.update(choices=[f"{i+1}. {bookmark['title']} (Category: {bookmark['category']})" for i, bookmark in enumerate(state_bookmarks)]), display_bookmarks(state_bookmarks)
+        return (
+            "⚠️ No new category selected.",
+            gr.update(choices=[f"{i+1}. {bookmark['title']} (Category: {bookmark['category']})" for i, bookmark in enumerate(state_bookmarks)]),
+            display_bookmarks(state_bookmarks)
+        )
 
     bookmarks = state_bookmarks.copy()
     indices = []
@@ -400,7 +452,7 @@ def edit_selected_bookmarks_category(selected_indices, new_category, state_bookm
     bookmarks_html = display_bookmarks(bookmarks)
 
     # Update the shared state
-    state_bookmarks = bookmarks.copy()
+    updated_state = bookmarks.copy()
 
     # Update choices for selection
     choices = [f"{i+1}. {bookmark['title']} (Category: {bookmark['category']})" for i, bookmark in enumerate(bookmarks)]
@@ -653,9 +705,12 @@ Navigate through the tabs to explore each feature in detail.
                 )
 
                 refresh_button.click(
-                    lambda bookmarks: ([
-                        f"{i+1}. {bookmark['title']} (Category: {bookmark['category']})" for i, bookmark in enumerate(bookmarks)
-                    ], display_bookmarks(bookmarks)),
+                    lambda bookmarks: (
+                        [
+                            f"{i+1}. {bookmark['title']} (Category: {bookmark['category']})" for i, bookmark in enumerate(bookmarks)
+                        ],
+                        display_bookmarks(bookmarks)
+                    ),
                     inputs=[state_bookmarks],
                     outputs=[bookmark_selector, bookmark_display_manage]
                 )