app.py

import gradio as gr
from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline

model_id = "HuggingFaceH4/zephyr-7b-alpha"  # ✅ sin restricciones

tokenizer = AutoTokenizer.from_pretrained(model_id)
model = AutoModelForCausalLM.from_pretrained(model_id)

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()