强制改为jpg格式

app.py

@@ -177,12 +177,10 @@ def generate(
                 callback_on_step_end=callback2
             ).images
         out_img = images[0] 
-        out_img = out_img.convert("RGB")
-        img_io = io.BytesIO()
-        out_img.save(img_io, format="PNG")  # 使用 PNG 格式保存
-        img_io.seek(0)
+        path = utils.save_image(out_img, "./outputs")
+        logger.info(f"output path: {path}")
         progress(1, desc="Complete")
-        return Image.open(img_io)
+        return path
     except GenerationError as e:
         logger.warning(f"Generation validation error: {str(e)}")
         raise gr.Error(str(e))
@@ -309,6 +307,7 @@ with gr.Blocks(css=custom_css).queue() as demo:
         with gr.Column():
             gr.Markdown("### Output")
             result = gr.Image(
+                type="filepath",
                 label="Generated Image",
                 elem_id="output-image"
             )

utils.py

@@ -99,4 +99,11 @@ def free_memory() -> None:
     if torch.cuda.is_available():
         torch.cuda.empty_cache()
         torch.cuda.ipc_collect()
-    gc.collect()
+    gc.collect()
+
+def save_image(image, output_dir):
+    filename = str(uuid.uuid4()) + ".jpg"
+    os.makedirs(output_dir, exist_ok=True)
+    filepath = os.path.join(output_dir, filename)
+    image.save(filepath, "JPEG", quality=80)
+    return filepath