"""Folding studio CLI."""

import typer

from folding_studio.commands.experiment import app as experiment_app
from folding_studio.commands.key import app as key_app
from folding_studio.commands.msa import app as msa_app
from folding_studio.commands.predict import (
    af2,
    boltz,
    chai,
    openfold,
    protenix,
    soloseq,
)

app = typer.Typer(
    no_args_is_help=True, context_settings={"help_option_names": ["-h", "--help"]}
)
app.add_typer(experiment_app, name="experiment")
app.add_typer(msa_app, name="msa")
app.add_typer(key_app, name="key")

predict_app = typer.Typer(no_args_is_help=True, help="Submit folding jobs")
predict_app.command()(af2)
predict_app.command()(openfold)
predict_app.command()(boltz)
predict_app.command()(chai)
predict_app.command()(protenix)
predict_app.command()(soloseq)


app.add_typer(predict_app, name="predict")

if __name__ == "__main__":
    app()