#!/usr/bin/env bash
# Creative showcase — push the envelope.
# Vertical social-media format (1440x2560) + cinema ultrawide (3104x1312).
set -euo pipefail

LAB="$(cd "$(dirname "${BASH_SOURCE[0]}")/../.." && pwd)"
PY="$LAB/.venv/bin/python"
MODEL="$LAB/mlx_models/hidream-o1-dev-q6"
OUT_V="$LAB/sample_outputs/showcase_creative/vertical_1440x2560"
OUT_W="$LAB/sample_outputs/showcase_creative/wide_3104x1312"
mkdir -p "$OUT_V" "$OUT_W"

run() {
  local out_dir="$1" name="$2" w="$3" h="$4" prompt="$5" seed="${6:-42}"
  echo "=== $name ${w}x${h} (seed=$seed) ==="
  cd "$LAB" && /usr/bin/time -l "$PY" scripts/hidream_o1/generate_hidream_o1_mlx.py \
    --model-path "$MODEL" \
    --prompt "$prompt" \
    --width "$w" --height "$h" \
    --output "$out_dir/$name.png" \
    --seed "$seed" 2>&1 | grep -E "loaded|using|generation:|saved" | tail -3
  echo ""
}

# === VERTICAL — social media influencer aesthetic (9:16 trained = 1440x2560) ===
run "$OUT_V" "01_fitness_influencer" 1440 2560 \
  "vertical full-body shot of a strong female fitness influencer in matching black lululemon top and high-waist leggings, mid-deadlift in a sunlit industrial gym, hair in a slick high ponytail, chalk dust in the air around her hands, hyperreal sweat detail, dramatic side lighting from large factory windows, professional sports photography, hyper sharp focus, 50mm lens" \
  101

run "$OUT_V" "02_glamour_mirror" 1440 2560 \
  "vertical full-body mirror selfie of a stylish female fashion influencer in a luxury Paris hotel bathroom, wearing a black silk slip dress and gold hoop earrings, holding a vintage Polaroid camera up to take the shot, marble walls and warm sconce lighting behind her, vogue magazine aesthetic, soft glamour, photoreal" \
  202

run "$OUT_V" "03_travel_iceland" 1440 2560 \
  "vertical full-body shot of a male travel photographer standing alone on a black volcanic Iceland beach with massive basalt sea stacks behind him, wearing a heavy grey wool turtleneck, dark technical pants, and a beanie, breath visible in the cold air, dramatic overcast moody light, churning North Atlantic waves, cinematic landscape photography, 35mm" \
  303

run "$OUT_V" "04_streetwear_tokyo" 1440 2560 \
  "vertical full-body shot of a Japanese streetwear influencer in front of a graffiti-covered wall in Harajuku, wearing oversized black raf simons hoodie with white text, baggy washed denim, chunky asics sneakers, ear-length neon green dyed hair, hands in pockets, head tilted with a slight smirk, golden-hour rim light, fashion editorial, photoreal" \
  404

# === CINEMA — ultrawide environments (21:9 trained = 3104x1312) ===
run "$OUT_W" "05_samurai_bamboo" 3104 1312 \
  "ultrawide cinematic shot of two samurai facing off in a misty bamboo forest at dawn, swords drawn, kimonos in deep red and indigo, golden first light filtering through tall bamboo stalks, particles of mist floating between them, anamorphic 35mm film, Akira Kurosawa composition, Roger Deakins cinematography" \
  505

run "$OUT_W" "06_astronaut_mars" 3104 1312 \
  "ultrawide cinematic shot of a single astronaut walking across a vast Martian canyon floor, deep red rocks rising hundreds of meters on either side, footprints behind in the rust dust, faint planet earth visible as a blue dot in the salmon sky, helmet visor reflecting the alien landscape, sense of cosmic scale and isolation, Denis Villeneuve aesthetic" \
  606

run "$OUT_W" "07_dragon_kingdom" 3104 1312 \
  "ultrawide cinematic shot of an enormous black dragon flying low over a medieval mountain kingdom at dusk, scales glistening, wings catching the last orange sunlight, peasants in the foreground looking up in awe from a stone bridge, the kingdom castle and snow-capped peaks in the deep background, fantasy epic, Peter Jackson composition, wide-angle lens, painterly atmosphere" \
  707

echo "=== creative showcase complete ==="
ls -la "$OUT_V" "$OUT_W"