kumar1907/Code_Review_Agent

1from transformers import pipeline2 3# Load once4reviewer = pipeline("text2text-generation", model="google/flan-t5-base")5 6def review_code(code_text):7    # Ask model for both feedback + improved code8    prompt = f"""9    You are a code review assistant. Analyze the following code diff or snippet:10    {code_text}11 12    1. Give clear review feedback with improvements.13    2. Then provide a refactored/improved version of the code.14    Format the response as:15    FEEDBACK:16    <your feedback here>17 18    CODE:19    <improved code here>20    """21 22    output = reviewer(prompt, max_length=800, do_sample=False)23    response = output[0]['generated_text']24 25    # Split into feedback & code26    if "CODE:" in response:27        feedback, code = response.split("CODE:", 1)28        feedback = feedback.replace("FEEDBACK:", "").strip()29        return feedback, code.strip()30    else:31        return response, ""32