Update app.py

app.py

@@ -118,22 +118,11 @@ def add_gallery(image, model_str, gallery):
 JS="""
 <script>
 
-/*
-function simulateButtonPress_() {
-    const button = document.getElementById('simulate-button');
-    if (button) {
-        button.click();  // Simulate the button press
-        console.log('Button Pressed!');
-    }
-}
-*/
-
-function simulateButtonPress() {
-    console.log('Button Pressed!');
-}
-
 // Function to monitor image src changes and automatically download the image
 function monitorImageSrcChanges() {
+  // Set of recently downloaded image URLs to avoid re-triggering the download
+  const downloadedImages = new Set();
+
   // Create a MutationObserver instance
   const observer = new MutationObserver((mutationsList, observer) => {
     // Loop through all mutations
@@ -181,8 +170,17 @@ function monitorImageSrcChanges() {
     srcObserver.observe(img, { attributes: true, attributeFilter: ['src'] });
   }
 
-  // Function to download an image automatically
+  // Function to download an image automatically with a cooldown to prevent multiple downloads
   function downloadImage(src) {
+    // Check if the image has been downloaded recently
+    if (downloadedImages.has(src)) {
+      return; // Prevent duplicate downloads
+    }
+
+    // Add the image src to the set of downloaded images
+    downloadedImages.add(src);
+
+    // Trigger the download
     const link = document.createElement('a');
     link.href = src;
     link.download = src.split('/').pop(); // Use the file name from the URL (last part of the src)
@@ -190,6 +188,11 @@ function monitorImageSrcChanges() {
     document.body.appendChild(link);
     link.click(); // Trigger the download
     document.body.removeChild(link); // Clean up the DOM by removing the link after download
+
+    // Set a cooldown to allow the download to be triggered again after a delay (e.g., 500ms)
+    setTimeout(() => {
+      downloadedImages.delete(src); // Remove from the set after the cooldown
+    }, 500); // 500ms cooldown (adjust as needed)
   }
 }