umair894/quickstart-trackio

1import argparse2 3from trackio import show4 5 6def main():7    parser = argparse.ArgumentParser(description="Trackio CLI")8    subparsers = parser.add_subparsers(dest="command")9 10    ui_parser = subparsers.add_parser(11        "show", help="Show the Trackio dashboard UI for a project"12    )13    ui_parser.add_argument(14        "--project", required=False, help="Project name to show in the dashboard"15    )16    ui_parser.add_argument(17        "--theme",18        required=False,19        default="citrus",20        help="A Gradio Theme to use for the dashboard instead of the default, can be a built-in theme (e.g. 'soft', 'citrus'), or a theme from the Hub (e.g. 'gstaff/xkcd').",21    )22    ui_parser.add_argument(23        "--mcp-server",24        action="store_true",25        help="Enable MCP server functionality. The Trackio dashboard will be set up as an MCP server and certain functions will be exposed as MCP tools.",26    )27 28    args = parser.parse_args()29 30    if args.command == "show":31        show(args.project, args.theme, args.mcp_server)32    else:33        parser.print_help()34 35 36if __name__ == "__main__":37    main()38