diff --git a/pyml_cli/cli/project.py b/pyml_cli/cli/project.py index 7408705..b8ba850 100644 --- a/pyml_cli/cli/project.py +++ b/pyml_cli/cli/project.py @@ -1,8 +1,11 @@ """Project initialization CLI.""" +import os from pathlib import Path from cookiecutter.main import cookiecutter +from loguru import logger +from sh import git from typer import Typer app = Typer() @@ -13,4 +16,13 @@ @app.command() def init(): """Initialize project from template.""" - cookiecutter(str(PROJECT_TEMPLATE_DIR.resolve())) + output_dir = cookiecutter(str(PROJECT_TEMPLATE_DIR.resolve())) + + if output_dir: + project_path = Path(output_dir) + try: + os.chdir(project_path) + git("init", "-b", "main") + logger.info(f"✓ Initialized git repository in {project_path}") + except Exception as e: + print(f"⚠ Git initialization failed: {e} - skipping git initialization")