giers10/skymap-gen
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Refactor prompt enrichment logic in generate_equirect.py

Browse Source
This commit is contained in:
Victor Giers
2026-05-07 16:19:45 +02:00
parent ac98a4de0f
commit ef2ab49260
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
generate_equirect.py
View File

@@ -846,12 +846,14 @@ def main():
if args.allow_downloads: if args.allow_downloads:
os.environ["SKYMAP_ALLOW_DOWNLOADS"] = "1" os.environ["SKYMAP_ALLOW_DOWNLOADS"] = "1"
prompt = enrich_prompt(args.prompt)
if args.postprocess_input: if args.postprocess_input:
if not args.postprocess_output or not args.postprocess_tempdir: if not args.postprocess_output or not args.postprocess_tempdir:
parser.error("--postprocess-input requires --postprocess-output and --postprocess-tempdir") parser.error("--postprocess-input requires --postprocess-output and --postprocess-tempdir")
try: try:
result_path = postprocess_image( result_path = postprocess_image(
args.prompt, prompt,
os.path.abspath(args.postprocess_input), os.path.abspath(args.postprocess_input),
os.path.abspath(args.postprocess_output), os.path.abspath(args.postprocess_output),
os.path.abspath(args.postprocess_tempdir), os.path.abspath(args.postprocess_tempdir),
@@ -869,7 +871,6 @@ def main():
print(f"Generation failed: {e}") print(f"Generation failed: {e}")
raise raise
prompt = enrich_prompt(args.prompt)
base = sanitize_name(prompt) base = sanitize_name(prompt)
target = args.output or next_filename(args.output_dir, base, args.width, args.height) target = args.output or next_filename(args.output_dir, base, args.width, args.height)
output_abs = os.path.abspath(target) output_abs = os.path.abspath(target)