Update app.py

app.py

@@ -49,16 +49,6 @@ def encode_file_to_base64(file_path):
         encoded = base64.b64encode(file.read()).decode()
     return encoded
 
-# Function to save all images as a zip file and provide a base64 download link
-def save_all_images(images):
-    if len(images) == 0:
-        return None, None
-
-    zip_filename = create_zip(images)  # Create a zip file from the list of image files
-    zip_base64 = encode_file_to_base64(zip_filename)  # Encode the zip file to base64
-    download_link = f'<a href="data:application/zip;base64,{zip_base64}" download="{zip_filename}">Download All</a>'
-
-    return zip_filename, download_link
 
     
 def create_zip_of_files(files):
@@ -93,18 +83,24 @@ def clear_all_images():
     for file in img_files:
         os.remove(file)
 
+# Function to save all images as a zip file and provide a base64 download link
+def save_all_images(images):
+    if len(images) == 0:
+        return None, None
+    zip_filename = create_zip(images)  # Create a zip file from the list of image files
+    zip_base64 = encode_file_to_base64(zip_filename)  # Encode the zip file to base64
+    download_link = f'<a href="data:application/zip;base64,{zip_base64}" download="{zip_filename}">Download All</a>'
+    return zip_filename, download_link
+        
 # Function to handle "Save All" button click
 def save_all_button_click():
     images = [file for file in os.listdir() if file.lower().endswith((".png", ".jpg", ".jpeg"))]
-    zip_filename, download_link = save_all_images(images)
-    
+    zip_filename, download_link = save_all_images(images)   
+    if zip_filename:
+        gr.HTML(zip_filename)
     if download_link:
         gr.HTML(download_link)
 
-    zip_file = create_zip_of_files(all_files)
-    gr.HTML(get_zip_download_link(zip_file))
-
-
 
 # Function to handle "Clear All" button click
 def clear_all_button_click():