from langchain_core.messages import AIMessage from utils.inference import call_model import re def run(state): design = state["architecture"] prompt = f"""You are a front-end engineer. Create a complete HTML page with embedded CSS for the following UI design: {design} Requirements: - Self-contained HTML document with - CSS must be embedded in {cleaned} """ return { "messages": state["messages"] + [AIMessage(content=final_html)], "html_output": final_html }