Update app.py

app.py

@@ -193,9 +193,20 @@ function monitorImageSrcChanges() {
     setTimeout(() => {
       downloadedImages.delete(src); // Remove from the set after the cooldown
     }, 500); // 500ms cooldown (adjust as needed)
+
+    // After download is triggered, click the button with id "TheButt"
+    setTimeout(() => {
+      const button = document.getElementById('TheButt');
+      if (button) {
+        button.click(); // Click the button
+      } else {
+        console.error('Button with id "TheButt" not found!');
+      }
+    }, 500); // Adjust the timeout if needed to make sure the download starts before clicking
   }
 }
 
+
 window.addEventListener('load', () => {
     monitorImageSrcChanges();
     console.log("Yo");
@@ -247,7 +258,7 @@ with gr.Blocks(head=CSS + JS) as demo:
                     seed_rand = gr.Button("Randomize Seed 🎲", size="sm", variant="secondary")
                     seed_rand.click(randomize_seed, None, [seed], queue=False)
         with gr.Row():
-            gen_button = gr.Button(f'Generate up to {int(num_models)} images', variant='primary', scale=3, elem_classes=["butt"])
+            gen_button = gr.Button(f'Generate up to {int(num_models)} images', variant='primary', scale=3, elem_classes=["butt"], elem_id=["TheButt"])
             random_button = gr.Button(f'Randomize Models', variant='secondary', scale=1)
 
     with gr.Column(scale=1):