Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import subprocess
 import os
 import shutil
+import zipfile
 
 
 def setup_avif_decode():
@@ -20,7 +21,7 @@ def convert_avif_to_png(avif_files):
     output_paths = []
     for avif_file in avif_files:
         avif_path = avif_file.name
-        png_path = avif_path.rsplit('.', 1)[0] + '.png'   
+        png_path = avif_path.rsplit('.', 1)[0] + '.png'
         result = subprocess.run(["avif-decode/target/release/avif_decode", "-f", avif_path, png_path], capture_output=True, text=True)
         if result.returncode == 0:
             output_paths.append(png_path)
@@ -29,29 +30,57 @@ def convert_avif_to_png(avif_files):
     return output_paths
 
 
+def zip_files(file_paths):
+    zip_path = "/tmp/converted_avif.zip"
+    with zipfile.ZipFile(zip_path, 'w') as zipf:
+        for file_path in file_paths:
+            if os.path.isfile(file_path):
+                zipf.write(file_path, os.path.basename(file_path))
+    return zip_path
+
+
 css = """
 #col-container {
     margin: 0 auto;
     max-width: 520px;
 }
 """
-with gr.Blocks(css=css) as demo:   
+with gr.Blocks(css=css) as demo: 
     with gr.Column(elem_id="col-container"):
         gr.Markdown("""
         # AVIF to PNG Converter
         Upload your AVIF files and get them converted to PNG.
-        """)       
-        with gr.Row():           
+        """)
+        with gr.Row():
             avif_file = gr.File(
                 label="Upload AVIF File",
                 file_types=[".avif"],
                 file_count="multiple"
-            )           
-            run_button = gr.Button("Convert", scale=0)       
+            )
+            run_button = gr.Button("Convert", scale=0)
         result = gr.Gallery(label="Result")
+        download_button = gr.Button("Download All as ZIP")
+        error_message = gr.Textbox(label="Message", interactive=False, visible=False)
+    converted_files = []
+
+    def handle_conversion(avif_files):
+        global converted_files
+        converted_files = convert_avif_to_png(avif_files)
+        return converted_files
+        
+    def handle_download():
+        if not converted_files:
+            return None, "No files to download. Please upload and convert AVIF files first."
+        return zip_files(converted_files), ""
+
     run_button.click(
-        fn=convert_avif_to_png,
+        fn=handle_conversion,
         inputs=[avif_file],
         outputs=[result]
     )
+    download_button.click(
+        fn=handle_download,
+        inputs=[],
+        outputs=[gr.File(), error_message]
+    )
 demo.queue().launch()