Update app.py

app.py

@@ -98,12 +98,9 @@ def save_all_images(images):
     # 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(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');")
 
+    # 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
         
@@ -112,18 +109,18 @@ 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)
     if download_link:
+        print(download_link)
         gr.HTML(download_link)
-
+        gr.Button(link=download_link)
 
 # Function to handle "Clear All" button click
 def clear_all_button_click():
     clear_all_images()
 
-
-
-
 print(f"SAFETY_CHECKER: {SAFETY_CHECKER}")
 print(f"TORCH_COMPILE: {TORCH_COMPILE}")
 print(f"device: {device}")
@@ -313,5 +310,7 @@ with gr.Blocks(css=css) as demo:
         save_all_button.click(save_all_button_click)
         clear_all_button.click(clear_all_button_click)
 
+        
+
 demo.queue()
 demo.launch()