Update app.py

app.py

@@ -7,6 +7,7 @@ from PIL import Image
 import re
 import base64
 from io import BytesIO
+import pytz
 
 try:
     import intel_extension_for_pytorch as ipex
@@ -98,9 +99,16 @@ def predict(prompt, guidance, steps, seed=1231231):
         nsfw=gr.Button("🕹️NSFW🎨", scale=1)
 
     # 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"
+    #date_str = datetime.datetime.now().strftime("%Y%m%d")
+    #safe_prompt = prompt.replace(" ", "_")[:50]  # Truncate long prompts
+    #filename = f"{date_str}_{safe_prompt}.png"
+
+    central = pytz.timezone('US/Central')
+    safe_date_time = datetime.now(central).strftime("%m%d_%H%M")
+    replaced_prompt = prompt.replace(" ", "_").replace("\n", "_")
+    safe_prompt = "".join(x for x in replaced_prompt if x.isalnum() or x == "_")[:90]
+    filename = f{safe_date_time}_{safe_prompt}.{file_type}"
+    
 
     # Save the image
     if len(results.images) > 0: