from model_loader import load_model

model = load_model()

def tailor_resume(resume, job_description):
    prompt = f"""
    You are an AI resume assistant. Your job is to tailor this resume to the job description below.

    Resume:
    {resume}

    Job Description:
    {job_description}

    Tailored Resume:
    """
    # Truncate to fit distilgpt2 token limit (~1024 tokens = ~4000 characters)
    if len(prompt) > 3500:
        prompt = prompt[:3500]

    response = model(prompt, max_new_tokens=300)[0]
    return response[0]["generated_text"]