added Save

check_app.py

@@ -15,7 +15,9 @@ from diffusers import (
 )
 import gradio as gr
 from diffusers.pipelines.pipeline_utils import DiffusionPipeline
-
+from pathlib import Path
+import time
+from datetime import datetime
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 1024
 
@@ -163,7 +165,8 @@ def generate_image_with_progress(model_name,pipe, prompt, num_steps, guidance_sc
     
     # Generate image
     image = pipe(**common_args).images[0]
-
+    filepath = save_generated_image(image, model_name, prompt)
+    print(f"Saved image to: {filepath}")
     return seed, image
 
 @spaces.GPU(duration=170)
@@ -264,6 +267,17 @@ def main():
 
     app.launch()
 
+def save_generated_image(image, model_name, prompt):
+    """Save generated image with timestamp and model name"""
+    timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+    # Create sanitized filename from prompt (first 30 chars)
+    prompt_part = "".join(c for c in prompt[:30] if c.isalnum() or c in (' ', '-', '_')).strip()
+    filename = f"{timestamp}_{model_name}_{prompt_part}.png"
+    path = Path(model_name)
+    path.mkdir(parents=True, exist_ok=True)
+    filepath = os.path.join(model_name, filename)
+    image.save(filepath)
+    return filepath
 
 if __name__ == "__main__":
     main()