Update app.py

app.py

@@ -469,23 +469,23 @@ def process_uploaded_file(file, state_bookmarks):
 
     if file is None:
         logger.warning("No file uploaded")
-        return "Please upload a bookmarks HTML file.", '', state_bookmarks, gr.update(choices=[])
+        return "Please upload a bookmarks HTML file.", '', state_bookmarks, gr.CheckboxGroup.update(choices=[])
 
     try:
         file_content = file.decode('utf-8')
     except UnicodeDecodeError as e:
         logger.error(f"Error decoding the file: {e}", exc_info=True)
-        return "Error decoding the file. Please ensure it's a valid HTML file.", '', state_bookmarks, gr.update(choices=[])
+        return "Error decoding the file. Please ensure it's a valid HTML file.", '', state_bookmarks, gr.CheckboxGroup.update(choices=[])
 
     try:
         bookmarks = parse_bookmarks(file_content)
     except Exception as e:
         logger.error(f"Error parsing bookmarks: {e}", exc_info=True)
-        return "Error parsing the bookmarks HTML file.", '', state_bookmarks, gr.update(choices=[])
+        return "Error parsing the bookmarks HTML file.", '', state_bookmarks, gr.CheckboxGroup.update(choices=[])
 
     if not bookmarks:
         logger.warning("No bookmarks found in the uploaded file")
-        return "No bookmarks found in the uploaded file.", '', state_bookmarks, gr.update(choices=[])
+        return "No bookmarks found in the uploaded file.", '', state_bookmarks, gr.CheckboxGroup.update(choices=[])
 
     # Assign unique IDs to bookmarks
     for idx, bookmark in enumerate(bookmarks):
@@ -505,7 +505,7 @@ def process_uploaded_file(file, state_bookmarks):
         faiss_index = vectorize_and_index(bookmarks)
     except Exception as e:
         logger.error(f"Error building FAISS index: {e}", exc_info=True)
-        return "Error building search index.", '', state_bookmarks, gr.update(choices=[])
+        return "Error building search index.", '', state_bookmarks, gr.CheckboxGroup.update(choices=[])
 
     message = f"✅ Successfully processed {len(bookmarks)} bookmarks."
     logger.info(message)
@@ -518,7 +518,7 @@ def process_uploaded_file(file, state_bookmarks):
     # Update state
     state_bookmarks = bookmarks.copy()
 
-    return message, bookmark_html, state_bookmarks, gr.update(choices=choices)
+    return message, bookmark_html, state_bookmarks, gr.CheckboxGroup.update(choices=choices)
 
 def delete_selected_bookmarks(selected_indices, state_bookmarks):
     """
@@ -526,7 +526,7 @@ def delete_selected_bookmarks(selected_indices, state_bookmarks):
     """
     global bookmarks, faiss_index
     if not selected_indices:
-        return "⚠️ No bookmarks selected.", gr.update(choices=[]), display_bookmarks()
+        return "⚠️ No bookmarks selected.", gr.CheckboxGroup.update(choices=[]), display_bookmarks()
 
     ids_to_delete = []
     indices_to_delete = []
@@ -554,16 +554,16 @@ def delete_selected_bookmarks(selected_indices, state_bookmarks):
     # Update state
     state_bookmarks = bookmarks.copy()
 
-    return message, gr.update(choices=choices), display_bookmarks()
+    return message, gr.CheckboxGroup.update(choices=choices), display_bookmarks()
 
 def edit_selected_bookmarks_category(selected_indices, new_category, state_bookmarks):
     """
     Edit category of selected bookmarks.
     """
     if not selected_indices:
-        return "⚠️ No bookmarks selected.", gr.update(choices=[]), display_bookmarks(), state_bookmarks
+        return "⚠️ No bookmarks selected.", gr.CheckboxGroup.update(choices=[]), display_bookmarks(), state_bookmarks
     if not new_category:
-        return "⚠️ No new category selected.", gr.update(choices=[]), display_bookmarks(), state_bookmarks
+        return "⚠️ No new category selected.", gr.CheckboxGroup.update(choices=[]), display_bookmarks(), state_bookmarks
 
     indices = [int(s.split('.')[0])-1 for s in selected_indices]
     for idx in indices:
@@ -581,7 +581,7 @@ def edit_selected_bookmarks_category(selected_indices, new_category, state_bookm
     # Update state
     state_bookmarks = bookmarks.copy()
 
-    return message, gr.update(choices=choices), display_bookmarks(), state_bookmarks()
+    return message, gr.CheckboxGroup.update(choices=choices), display_bookmarks(), state_bookmarks
 
 def export_bookmarks():
     """
@@ -722,7 +722,7 @@ def build_app():
             # Initialize state
             state_bookmarks = gr.State([])
 
-            # Define 'bookmark_selector' first
+            # Define 'bookmark_selector' once
             bookmark_selector = gr.CheckboxGroup(
                 label="✅ Select Bookmarks",
                 choices=[]
@@ -773,7 +773,7 @@ Navigate through the tabs to explore each feature in detail.
                 process_button.click(
                     process_uploaded_file,
                     inputs=[upload, state_bookmarks],
-                    outputs=[output_text, bookmark_display, state_bookmarks, bookmark_display, bookmark_selector]
+                    outputs=[output_text, bookmark_display, state_bookmarks, bookmark_selector]
                 )
 
             with gr.Tab("Chat with Bookmarks"):
@@ -843,8 +843,8 @@ Navigate through the tabs to explore each feature in detail.
                 bookmark_display_manage = gr.HTML(label="📄 Bookmarks")
 
                 with gr.Row():
-                    # Render 'bookmark_selector' inside the tab
-                    bookmark_selector.render()
+                    # Include 'bookmark_selector' directly in the layout
+                    bookmark_selector
 
                 with gr.Row():
                     delete_button = gr.Button("🗑️ Delete Selected")
@@ -885,6 +885,9 @@ Navigate through the tabs to explore each feature in detail.
 
         logger.info("Launching Gradio app")
         demo.launch(debug=True)
+    except gr.Error as e:
+        logger.error(f"Gradio Error: {e}", exc_info=True)
+        print(f"Gradio Error: {e}")
     except Exception as e:
         logger.error(f"Error building the app: {e}", exc_info=True)
         print(f"Error building the app: {e}")