# shopmate/vision/photo_description.py
# Maya uploads a product photo -> ShopMate writes the description
import anthropic, base64
from pathlib import Path
client = anthropic.Anthropic()

SYSTEM = """You are a ThreadCo copywriter. When given a product photo:
1. Identify the garment type, colours, print/design, and apparent material
2. Write a 2-sentence product description in ThreadCo's voice:
   - Friendly, direct, never corporate
   - Always mention sustainability
   - No exclamation marks
   - Forbidden words: vibrant, perfect, stylish, must-have"""

def describe_from_photo(image_path: str, extra_info: str = "") -> str:
    """Generate a product description directly from a product photo."""
    data = base64.b64encode(Path(image_path).read_bytes()).decode()
    mime = "image/png" if image_path.endswith(".png") else "image/jpeg"
    resp = client.messages.create(
        model="claude-sonnet-4-6", max_tokens=150,
        system=SYSTEM,
        messages=[{"role":"user", "content":[
            {"type":"image","source":{"type":"base64","media_type":mime,"data":data}},
            {"type":"text", "text": f"Write the ThreadCo product description.{' Extra info: ' + extra_info if extra_info else ''}"}
        ]}]
    )
    return resp.content[0].text

def batch_describe_photos(photo_dir: str) -> list[dict]:
    """Process a folder of product photos in one go."""
    results = []
    for path in Path(photo_dir).glob("*.jpg"):
        desc = describe_from_photo(str(path))
        results.append({"file": path.name, "description": desc})
        print(f"{path.name}: {desc[:80]}...")
    return results

# Usage: Maya drags her product photos into a folder, runs one command
# results = batch_describe_photos("new_season_photos/")