Update app.py

app.py

@@ -79,9 +79,23 @@ def predict(prompt, guidance, steps, seed=1231231):
     )
     if nsfw_content_detected:
         nsfw=gr.Button("🕹️NSFW🎨", scale=1)
-    #    gr.Warning("NSFW content detected.")
-    #    return Image.new("RGB", (512, 512))
-    return results.images[0]
+
+
+    # Generate file name
+    date_str = datetime.datetime.now().strftime("%Y%m%d")
+    safe_prompt = prompt.replace(" ", "_")[:50]  # Truncate long prompts
+    filename = f"{date_str}_{safe_prompt}.png"
+
+    # Save the image
+    if len(results.images) > 0:
+        image_path = os.path.join("path_to_save", filename)  # Specify your directory
+        results.images[0].save(image_path)
+        print(f"Image saved as {image_path}")
+    
+    return results.images[0] if len(results.images) > 0 else None
+
+        
+    #return results.images[0]
 
 
 css = """