Use enriched prompt suffix in generation functions

generate_equirect.py

@@ -545,6 +545,7 @@ def generate(
     height: int = 512,
     seam_inpaint: bool = False,
 ) -> str:
+    prompt = enrich_prompt(prompt)
     model_path = resolve_generation_model_path(model_path, work_dir)
 
     device = select_device()
@@ -868,13 +869,14 @@ def main():
             print(f"Generation failed: {e}")
             raise
 
-    base = sanitize_name(args.prompt)
+    prompt = enrich_prompt(args.prompt)
+    base = sanitize_name(prompt)
     target = args.output or next_filename(args.output_dir, base, args.width, args.height)
     output_abs = os.path.abspath(target)
 
     try:
         result_path = generate(
-            args.prompt,
+            prompt,
             output_abs,
             args.work_dir,
             upscale=args.upscale,