python.py

import subprocess
import requests
import sys
import json
import os

config_path = os.path.join(os.path.dirname(__file__), 'config.json')
with open(config_path, 'r') as f:
    config = json.load(f)

MODEL = config.get('model', 'qwen2.5-coder:7b')

def get_diff():
    result = subprocess.run(
        [
            'git', 'diff', '--staged',
            '-U5',               
            '--function-context',   
            '-w',               
            '--ignore-blank-lines'
        ],
        capture_output=True,
        text=True
    )
    return result.stdout
    
def classify_diff(diff):
    added = []
    removed = []

    for line in diff.splitlines():
        if line.startswith('+') and not line.startswith('+++'):
            added.append(line[1:].strip())
        elif line.startswith('-') and not line.startswith('---'):
            removed.append(line[1:].strip())

    added = [l for l in added if l]
    removed = [l for l in removed if l]

    total_changes = len(added) + len(removed)

    if total_changes <= 2:
        return "trivial"

    if sorted(added) == sorted(removed):
        if total_changes > 10:
            return "reorder"
        else:
            return "trivial"

    return "functional"

def clean_message(msg):
    msg = msg.strip().lower()

    if not msg:
        return "update code logic"

    if len(msg.split()) > 10:
        return "update code logic"

    if any(word in msg for word in ["format", "whitespace", "blank"]):
        return "minor code tweak"

    return msg

def generate_commit_message(diff):
    try:
        response = requests.post(
            'http://localhost:11434/api/generate',
            json={
                "model": "qwen2.5-coder:7b",
                "prompt": f"""Write a git commit message (4-8 words).

            ONLY describe meaningful behavioral changes.

            Make your responces short but NO LONGER THAN 8 WORDS.

            The Commit message could be what then theme of the changes was and them __ improvement or the them of the change with __ added

            Use the surrounding function context to understand intent.

            IGNORE:
            - formatting
            - whitespace
            - reordering

            Diff with context:
            {diff}
            """,
                "stream": False
            },
            timeout=10
        )

        data = response.json()

        if 'response' not in data:
            print("API error:", data)
            return "update code logic"

        return data['response'].strip()

    except Exception as e:
        print("Request failed:", e)
        return "update code logic"


if __name__ == "__main__":

    dry_run = '--dry-run' in sys.argv
    verbose = '--verbose' in sys.argv

    if verbose:
        print(f"Diff being sent:\n{diff}\n")

    diff = get_diff()
    if not diff.strip():
        print("No staged changes — run 'git add' first")
        exit(0)


    dry_run = '--dry-run' in sys.argv

    change_type = classify_diff(diff)
    if change_type == "trivial":
        message = "minor code tweak"
    elif change_type == "reorder":
        message = "refactor code structure"
    else:
        message = generate_commit_message(diff)
        message = clean_message(message)

    print(f"Change type: {change_type}")  
    print(f"Committing: {message}")

    if dry_run:
        print("Dry run — skipping commit")
    else:
        subprocess.run(['git', 'commit', '-m', message])