#!/usr/bin/env python

# Copyright 2026 The HuggingFace Team. All rights reserved.

#

# Licensed under the Apache License, Version 2.0 (the "License");

# you may not use this file except in compliance with the License.

# You may obtain a copy of the License at

#

# http://www.apache.org/licenses/LICENSE-2.0

#

# Unless required by applicable law or agreed to in writing, software

# distributed under the License is distributed on an "AS IS" BASIS,

# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

# See the License for the specific language governing permissions and

# limitations under the License.

from argparse import ArgumentParser

from huggingface_hub.cli._output import OutputFormat, out

from .custom_blocks import CustomBlocksCommand

from .env import EnvironmentCommand

from .fp16_safetensors import FP16SafetensorsCommand

from .run import RunCommand

from .schema import SchemaCommand

from .skills import SkillsCommand

def main():

parser = ArgumentParser(

prog="diffusers-cli",

usage="\n diffusers-cli [--format <fmt>] <command> [options]",

)

parser._optionals.title = "Options"

parser.add_argument(

"--format",

choices=[m.value for m in OutputFormat],

default=OutputFormat.auto.value,

help=(

"Output format. 'auto' (default) picks 'agent' when an AI coding agent is detected "

"(via CLAUDECODE/CURSOR_AI/AIDER_AI_CONTEXT/... env vars) and 'human' otherwise. "

"Must appear before the subcommand."

),

)

commands_parser = parser.add_subparsers(title="Commands", metavar="<command>")

# Register commands

EnvironmentCommand.register_subcommand(commands_parser)

FP16SafetensorsCommand.register_subcommand(commands_parser)

CustomBlocksCommand.register_subcommand(commands_parser)

RunCommand.register_subcommand(commands_parser)

SchemaCommand.register_subcommand(commands_parser)

SkillsCommand.register_subcommand(commands_parser)

# Let's go

args = parser.parse_args()

out.set_mode(OutputFormat(args.format))

if not hasattr(args, "func"):

parser.print_help()

exit(1)

# Run

service = args.func(args)

service.run()

if __name__ == "__main__":

main()