Update app.py

app.py

@@ -15,9 +15,21 @@ DESCRIPTION = """
 # DALL•E 3 XL v2
 """
 
-def save_image(img):
+def create_download_link(filename):
+    with open(filename, "rb") as file:
+        encoded_string = base64.b64encode(file.read()).decode('utf-8')
+        download_link = f'<a href="data:image/png;base64,{encoded_string}" download="{filename}">Download Image</a>'
+        return download_link
+        
+def save_image(img, prompt):
     unique_name = str(uuid.uuid4()) + ".png"
     img.save(unique_name)
+
+    # save with promp to save prompt as image file name
+    filename = f"{prompt}.png"
+    img.save(filename)
+    return filename
+    
     return unique_name
 
 def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
@@ -83,8 +95,12 @@ def generate(
         output_type="pil",
     ).images
     image_paths = [save_image(img) for img in images]
+    download_links = [create_download_link(path) for path in image_paths]
+
     print(image_paths)
-    return image_paths, seed
+    #return image_paths, seed
+    return image_paths, seed, download_links
+
 
 #examples = [
 #    "neon holography crystal cat",