"""文生 3D — text to model.

    tripocli text -p "a low-poly wooden barrel" --smart-mesh --wait -o barrel.glb
"""

from . import _common

NAME = "text"
HELP = "文本生成 3D 模型（文生 3D）"


def add_arguments(parser):
    parser.add_argument("-p", "--prompt", required=True, help="模型描述提示词（英文效果更佳）")
    _common.add_generation_options(parser)
    _common.add_wait_options(parser, "生成完成后保存的 .glb 路径")


def run(ctx):
    args = ctx.args
    body = _common.generation_body(args)
    data = ctx.client.get_data(
        "POST", "/v2/studio/operation/text_to_model", body={"prompt": args.prompt, **body}
    )
    return _common.finish_single(ctx, data, output=args.output, wait=args.wait)