1#!/usr/bin/env python3
2"""
3Background Image Fetcher
4Fetches real images from Pexels for slide backgrounds.
5Uses web scraping (no API key required) or WebFetch tool integration.
6"""
7 
8import json
9import csv
10import re
11import sys
12from pathlib import Path
13 
14# Project root relative to this script
15PROJECT_ROOT = Path(__file__).parent.parent.parent.parent.parent
16TOKENS_PATH = PROJECT_ROOT / 'assets' / 'design-tokens.json'
17BACKGROUNDS_CSV = Path(__file__).parent.parent / 'data' / 'slide-backgrounds.csv'
18 
19 
20def resolve_token_reference(ref: str, tokens: dict) -> str:
21    """Resolve token reference like {primitive.color.ocean-blue.500} to hex value."""
22    if not ref or not ref.startswith('{') or not ref.endswith('}'):
23        return ref  # Already a value, not a reference
24 
25    # Parse reference: {primitive.color.ocean-blue.500}
26    path = ref[1:-1].split('.')  # ['primitive', 'color', 'ocean-blue', '500']
27    current = tokens
28    for key in path:
29        if isinstance(current, dict):
30            current = current.get(key)
31        else:
32            return None  # Invalid path
33    # Return $value if it's a token object
34    if isinstance(current, dict) and '$value' in current:
35        return current['$value']
36    return current
37 
38 
39def load_brand_colors():
40    """Load colors from assets/design-tokens.json for overlay gradients.
41 
42    Resolves semantic token references to actual hex values.
43    """
44    try:
45        with open(TOKENS_PATH) as f:
46            tokens = json.load(f)
47 
48        colors = tokens.get('primitive', {}).get('color', {})
49        semantic = tokens.get('semantic', {}).get('color', {})
50 
51        # Try semantic tokens first (preferred) - resolve references
52        if semantic:
53            primary_ref = semantic.get('primary', {}).get('$value')
54            secondary_ref = semantic.get('secondary', {}).get('$value')
55            accent_ref = semantic.get('accent', {}).get('$value')
56            background_ref = semantic.get('background', {}).get('$value')
57 
58            primary = resolve_token_reference(primary_ref, tokens)
59            secondary = resolve_token_reference(secondary_ref, tokens)
60            accent = resolve_token_reference(accent_ref, tokens)
61            background = resolve_token_reference(background_ref, tokens)
62 
63            if primary and secondary:
64                return {
65                    'primary': primary,
66                    'secondary': secondary,
67                    'accent': accent or primary,
68                    'background': background or '#0D0D0D',
69                }
70 
71        # Fallback: find first color palette with 500 value (primary)
72        primary_keys = ['ocean-blue', 'coral', 'blue', 'primary']
73        secondary_keys = ['golden-amber', 'purple', 'amber', 'secondary']
74        accent_keys = ['emerald', 'mint', 'green', 'accent']
75 
76        primary_color = None
77        secondary_color = None
78        accent_color = None
79 
80        for key in primary_keys:
81            if key in colors and isinstance(colors[key], dict):
82                primary_color = colors[key].get('500', {}).get('$value')
83                if primary_color:
84                    break
85 
86        for key in secondary_keys:
87            if key in colors and isinstance(colors[key], dict):
88                secondary_color = colors[key].get('500', {}).get('$value')
89                if secondary_color:
90                    break
91 
92        for key in accent_keys:
93            if key in colors and isinstance(colors[key], dict):
94                accent_color = colors[key].get('500', {}).get('$value')
95                if accent_color:
96                    break
97 
98        background = colors.get('dark', {}).get('800', {}).get('$value', '#0D0D0D')
99 
100        return {
101            'primary': primary_color or '#3B82F6',
102            'secondary': secondary_color or '#F59E0B',
103            'accent': accent_color or '#10B981',
104            'background': background,
105        }
106    except (FileNotFoundError, KeyError, TypeError):
107        # Fallback defaults
108        return {
109            'primary': '#3B82F6',
110            'secondary': '#F59E0B',
111            'accent': '#10B981',
112            'background': '#0D0D0D',
113        }
114 
115 
116def load_backgrounds_config():
117    """Load background configuration from CSV."""
118    config = {}
119    try:
120        with open(BACKGROUNDS_CSV, newline='') as f:
121            reader = csv.DictReader(f)
122            for row in reader:
123                config[row['slide_type']] = row
124    except FileNotFoundError:
125        print(f"Warning: {BACKGROUNDS_CSV} not found")
126    return config
127 
128 
129def get_overlay_css(style: str, brand_colors: dict) -> str:
130    """Generate overlay CSS using brand colors from design-tokens.json."""
131    overlays = {
132        'gradient-dark': f"linear-gradient(135deg, {brand_colors['background']}E6, {brand_colors['background']}B3)",
133        'gradient-brand': f"linear-gradient(135deg, {brand_colors['primary']}CC, {brand_colors['secondary']}99)",
134        'gradient-accent': f"linear-gradient(135deg, {brand_colors['accent']}99, transparent)",
135        'blur-dark': f"rgba(13,13,13,0.8)",
136        'desaturate-dark': f"rgba(13,13,13,0.7)",
137    }
138    return overlays.get(style, overlays['gradient-dark'])
139 
140 
141# Curated high-quality images from Pexels (free to use, pre-selected for brand aesthetic)
142CURATED_IMAGES = {
143    'hero': [
144        'https://images.pexels.com/photos/3861969/pexels-photo-3861969.jpeg?auto=compress&cs=tinysrgb&w=1920',
145        'https://images.pexels.com/photos/2582937/pexels-photo-2582937.jpeg?auto=compress&cs=tinysrgb&w=1920',
146        'https://images.pexels.com/photos/1089438/pexels-photo-1089438.jpeg?auto=compress&cs=tinysrgb&w=1920',
147    ],
148    'vision': [
149        'https://images.pexels.com/photos/3183150/pexels-photo-3183150.jpeg?auto=compress&cs=tinysrgb&w=1920',
150        'https://images.pexels.com/photos/3182812/pexels-photo-3182812.jpeg?auto=compress&cs=tinysrgb&w=1920',
151        'https://images.pexels.com/photos/3184291/pexels-photo-3184291.jpeg?auto=compress&cs=tinysrgb&w=1920',
152    ],
153    'team': [
154        'https://images.pexels.com/photos/3184418/pexels-photo-3184418.jpeg?auto=compress&cs=tinysrgb&w=1920',
155        'https://images.pexels.com/photos/3184338/pexels-photo-3184338.jpeg?auto=compress&cs=tinysrgb&w=1920',
156        'https://images.pexels.com/photos/3182773/pexels-photo-3182773.jpeg?auto=compress&cs=tinysrgb&w=1920',
157    ],
158    'testimonial': [
159        'https://images.pexels.com/photos/3184465/pexels-photo-3184465.jpeg?auto=compress&cs=tinysrgb&w=1920',
160        'https://images.pexels.com/photos/1181622/pexels-photo-1181622.jpeg?auto=compress&cs=tinysrgb&w=1920',
161    ],
162    'cta': [
163        'https://images.pexels.com/photos/3184339/pexels-photo-3184339.jpeg?auto=compress&cs=tinysrgb&w=1920',
164        'https://images.pexels.com/photos/3184298/pexels-photo-3184298.jpeg?auto=compress&cs=tinysrgb&w=1920',
165    ],
166    'problem': [
167        'https://images.pexels.com/photos/3760529/pexels-photo-3760529.jpeg?auto=compress&cs=tinysrgb&w=1920',
168        'https://images.pexels.com/photos/897817/pexels-photo-897817.jpeg?auto=compress&cs=tinysrgb&w=1920',
169    ],
170    'solution': [
171        'https://images.pexels.com/photos/3184292/pexels-photo-3184292.jpeg?auto=compress&cs=tinysrgb&w=1920',
172        'https://images.pexels.com/photos/3184644/pexels-photo-3184644.jpeg?auto=compress&cs=tinysrgb&w=1920',
173    ],
174    'hook': [
175        'https://images.pexels.com/photos/2582937/pexels-photo-2582937.jpeg?auto=compress&cs=tinysrgb&w=1920',
176        'https://images.pexels.com/photos/1089438/pexels-photo-1089438.jpeg?auto=compress&cs=tinysrgb&w=1920',
177    ],
178    'social': [
179        'https://images.pexels.com/photos/3184360/pexels-photo-3184360.jpeg?auto=compress&cs=tinysrgb&w=1920',
180        'https://images.pexels.com/photos/3184287/pexels-photo-3184287.jpeg?auto=compress&cs=tinysrgb&w=1920',
181    ],
182    'demo': [
183        'https://images.pexels.com/photos/1181675/pexels-photo-1181675.jpeg?auto=compress&cs=tinysrgb&w=1920',
184        'https://images.pexels.com/photos/3861958/pexels-photo-3861958.jpeg?auto=compress&cs=tinysrgb&w=1920',
185    ],
186}
187 
188 
189def get_curated_images(slide_type: str) -> list:
190    """Get curated images for slide type."""
191    return CURATED_IMAGES.get(slide_type, CURATED_IMAGES.get('hero', []))
192 
193 
194def get_pexels_search_url(keywords: str) -> str:
195    """Generate Pexels search URL for manual lookup."""
196    import urllib.parse
197    return f"https://www.pexels.com/search/{urllib.parse.quote(keywords)}/"
198 
199 
200def get_background_image(slide_type: str) -> dict:
201    """
202    Get curated image matching slide type and brand aesthetic.
203    Uses pre-selected Pexels images (no API/scraping needed).
204    """
205    brand_colors = load_brand_colors()
206    config = load_backgrounds_config()
207 
208    slide_config = config.get(slide_type)
209    overlay_style = 'gradient-dark'
210    keywords = slide_type
211 
212    if slide_config:
213        keywords = slide_config.get('search_keywords', slide_config.get('image_category', slide_type))
214        overlay_style = slide_config.get('overlay_style', 'gradient-dark')
215 
216    # Get curated images
217    urls = get_curated_images(slide_type)
218    if urls:
219        return {
220            'url': urls[0],
221            'all_urls': urls,
222            'overlay': get_overlay_css(overlay_style, brand_colors),
223            'attribution': 'Photo from Pexels (free to use)',
224            'source': 'pexels-curated',
225            'search_url': get_pexels_search_url(keywords),
226        }
227 
228    # Fallback: provide search URL for manual selection
229    return {
230        'url': None,
231        'overlay': get_overlay_css(overlay_style, brand_colors),
232        'keywords': keywords,
233        'search_url': get_pexels_search_url(keywords),
234        'available_types': list(CURATED_IMAGES.keys()),
235    }
236 
237 
238def generate_css_for_background(result: dict, slide_class: str = '.slide-with-bg') -> str:
239    """Generate CSS for a background slide."""
240    if not result.get('url'):
241        search_url = result.get('search_url', '')
242        return f"""/* No image scraped. Search manually: {search_url} */
243/* Overlay ready: {result.get('overlay', 'gradient-dark')} */
244"""
245 
246    return f"""{slide_class} {{
247    background-image: url('{result['url']}');
248    background-size: cover;
249    background-position: center;
250    position: relative;
251}}
252 
253{slide_class}::before {{
254    content: '';
255    position: absolute;
256    inset: 0;
257    background: {result['overlay']};
258}}
259 
260{slide_class} .content {{
261    position: relative;
262    z-index: 1;
263}}
264 
265/* {result.get('attribution', 'Pexels')} - {result.get('search_url', '')} */
266"""
267 
268 
269def main():
270    """CLI entry point."""
271    import argparse
272 
273    parser = argparse.ArgumentParser(description='Get background images for slides')
274    parser.add_argument('slide_type', nargs='?', help='Slide type (hero, vision, team, etc.)')
275    parser.add_argument('--list', action='store_true', help='List available slide types')
276    parser.add_argument('--css', action='store_true', help='Output CSS for the background')
277    parser.add_argument('--json', action='store_true', help='Output JSON')
278    parser.add_argument('--colors', action='store_true', help='Show brand colors')
279    parser.add_argument('--all', action='store_true', help='Show all curated URLs')
280 
281    args = parser.parse_args()
282 
283    if args.colors:
284        colors = load_brand_colors()
285        print("\nBrand Colors (from design-tokens.json):")
286        for name, value in colors.items():
287            print(f"  {name}: {value}")
288        return
289 
290    if args.list:
291        print("\nAvailable slide types (curated images):")
292        for slide_type, urls in CURATED_IMAGES.items():
293            print(f"  {slide_type}: {len(urls)} images")
294        return
295 
296    if not args.slide_type:
297        parser.print_help()
298        return
299 
300    result = get_background_image(args.slide_type)
301 
302    if args.json:
303        print(json.dumps(result, indent=2))
304    elif args.css:
305        print(generate_css_for_background(result))
306    elif args.all:
307        print(f"\nAll images for '{args.slide_type}':")
308        for i, url in enumerate(result.get('all_urls', []), 1):
309            print(f"  {i}. {url}")
310    else:
311        print(f"\nImage URL: {result['url']}")
312        print(f"Alternatives: {len(result.get('all_urls', []))} available (use --all)")
313        print(f"Overlay: {result['overlay']}")
314 
315 
316if __name__ == '__main__':
317    main()
318