Update app.py

app.py

@@ -45,7 +45,7 @@ css = """
     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
@@ -56,22 +56,19 @@ with gr.Blocks(css=css) as demo:
                 label="Upload AVIF File",
                 file_types=[".avif"],
                 file_count="multiple"
-            )
+            )            
             run_button = gr.Button("Convert", scale=0)
         result = gr.Gallery(label="Result")
         download_button = gr.Button("Download All as ZIP")
-        error_modal = gr.Modal("No files to download. Please upload and convert AVIF files first.")
+
     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:
-            error_modal.show()
-            return None
         return zip_files(converted_files)
 
     run_button.click(
@@ -83,5 +80,13 @@ with gr.Blocks(css=css) as demo:
         fn=handle_download,
         inputs=[],
         outputs=[gr.File()]
+    ).success(
+        lambda: gr.Info("Download started."),
+        inputs=None,
+        outputs=None
+    ).error(
+        lambda e: gr.Error(str(e)),
+        inputs=None,
+        outputs=None
     )
 demo.queue().launch()