Update app.py

app.py

@@ -88,12 +88,13 @@ def save_all_images(images):
     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)
+    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');")
     
@@ -106,10 +107,10 @@ def save_all_button_click():
     if zip_filename:
         print(zip_filename)
         gr.Button(link=zip_filename)
-        # gr.File(value=zip_filename)
+        gr.File(value=zip_filename)
     if download_link:
         print(download_link)
-        #gr.HTML(download_link)
+        gr.HTML(download_link)
         gr.Button(link=download_link)
 
 # Function to handle "Clear All" button click