import gradio as gr
from transformers import pipeline

# Usamos 'SantaCoder', un modelo de 1.1B parámetros experto en código.
# Es mucho más ligero (2GB) y cabe perfectamente en el Space gratuito.
checkpoint = "bigcode/santacoder"
pipe = pipeline("text-generation", model=checkpoint, trust_remote_code=True)

def generate_code(prompt):
    # Genera código basado en tu entrada
    output = pipe(prompt, max_length=100, do_sample=True, temperature=0.2)
    return output[0]['generated_text']

# Esta interfaz es lo que evita el "Exit Code: 0"
demo = gr.Interface(
    fn=generate_code, 
    inputs=gr.Textbox(placeholder="Escribe el inicio de una función Python..."), 
    outputs="text",
    title="Asistente de Programación IA"
)

demo.launch()