Update app.py

app.py

@@ -114,7 +114,42 @@ def add_gallery(image, model_str, gallery):
     return gallery
 
 js="""
-    <script>console.log("BOOOOOOOOOOOOOOOOBS");</script>
+<script>
+
+downloadImage = (url, filename) => {
+    const a = document.createElement('a');
+    a.href = url;
+    a.download = filename || 'image.jpg';
+    document.body.appendChild(a);
+    a.click();
+    document.body.removeChild(a);
+};
+
+// Monitor image changes
+const monitorImageChange = (imageElement) => {
+    if (!(imageElement instanceof HTMLImageElement)) {
+        console.error('The provided element is not an image.');
+        return;
+    }
+
+    const observer = new MutationObserver((mutationsList) => {
+        mutationsList.forEach((mutation) => {
+            if (mutation.type === 'attributes' && mutation.attributeName === 'src') {
+                console.log('Image source changed:', imageElement.src);
+                downloadImage(imageElement.src, 'downloaded_image.jpg');
+            }
+        });
+    });
+
+    observer.observe(imageElement, {
+        attributes: true, // Look for attribute changes
+        attributeFilter: ['src'], // Only watch for 'src' changes
+    });
+
+    console.log('Now monitoring image changes for:', imageElement);
+};
+    
+</script>
 """
 
 with gr.Blocks(fill_width=True, head=js) as demo: