Update app.py

app.py

@@ -83,6 +83,7 @@ def create_zip(files):
     with zipfile.ZipFile(zip_filename, 'w') as zipf:
         for file in files:
             zipf.write(file, os.path.basename(file))
+            print('added:' + file)
     return zip_filename
     
 # Function to save all images as a zip file and provide a base64 download link
@@ -90,23 +91,30 @@ 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_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.File(download_link)
-    #print(download_link)
+    gr.File(link=download_link)
+    print(download_link)
+    gr.HTML(download_link)
+    redirect_button = gr.Button("Clear", variant='secondary')
+    redirect_button.click(None, None,None, _js="window.location.assign('https://google.com');")
+
+    
     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:
-    #    gr.HTML(zip_filename)
-    #if download_link:
-    #    gr.HTML(download_link)
+    if zip_filename:
+        gr.Button(link=zip_filename)
+    if download_link:
+        gr.HTML(download_link)
 
 
 # Function to handle "Clear All" button click