feat(music-lab): background.py — Pexels Video API + orchestrator video_loop 분기

This commit is contained in:
2026-05-09 13:13:42 +09:00
parent f0c0c18beb
commit e754fb30f5
3 changed files with 131 additions and 1 deletions

View File

@@ -6,7 +6,8 @@ import os
import sqlite3
from app import db
from . import cover, video, thumb, metadata, review, youtube
from . import cover, video, thumb, metadata, review, youtube, background, storage
from .gradient import make_gradient_with_title
logger = logging.getLogger("music-lab.orchestrator")
@@ -167,13 +168,31 @@ def _fetch_track_fallback(track_id: int) -> dict | None:
async def _run_cover(p, ctx, feedback):
setup = db.get_youtube_setup()
vd = setup["visual_defaults"]
bg_mode = p.get("background_mode") or vd.get("default_background_mode", "static")
keyword = p.get("background_keyword") or vd.get("default_background_keyword", "")
if bg_mode == "video_loop":
# Pexels 영상 다운로드 시도 — 성공 여부와 무관하게 cover.jpg는 그라데이션으로 별도 생성
# (실패 시 video.py가 cover.jpg를 fallback 배경으로 사용 가능)
await background.fetch_video_loop(p["id"], keyword)
out_path = os.path.join(storage.pipeline_dir(p["id"]), "cover.jpg")
make_gradient_with_title(ctx["genre"], ctx["title"], out_path)
return {"next_state": "cover_pending",
"fields": {"cover_url": storage.media_url(p["id"], "cover.jpg")}}
# 정적 모드 — 기존 cover.generate 흐름
prompts = setup["cover_prompts"]
template = prompts.get(ctx["genre"].lower(), prompts.get("default", ""))
image_source = vd.get("background_image_source", "ai")
out = await cover.generate(
pipeline_id=p["id"], genre=ctx["genre"],
prompt_template=template,
mood=", ".join(ctx["moods"] or []),
track_title=ctx["title"], feedback=feedback,
image_source=image_source,
background_keyword=keyword,
)
return {"next_state": "cover_pending", "fields": {"cover_url": out["url"]}}