#!/bin/bash
# 图片生成入口
# 用法: ./image.sh --output-dir /abs/path --name 中文名 "描述" [--aspect-ratio 16:9]
set -euo pipefail

OUTPUT_DIR=""
NAME=""
PROMPT=""
ASPECT_RATIO=""

while [ $# -gt 0 ]; do
    case "$1" in
        --output-dir) OUTPUT_DIR="$2"; shift 2 ;;
        --name) NAME="$2"; shift 2 ;;
        --aspect-ratio) ASPECT_RATIO="$2"; shift 2 ;;
        *)
            if [ -z "$PROMPT" ]; then
                PROMPT="$1"
            fi
            shift
            ;;
    esac
done

if [ -z "$OUTPUT_DIR" ] || [ -z "$NAME" ] || [ -z "$PROMPT" ]; then
    echo "用法: ./image.sh --output-dir <绝对路径> --name <中文名> \"描述\" [--aspect-ratio 16:9]"
    echo "示例: ./image.sh --output-dir /root/.openclaw/workspace --name 日落风景 \"a sunset landscape\""
    exit 1
fi

SCRIPT_DIR="$(cd "$(dirname "$0")" && pwd)"

ARGS=()
ARGS+=("image")
ARGS+=("$PROMPT")
ARGS+=(--output-dir "$OUTPUT_DIR")
ARGS+=(--name "$NAME")
[ -n "$ASPECT_RATIO" ] && ARGS+=(--aspect-ratio "$ASPECT_RATIO")

python3 "$SCRIPT_DIR/generate.py" "${ARGS[@]}"
