import gradio as gr
from transformers import pipeline

generator = pipeline(
    "text-generation",
    model="distilgpt2",
    pad_token_id=50256  # DistilGPT2 uses the same special tokens as GPT2
)

def generate_cv(name, education, experience):
    prompt = (
        f"Generate a CV:\n"
        f"Name: {name}\n"
        f"Education: {education}\n"
        f"Experience: {experience}\n"
        "CV:\n"
    )
    outputs = generator(
        prompt,
        max_new_tokens=50,
        do_sample=True,
        top_k=50,
        temperature=0.7
    )
    return outputs[0]["generated_text"]

demo = gr.Interface(
    fn=generate_cv,
    inputs=["text", "text", "text"],
    outputs="text",
    title="Automated CV Generator"
)

demo.launch()