Fix batch download button - use Button+File instead of DownloadButton

ui/callbacks.py

@@ -1604,12 +1604,12 @@ def handle_batch_download(output_path_str: str):
     import tempfile
     
     if not output_path_str:
-        return None
+        return gr.update(visible=False)
     
     from pathlib import Path
     output_path = Path(output_path_str)
     if not output_path.exists() or not output_path.is_dir():
-        return None
+        return gr.update(visible=False)
     
     # Create a temporary ZIP file
     temp_zip = tempfile.NamedTemporaryFile(
@@ -1627,7 +1627,7 @@ def handle_batch_download(output_path_str: str):
                     arcname = file.relative_to(output_path)
                     zipf.write(file, arcname)
         
-        return temp_zip.name
+        return gr.update(value=temp_zip.name, visible=True)
     except Exception as e:
         print(f"Error creating ZIP file: {e}")
-        return None
+        return gr.update(visible=False)

ui/layout.py

@@ -368,11 +368,15 @@ def create_layout(
                                 "Cancel", variant="stop", visible=False
                             )
                         with gr.Row():
-                            batch_download_button = gr.DownloadButton(
+                            batch_download_button = gr.Button(
                                 "Download All Images as ZIP",
                                 variant="secondary",
                                 visible=False,
                             )
+                        batch_download_file = gr.File(
+                            label="Download ZIP",
+                            visible=False,
+                        )
 
             with gr.TabItem("Config", elem_id="settings-tab-container"):
                 config_initial_provider = initial_provider
@@ -2378,6 +2382,7 @@ def create_layout(
         batch_download_button.click(
             fn=callbacks.handle_batch_download,
             inputs=batch_output_path_state,
+            outputs=[batch_download_file],
         )
 
         app.load(