#!/usr/bin/env bash
set -euo pipefail

input_image_path=/Users/lehongwu/Projects/others/lyrics/VideoGeneration/output_video/leaves_video_0309184011/debug_frame6.png

export http_proxy="http://127.0.0.1:7890"
export https_proxy="http://127.0.0.1:7890"

if [[ -z "${GEMINI_API_KEY:-}" ]]; then
  echo "Error: GEMINI_API_KEY is not set."
  echo 'Run: export GEMINI_API_KEY="your_api_key"'
  exit 1
fi

# if ! python3 -c "from google import genai" >/dev/null 2>&1; then
#   echo "Installing dependency: google-genai"
#   python3 -m pip install --upgrade google-genai
# fi

datetime=$(date +%m%d%H%M%S)
name=flowers_to_leaves
output_dir="output_image/${name}_${datetime}"

# prompt="
# Transform this image into: Autumn leaves drifting on the black background, but the locations can follow the original image.
# The autumn leaves should look diverse in shapes, and colors should be red and gold.
# The sizes are diverse, especially small ones, as if they are far from the camera.
# Not all leaves are facing the camera, instead, they are in random directions as if drifting in the wind.
# But overall, the leaves should be sparse and not too large.
# "

prompt="Change the background to gold and with soft sun glows from top."

python gen_image_image_cond.py \
  --prompt "$prompt" \
  --input-image-path "$input_image_path" \
  --model gemini-3.1-flash-image-preview \
  --aspect-ratio 16:9 \
  --resolution 2K \
  --number-of-images 1 \
  --name "$name" \
  --output-dir "$output_dir"