test_local.py

# test_local.py
import os
os.environ['OPENROUTER_API_KEY'] = 'your-api-key-here'

from app import create_ui_design

# Test single page
result = create_ui_design(
    "Create a landing page for a tech startup",
    os.environ['OPENROUTER_API_KEY']
)

# Test multi-page
result = create_ui_design(
    "Create 3 pages: landing, about, contact",
    os.environ['OPENROUTER_API_KEY']
)