Update app.py

app.py

@@ -88,19 +88,20 @@ def main_interface():
 
             with gr.Column(scale=2, elem_id="right-panel"):
                 preview_gallery = gr.Gallery(label="Preview Combined Images", columns=3, elem_id="preview-gallery")
-                download_zip_button = gr.Button("Download ZIP", elem_id="download-zip")
+                download_zip_button = gr.Button("Download ZIP", elem_id="download-zip", visible=False)
+                output_zip = gr.File(label="Download Combined ZIP", visible=False)
 
         # Button actions
         combine_button.click(
             process_images,
             inputs=[uploaded_files],
-            outputs=[preview_gallery, download_zip_button],
+            outputs=[preview_gallery, output_zip],
         )
 
         download_zip_button.click(
-            lambda x: x,
-            inputs=[],
-            outputs=[download_zip_button],
+            lambda zip_path: zip_path,
+            inputs=[output_zip],
+            outputs=[output_zip],
         )
 
     return interface