Update app.py

app.py

@@ -80,38 +80,29 @@ def create_zip(files):
             zipf.write(file, os.path.basename(file))
             print('added:' + file)
     return zip_filename
+
+def get_zip_download_link(zip_file):
+    """
+    Generate a link to download the zip file.
+    """
+    zip_base64 = encode_file_to_base64(zip_file)  # Encode the zip file to base64
+    href = f'<a href="data:application/zip;base64,{zip_base64}" download="{zip_file}">Download All</a>'
+    return href
     
-# 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
-    print(' Zip file created:' + zip_filename)
-
-    gr.Button(link="/file=" + zip_filename)
-
-    # remove?
-    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>'
-    gr.HTML(download_link)
-    
-    # redirect_button = gr.Button("Clear", variant='secondary')
-    # redirect_button.click(None, None,None, _js="window.location.assign('https://google.com');")
-    
+    zip_filename = create_zip_of_files(images)  # Create a zip file from the list of image files
+    print(f"Zip file created: {zip_filename}")
+    download_link = get_zip_download_link(zip_filename)
     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)   
-    if zip_filename:
-        print(zip_filename)
-        gr.Button(link=zip_filename)
-        gr.File(value=zip_filename)
+    zip_filename, download_link = save_all_images(images)
     if download_link:
-        print(download_link)
-        gr.HTML(download_link)
-        gr.Button(link=download_link)
+        gr.HTML(download_link).update()
+
 
 # Function to handle "Clear All" button click
 def clear_all_button_click():