Update app.py

app.py

@@ -5,6 +5,8 @@ import random
 import os
 from PIL import Image
 from datetime import datetime
+import string
+import re
 
 app = Flask(__name__)
 
@@ -36,7 +38,23 @@ def save_image(image, prompt):
     image.save(filepath, format='PNG')
     return filename
 
+def sanitize_filename(filename):
+    # Replace spaces with underscores
+    filename = filename.replace(" ", "_")
 
+    # Remove special characters
+    sanitized_filename = re.sub(r'[^\w\s-]', '', filename)
+    sanitized_filename = re.sub(r'[-\s]+', '-', sanitized_filename)
+
+    return sanitized_filename
+
+def save_image(image, prompt):
+    timestamp = datetime.now().strftime("%Y%m%d%H%M%S")
+    filename = f"{sanitize_filename(prompt)}-{random.randint(1, 100000)}-{timestamp}.png"
+    filepath = os.path.join(TEMP_DIR, filename)
+    image.save(filepath, format='PNG')
+    return filename
+    
 @app.route("/generate", methods=["POST"])
 def generate():
     try: