fixed download options

index.html

@@ -488,11 +488,33 @@
                 }
             }
 
+            // downloadMask() {
+            //     if (!this.predictedMask) return;
+            //     const link = document.createElement('a');
+            //     link.download = 'binary_mask.png';
+            //     link.href = this.predictedMask.src;
+            //     link.click();
+            // }
+
             downloadMask() {
-                if (!this.predictedMask) return;
+                if (!this.predictedMask) return; // No mask, nothing to download
+
                 const link = document.createElement('a');
-                link.download = 'binary_mask.png';
-                link.href = this.predictedMask.src;
+                let filename = 'segmentation_result.png';
+
+                // Determine filename based on current view
+                if (this.currentView === 'filled') {
+                    filename = 'filled_mask.png';
+                } else if (this.currentView === 'outline') {
+                    filename = 'outline_mask.png';
+                } else if (this.currentView === 'binary') {
+                    filename = 'binary_mask.png';
+                }
+
+                // Get the image data from the outputCanvas (which holds the current view)
+                link.download = filename;
+                link.href = this.outputCanvas.toDataURL('image/png'); // Gets the current content of the outputCanvas
+                
                 link.click();
             }
         }