Update app.py

app.py

@@ -408,19 +408,27 @@ def zip_outputs_folder():
         outputs_dir = "Outputs"
         zip_path = os.path.join("tmp", "outputs.zip")
         os.makedirs("tmp", exist_ok=True)
-        
-        if os.path.exists(outputs_dir) and os.path.isdir(outputs_dir):
-            with zipfile.ZipFile(zip_path, 'w', zipfile.ZIP_DEFLATED) as zipf:
-                for root, _, files in os.walk(outputs_dir):
-                    for file in files:
-                        file_path = os.path.join(root, file)
-                        arcname = os.path.relpath(file_path, outputs_dir)
-                        zipf.write(file_path, os.path.join("Outputs", arcname))
-            return zip_path, "Outputs folder successfully zipped as outputs.zip"
-        else:
-            return None, "Outputs folder does not exist or is empty"
+
+        if not os.path.exists(outputs_dir) or not os.path.isdir(outputs_dir):
+            return "Outputs folder does not exist or is empty", gr.DownloadButton(visible=False)
+
+        with zipfile.ZipFile(zip_path, 'w', zipfile.ZIP_DEFLATED) as zipf:
+            for root, _, files in os.walk(outputs_dir):
+                for file in files:
+                    file_path = os.path.join(root, file)
+                    arcname = os.path.relpath(file_path, outputs_dir)
+                    zipf.write(file_path, os.path.join("Outputs", arcname))
+
+        return (
+            "outputs.zip created successfully",
+            gr.DownloadButton(
+                value=zip_path,
+                label="Download Outputs.zip",
+                visible=True
+            )
+        )
     except Exception as e:
-        return None, f"Failed to zip Outputs folder: {str(e)}"
+        return f"Zip failed: {str(e)}", gr.DownloadButton(visible=False)
 
 def compress_wav_to_mp3():
     """Convert .wav files in Outputs folder and subfolders to 80kbps .mp3 files, replacing the originals"""
@@ -793,11 +801,6 @@ def main():
             outputs=[zip_status]
         )
 
-        demo.load(
-            fn=get_download_zip_state,
-            inputs=[],
-            outputs=[download_zip_button, zip_status]
-        )
 
         seed_id_input.change(
             fn=get_pt_file_for_download,