Marketing videos and image assets

import base64
import os
import time

import requests
from openai import OpenAI

API_KEY = os.environ["TUTUJIN_API_KEY"]
SITE = "https://www.tutujin.com"
client = OpenAI(base_url="https://www.tutujin.com/v1", api_key=API_KEY, timeout=300)
HEADERS = {"Authorization": "Bearer " + API_KEY}
UA = {"User-Agent": "Mozilla/5.0"}  # some file hosts reject requests without a browser-like User-Agent

# One template, many products and selling points (the fields can be filled in by qwen3.5-plus).
FRAME = "{product}, {scene}, clean commercial photography, soft daylight, brand colors {colors}, no added text"


def download(url, path):
    r = requests.get(url, headers=UA, timeout=300)
    r.raise_for_status()
    with open(path, "wb") as f:
        f.write(r.content)


def key_frame(path, **fields):
    image = client.images.generate(model="nano-banana-pro", prompt=FRAME.format(**fields)).data[0]
    if image.b64_json:
        with open(path, "wb") as f:
            f.write(base64.b64decode(image.b64_json))
    else:
        download(image.url, path)  # temporary URL: store the file right away


def video_clips(first_frame_url, motion_prompt):
    """Video generation is an async task: submit, poll, download. One task returns 4 clips."""
    submit = requests.post(
        SITE + "/mj/submit/video",
        headers=HEADERS,
        json={"prompt": motion_prompt, "image": first_frame_url, "motion": "low"},
        timeout=60,
    ).json()
    if submit.get("code") != 1:
        raise RuntimeError(submit.get("description"))
    while True:
        time.sleep(10)
        task = requests.get(f"{SITE}/mj/task/{submit['result']}/fetch", headers=HEADERS, timeout=30).json()
        if task.get("status") in ("SUCCESS", "FAILURE"):
            break
    if task["status"] == "FAILURE":
        raise RuntimeError(task.get("failReason"))
    return [v["url"] for v in task.get("videoUrls") or []]


key_frame("mug-shot-1.png", product="A white ceramic mug", scene="on a wooden breakfast table", colors="white and teal")
# after a person picks the frame, upload it to your own storage and pass its URL
for i, url in enumerate(video_clips("https://cdn.example.com/mug-shot-1.png", "slow push-in, steam rising from the mug")):
    download(url, f"mug-shot-1-clip-{i + 1}.mp4")