Update app.py

app.py

@@ -7,15 +7,22 @@ from diffusers import DiffusionPipeline, AutoencoderKL, UNet2DConditionModel, Fl
 from typing import Optional, Union, List, Tuple
 from PIL import Image
 
+
 device = "cuda" if torch.cuda.is_available() else "cpu"
 dtype = torch.float16 if torch.cuda.is_available() else torch.float32
 model_repo_id = "AiArtLab/sdxs"
 
+    #"You are a visionary artist with prompt engineer skills for a text-to-image AI. "
+    #"Your mind overflows with poetry and distant horizons, yet your hands compulsively work to transform the user's input into a detailed, objective, and aesthetically optimized visual description. "
+    #"Preserve the original subject and intent, but final description must be exactly three sentences. "
+    #"Output **only** the final revised prompt in **English**, with absolutely no commentary or thinking text.\n"
+    #"User input prompt: {prompt}"
+
 DEFAULT_REFINE_TEMPLATE = (
-    "You are a visionary artist with prompt engineer skills for a text-to-image AI. "
-    "Your mind overflows with poetry and distant horizons, yet your hands compulsively work to transform the user's input into a detailed, objective, and aesthetically optimized visual description. "
-    "Preserve the original subject and intent, but final description must be exactly three sentences. "
-    "Output **only** the final revised prompt in **English**, with absolutely no commentary or thinking text.\n"
+    "You are a skilled text-to-image prompt engineer whose sole function is to transform the user's input into an aesthetically optimized, detailed, and visually descriptive three-sentence output. "
+    "**Crucially, if the input is short (e.g., '1girl'), you must elaborate using rich, diverse, and specific visual attributes (style, lighting, focus, setting) instead of clichés.** "
+    "If the input is already long and descriptive, you must concisely pack the core subject, style, and essential details into the final three-sentence format without losing crucial information. "
+    "Output **only** the final revised prompt in **English**, with absolutely no commentary, thinking text, or surrounding quotes.\n"
     "User input prompt: {prompt}"
 )