Update app.py

app.py

@@ -8,6 +8,7 @@ from diffusers import DDIMScheduler, EulerAncestralDiscreteScheduler
 import cv2
 import torch
 import os
+import uuid  # UUIDを使用するために追加
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
@@ -40,11 +41,16 @@ def infer(prompt, negative_prompt, seed, randomize_seed, width, height, guidance
         generator=generator
     ).images[0]
 
-    # PNG形式で一時的に保存
-    output_path = "output_image.png"
-    output_image.save(output_path, format="PNG")
+    # UUIDを使用して一意のファイル名を生成
+    unique_filename = f"output_image_{uuid.uuid4().hex}.png"
+    output_image.save(unique_filename, format="PNG")
 
-    return output_path  # ファイルパスを返す
+    return unique_filename  # ファイルパスを返す
+
+def cleanup(filename):
+    if os.path.exists(filename):
+        os.remove(filename)
+        print(f"Deleted temporary file: {filename}")
 
 css = """
 #col-container {
@@ -130,6 +136,10 @@ with gr.Blocks(css=css) as demo:
         fn=infer,
         inputs=[prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps],
         outputs=[result]
+    ).then(
+        fn=cleanup,
+        inputs=[result],
+        outputs=[]
     )
 
 demo.queue().launch()