GreenLeaf E-commerce — AI Shopping Visibility for Organic Cosmetics

"""
product_schema_generator.py — Enhanced Product schema for AI visibility
Generates comprehensive JSON-LD for Shopify products with
ingredient data, certifications, and review aggregation.
"""

import json
from typing import List, Dict, Optional
from dataclasses import dataclass, field


@dataclass
class ProductGEO:
    name: str
    sku: str
    description: str
    ingredients: List[str]
    certifications: List[str]
    skin_types: List[str]
    concerns: List[str]
    price: float
    currency: str = "USD"
    reviews: List[Dict] = field(default_factory=list)

    def generate_product_schema(self) -> Dict:
        """Build comprehensive Product JSON-LD for AI comprehension."""
        aggregate = self._compute_aggregate_rating()
        return {
            "@context": "https://schema.org",
            "@type": "Product",
            "name": self.name,
            "sku": self.sku,
            "brand": {"@type": "Brand", "name": "GreenLeaf"},
            "hasIngredient": [
                {"@type": "ChemicalSubstance", "name": ing}
                for ing in self.ingredients
            ],
            "certification": [
                {"@type": "Certification", "name": cert}
                for cert in self.certifications
            ],
            "audience": {
                "@type": "PeopleAudience",
                "healthCondition": ", ".join(self.concerns)
            },
            "aggregateRating": aggregate,
            "review": self._build_review_schema()[:5]
        }

    def _compute_aggregate_rating(self) -> Dict:
        if not self.reviews: return {}
        avg = sum(r["rating"] for r in self.reviews) / len(self.reviews)
        return {
            "@type": "AggregateRating",
            "ratingValue": round(avg, 1),
            "reviewCount": len(self.reviews),
            "bestRating": 5
        }