From ac98a4de0f18a0f79cd9e2b75875accfffe658d0 Mon Sep 17 00:00:00 2001
From: Victor Giers <mail@victorgiers.com>
Date: Thu, 7 May 2026 16:19:07 +0200
Subject: [PATCH] Use enriched prompt suffix in generation functions

---
 generate_equirect.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/generate_equirect.py b/generate_equirect.py
index 46020b4..8c60a36 100644
--- a/generate_equirect.py
+++ b/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,