import requests
import os
import base64

AIPASS_API_KEY = os.environ["AIPASS_API_KEY"]
BASE_URL = "https://aipass.one/apikey/v1"

def redesign_room(image_path, style="modern minimalist"):
    """Redesign a room in the specified style.
    
    Args:
        image_path: Path to room photo
        style: Design style (e.g. "Scandinavian", "industrial", "Japanese zen")
    
    Returns:
        str: AI response with redesigned image
    """
    with open(image_path, "rb") as f:
        b64 = base64.b64encode(f.read()).decode()
    
    ext = image_path.rsplit(".", 1)[-1].lower()
    mime = {"png": "image/png", "jpg": "image/jpeg", "jpeg": "image/jpeg", "webp": "image/webp"}.get(ext, "image/png")
    
    r = requests.post(
        f"{BASE_URL}/chat/completions",
        headers={
            "Authorization": f"Bearer {AIPASS_API_KEY}",
            "Content-Type": "application/json"
        },
        json={
            "model": "gemini/gemini-3-pro-image-preview",
            "messages": [{
                "role": "user",
                "content": [
                    {"type": "text", "text": f"Redesign this room in {style} style. Keep the layout but change all furniture, decor, and colors to match {style} design."},
                    {"type": "image_url", "image_url": {"url": f"data:{mime};base64,{b64}"}}
                ]
            }]
        }
    )
    return r.json()["choices"][0]["message"]["content"]