commit

commit.py

@@ -0,0 +1,36 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""
+Created on Thu Apr 17 20:18:35 2025
+
+@author: rick
+"""
+
+import gradio as gr
+from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
+
+model_id = "mistralai/Mistral-7B-Instruct-v0.1"
+
+tokenizer = AutoTokenizer.from_pretrained(model_id)
+model = AutoModelForCausalLM.from_pretrained(
+    model_id,
+    device_map="auto",
+    torch_dtype="auto",
+)
+
+pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
+
+def generar_commit(diff):
+    prompt = f"Escribí un mensaje de commit claro, corto y técnico basado en los siguientes cambios:\n{diff}"
+    output = pipe(prompt, max_new_tokens=60, do_sample=True, temperature=0.7)
+    return output[0]["generated_text"]
+
+demo = gr.Interface(
+    fn=generar_commit,
+    inputs=gr.Textbox(lines=10, label="Código diff"),
+    outputs="text",
+    title="🤖 Commit Generator",
+    description="Pega aquí tus cambios (git diff) y generá un mensaje de commit automático.",
+)
+
+demo.launch()

requirements.txt

@@ -0,0 +1,3 @@
+transformers
+torch
+gradio