Update app.py

app.py

@@ -33,15 +33,6 @@ torch_device = device
 torch_dtype = torch.float16
 
 
-# add file save and download and clear:
-# Function to create a zip file from a list of files
-def create_zip(files):
-    timestamp = datetime.datetime.now().strftime("%Y%m%d%H%M%S")
-    zip_filename = f"images_{timestamp}.zip"
-    with zipfile.ZipFile(zip_filename, 'w') as zipf:
-        for file in files:
-            zipf.write(file, os.path.basename(file))
-    return zip_filename
 
 # Function to encode a file to base64
 def encode_file_to_base64(file_path):
@@ -83,6 +74,17 @@ def clear_all_images():
     for file in img_files:
         os.remove(file)
 
+        
+# add file save and download and clear:
+# Function to create a zip file from a list of files
+def create_zip(files):
+    timestamp = datetime.datetime.now().strftime("%Y%m%d%H%M%S")
+    zip_filename = f"images_{timestamp}.zip"
+    with zipfile.ZipFile(zip_filename, 'w') as zipf:
+        for file in files:
+            zipf.write(file, os.path.basename(file))
+    return zip_filename
+    
 # Function to save all images as a zip file and provide a base64 download link
 def save_all_images(images):
     if len(images) == 0:
@@ -90,6 +92,8 @@ def save_all_images(images):
     zip_filename = create_zip(images)  # Create a zip file from the list of image files
     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)
+    print(download_link)
     return zip_filename, download_link
         
 # Function to handle "Save All" button click