from llm import llm
from state import AgentState

PROMPT = """
Review the following HTML code.
Fix errors if any.
Improve performance, accessibility, and responsiveness.
Return the corrected FULL HTML code.

HTML:
{}
"""

def reviewer(state: AgentState):
    result = llm.invoke(PROMPT.format(state["code"]))
    return {"code": result.content}