correct small bug

segment-anything-models-java · Mar 14, 2024 · fafb9e1 · fafb9e1
1 parent 9c06b09
commit fafb9e1
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/main/java/ai/nets/samj/gui/SAMJDialog.java b/src/main/java/ai/nets/samj/gui/SAMJDialog.java
@@ -551,8 +551,11 @@ public void drop(DropTargetDropEvent e) {
 	 * Close the panel and the plugin, deleting the python process if it exists
 	 */
 	public void close() {
-		if (display != null)
+		if (display != null) {
 			display.notifyNetToClose();
+			display.switchToNone();
+		}
+		display = null;
 		this.panelModel.interrupExistingThreads();
 		if (SwingUtilities.windowForComponent(this).isDisplayable())
 			SwingUtilities.windowForComponent(this).dispose();