#!/usr/bin/env bash
# Image2World: one DROID start frame -> video.
# First frame is at assets/droid/frames/droid_start_frame.jpg (extracted from a DROID MP4).
# Requires: HF access to nvidia/Cosmos-Predict2.5-2B and hf auth login.

set -e
cd "$(dirname "$0")"
export HF_HOME="${HF_HOME:-/data/cameron/vidgen/.cache/huggingface}"

# Use repo .venv if conda env doesn't have packages
PYTHON="${PYTHON:-.venv/bin/python}"
if ! "$PYTHON" -c "import cosmos_predict2" 2>/dev/null; then
  PYTHON="python"
fi

"$PYTHON" examples/inference.py \
  -i assets/droid/droid_image2world.json \
  -o outputs/droid_image2world \
  --model=2B/post-trained \
  --inference-type=image2world

echo "Outputs: outputs/droid_image2world/"