Update app.py

app.py

@@ -49,7 +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:
@@ -79,6 +78,11 @@ def save_all_button_click():
     if download_link:
         gr.write(download_link)
 
+    zip_file = create_zip_of_files(all_files)
+    st.markdown(get_zip_download_link(zip_file), unsafe_allow_html=True)
+
+
+
 # Function to handle "Clear All" button click
 def clear_all_button_click():
     clear_all_images()
@@ -86,8 +90,6 @@ def clear_all_button_click():
 
 
 
-
-
 print(f"SAFETY_CHECKER: {SAFETY_CHECKER}")
 print(f"TORCH_COMPILE: {TORCH_COMPILE}")
 print(f"device: {device}")