From d21018b42b2b3af4d9f3214685a59d8a2dc7d779 Mon Sep 17 00:00:00 2001
From: Thomas <thomas@example.com>
Date: Sun, 10 May 2026 19:03:57 +0200
Subject: [PATCH] fix: ensure notifyComfyUI is called before blocking dialogs
 during restoration

---
 src/main/java/de/tki/comfymodels/Main.java | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/src/main/java/de/tki/comfymodels/Main.java b/src/main/java/de/tki/comfymodels/Main.java
index 204420a..74ad3e1 100644
--- a/src/main/java/de/tki/comfymodels/Main.java
+++ b/src/main/java/de/tki/comfymodels/Main.java
@@ -748,6 +748,9 @@ public class Main extends JFrame {
                 }
             }
             
+            // After all restores, notify ComfyUI
+            downloadManager.notifyComfyUI();
+            
             // After restores, start the actual download manager for the remaining models
             SwingUtilities.invokeLater(() -> {
                 // Update selected array (some might have been unselected by restore)
@@ -1664,10 +1667,11 @@ public class Main extends JFrame {
                 for (int r : processedRowIndices) sourceModel.removeRow(r);
                 for (Object[] rowData : rowsToMove) targetModel.addRow(rowData);
                 
-                JOptionPane.showMessageDialog(this, finalSuccess + " models successfully " + title.toLowerCase() + "ed.", 
-                    "Operation Complete", JOptionPane.INFORMATION_MESSAGE);
                 analyzeJsonContent();
                 downloadManager.notifyComfyUI();
+                
+                JOptionPane.showMessageDialog(this, finalSuccess + " models successfully " + title.toLowerCase() + "ed.", 
+                    "Operation Complete", JOptionPane.INFORMATION_MESSAGE);
             });
         }).start();