code_reviewer.py

from transformers import pipeline

# Load model once
reviewer = pipeline("text2text-generation", model="google/flan-t5-base")

def review_code(code_text):
    prompt = f"Review this code and give feedback:\n{code_text}"
    output = reviewer(prompt, max_length=512, do_sample=False)
    return output[0]['generated_text']