Update app.py

app.py

@@ -187,9 +187,20 @@ desc_html='''
 </div>
 '''
 def save_image(img):
-    unique_name = str(uuid.uuid4()) + ".png"
-    img.save(unique_name)
-    return unique_name
+    # Generate a unique filename
+    unique_name = str(uuid.uuid4()) + ".webp"
+    
+    # Convert the image to WebP format
+    webp_img = img.convert("RGB")  # Ensure the image is in RGB mode
+    
+    # Save the image in WebP format with high quality
+    webp_img.save(unique_name, "WEBP", quality=90)
+    
+    # Open the saved WebP file and return it as a PIL Image object
+    with Image.open(unique_name) as webp_file:
+        webp_image = webp_file.copy()
+    
+    return webp_image, unique_name
     
 def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
     if randomize_seed:
@@ -223,10 +234,10 @@ def generate(prompt, progress=gr.Progress(track_tqdm=True)):
         num_images_per_prompt=NUM_IMAGES_PER_PROMPT,
         output_type="pil",
     ).images
-    
-    image_paths = [save_image(img) for img in images]
+    images = [save_image(img) for img in images]
+    image_paths = [i[1] for i in images]
     print(prompt_str, image_paths)
-    return image_paths
+    return [i[0] for i in images]
 
 with gr.Blocks(css=css,head=js,fill_height=True) as demo:
     with gr.Row(equal_height=False):