Merge pull request #242 from Thawtz/fix-database-refresh

Fix: force refresh for embedding and hypernetwork tables initialization

Author: anapnoe

extensions-builtin/sd-webui-ux/scripts/anapnoe/database_manager.py

@@ -94,8 +94,13 @@ def import_tables_generator(self, table_types: List[str], refresh: bool = False)
                 "message": "No valid table types to process"
             }) + "\n"
             return
-            
+
+        if refresh:
+            for page in pages.values():
+                page.refresh()
+
         total_items = sum(len(list(page.list_items()) or []) for page in pages.values())
+
 
         pbar = tqdm(total=total_items, unit='item', 
                     bar_format='{l_bar}{bar}| {n_fmt}/{total_fmt} [{percentage:3.0f}%]')

extensions-builtin/sd-webui-ux/scripts/anapnoe_sd_uiux_db.py

@@ -78,13 +78,15 @@ def initialize_tables():
     else:
         logger.debug("All tables already exist in database")
 
-initialize_tables()
-
 def on_ui_tabs():
+    # Make sure tables are initialized when UI loads
+    initialize_tables()
+
     with gr.Blocks(analytics_enabled=False) as anapnoe_sd_uiux_db:
         refresh_button = gr.Button("Refresh Database", elem_id="refresh_database")
         refresh_button.click(fn=initialize_tables, inputs=[], outputs=[])
 
     return (anapnoe_sd_uiux_db, 'Init DB', 'anapnoe_sd_uiux_db'),
 
 script_callbacks.on_ui_tabs(on_ui_tabs)
+