main.py

from cli import parse_args
from git_utils import get_staged_files
from commit_logic import create_commit_prompt
from helper.ollama_client_direct import generate_commit_direct

def main():
    args = parse_args()
    repo_type = "yadm" if args.yadm else "git"
    files = get_staged_files(repo_type)

    if not files:
        print("No changes to commit.")
        return

    use_content = args.smallchange or (len(files) <= 2 and not args.bigchange)
    prompt = create_commit_prompt(files, use_content, repo_type, args.experimentalSum)
    message = generate_commit_message(prompt, args.model, args.temperature)
    print(message)

if __name__ == "__main__":
    main()