1#!/usr/bin/env python3
2# -*- coding: utf-8 -*-
3"""
4Slide Search CLI - Search slide design databases for strategies, layouts, copy, and charts
5"""
6 
7import sys
8import json
9import argparse
10from slide_search_core import (
11    search, search_all, AVAILABLE_DOMAINS,
12    search_with_context, get_layout_for_goal, get_typography_for_slide,
13    get_color_for_emotion, get_background_config
14)
15 
16 
17def format_result(result, domain):
18    """Format a single search result for display"""
19    output = []
20 
21    if domain == "strategy":
22        output.append(f"**{result.get('strategy_name', 'N/A')}**")
23        output.append(f"  Slides: {result.get('slide_count', 'N/A')}")
24        output.append(f"  Structure: {result.get('structure', 'N/A')}")
25        output.append(f"  Goal: {result.get('goal', 'N/A')}")
26        output.append(f"  Audience: {result.get('audience', 'N/A')}")
27        output.append(f"  Tone: {result.get('tone', 'N/A')}")
28        output.append(f"  Arc: {result.get('narrative_arc', 'N/A')}")
29        output.append(f"  Source: {result.get('sources', 'N/A')}")
30 
31    elif domain == "layout":
32        output.append(f"**{result.get('layout_name', 'N/A')}**")
33        output.append(f"  Use case: {result.get('use_case', 'N/A')}")
34        output.append(f"  Zones: {result.get('content_zones', 'N/A')}")
35        output.append(f"  Visual weight: {result.get('visual_weight', 'N/A')}")
36        output.append(f"  CTA: {result.get('cta_placement', 'N/A')}")
37        output.append(f"  Recommended: {result.get('recommended_for', 'N/A')}")
38        output.append(f"  Avoid: {result.get('avoid_for', 'N/A')}")
39        output.append(f"  CSS: {result.get('css_structure', 'N/A')}")
40 
41    elif domain == "copy":
42        output.append(f"**{result.get('formula_name', 'N/A')}**")
43        output.append(f"  Components: {result.get('components', 'N/A')}")
44        output.append(f"  Use case: {result.get('use_case', 'N/A')}")
45        output.append(f"  Template: {result.get('example_template', 'N/A')}")
46        output.append(f"  Emotion: {result.get('emotion_trigger', 'N/A')}")
47        output.append(f"  Slide type: {result.get('slide_type', 'N/A')}")
48        output.append(f"  Source: {result.get('source', 'N/A')}")
49 
50    elif domain == "chart":
51        output.append(f"**{result.get('chart_type', 'N/A')}**")
52        output.append(f"  Best for: {result.get('best_for', 'N/A')}")
53        output.append(f"  Data type: {result.get('data_type', 'N/A')}")
54        output.append(f"  When to use: {result.get('when_to_use', 'N/A')}")
55        output.append(f"  When to avoid: {result.get('when_to_avoid', 'N/A')}")
56        output.append(f"  Max categories: {result.get('max_categories', 'N/A')}")
57        output.append(f"  Slide context: {result.get('slide_context', 'N/A')}")
58        output.append(f"  CSS: {result.get('css_implementation', 'N/A')}")
59        output.append(f"  Accessibility: {result.get('accessibility_notes', 'N/A')}")
60 
61    return "\n".join(output)
62 
63 
64def format_context(context):
65    """Format contextual recommendations for display."""
66    output = []
67    output.append(f"\n=== CONTEXTUAL RECOMMENDATIONS ===")
68    output.append(f"Inferred Goal: {context.get('inferred_goal', 'N/A')}")
69    output.append(f"Position: Slide {context.get('slide_position')} of {context.get('total_slides')}")
70 
71    if context.get('recommended_layout'):
72        output.append(f"\n📐 Layout: {context['recommended_layout']}")
73        output.append(f"   Direction: {context.get('layout_direction', 'N/A')}")
74        output.append(f"   Visual Weight: {context.get('visual_weight', 'N/A')}")
75 
76    if context.get('typography'):
77        typo = context['typography']
78        output.append(f"\n📝 Typography:")
79        output.append(f"   Primary: {typo.get('primary_size', 'N/A')}")
80        output.append(f"   Secondary: {typo.get('secondary_size', 'N/A')}")
81        output.append(f"   Contrast: {typo.get('weight_contrast', 'N/A')}")
82 
83    if context.get('color_treatment'):
84        color = context['color_treatment']
85        output.append(f"\n🎨 Color Treatment:")
86        output.append(f"   Background: {color.get('background', 'N/A')}")
87        output.append(f"   Text: {color.get('text_color', 'N/A')}")
88        output.append(f"   Accent: {color.get('accent_usage', 'N/A')}")
89 
90    if context.get('should_break_pattern'):
91        output.append(f"\n⚡ Pattern Break: YES (use contrasting layout)")
92 
93    if context.get('should_use_full_bleed'):
94        output.append(f"\n🖼️ Full Bleed: Recommended for emotional impact")
95 
96    if context.get('use_background_image') and context.get('background'):
97        bg = context['background']
98        output.append(f"\n📸 Background Image:")
99        output.append(f"   Category: {bg.get('image_category', 'N/A')}")
100        output.append(f"   Overlay: {bg.get('overlay_style', 'N/A')}")
101        output.append(f"   Keywords: {bg.get('search_keywords', 'N/A')}")
102 
103    output.append(f"\n✨ Animation: {context.get('animation_class', 'animate-fade-up')}")
104 
105    return "\n".join(output)
106 
107 
108def main():
109    parser = argparse.ArgumentParser(
110        description="Search slide design databases",
111        formatter_class=argparse.RawDescriptionHelpFormatter,
112        epilog="""
113Examples:
114  search-slides.py "investor pitch"           # Auto-detect domain (strategy)
115  search-slides.py "funnel conversion" -d chart
116  search-slides.py "headline hook" -d copy
117  search-slides.py "two column" -d layout
118  search-slides.py "startup funding" --all    # Search all domains
119  search-slides.py "metrics dashboard" --json # JSON output
120 
121Contextual Search (Premium System):
122  search-slides.py "problem slide" --context --position 2 --total 9
123  search-slides.py "cta" --context --position 9 --total 9 --prev-emotion frustration
124        """
125    )
126 
127    parser.add_argument("query", help="Search query")
128    parser.add_argument("-d", "--domain", choices=AVAILABLE_DOMAINS,
129                        help="Specific domain to search (auto-detected if not specified)")
130    parser.add_argument("-n", "--max-results", type=int, default=3,
131                        help="Maximum results to return (default: 3)")
132    parser.add_argument("--all", action="store_true",
133                        help="Search across all domains")
134    parser.add_argument("--json", action="store_true",
135                        help="Output as JSON")
136 
137    # Contextual search options
138    parser.add_argument("--context", action="store_true",
139                        help="Use contextual search with layout/typography/color recommendations")
140    parser.add_argument("--position", type=int, default=1,
141                        help="Slide position in deck (1-based, default: 1)")
142    parser.add_argument("--total", type=int, default=9,
143                        help="Total slides in deck (default: 9)")
144    parser.add_argument("--prev-emotion", type=str, default=None,
145                        help="Previous slide's emotion for contrast calculation")
146 
147    args = parser.parse_args()
148 
149    # Contextual search mode
150    if args.context:
151        result = search_with_context(
152            args.query,
153            slide_position=args.position,
154            total_slides=args.total,
155            previous_emotion=args.prev_emotion
156        )
157 
158        if args.json:
159            print(json.dumps(result, indent=2))
160        else:
161            print(format_context(result['context']))
162 
163            # Also show base search results
164            if result.get('base_results'):
165                print("\n\n=== RELATED SEARCH RESULTS ===")
166                for domain, data in result['base_results'].items():
167                    print(f"\n--- {domain.upper()} ---")
168                    for item in data['results']:
169                        print(format_result(item, domain))
170                        print()
171        return
172 
173    if args.all:
174        results = search_all(args.query, args.max_results)
175 
176        if args.json:
177            print(json.dumps(results, indent=2))
178        else:
179            if not results:
180                print(f"No results found for: {args.query}")
181                return
182 
183            for domain, data in results.items():
184                print(f"\n=== {domain.upper()} ===")
185                print(f"File: {data['file']}")
186                print(f"Results: {data['count']}")
187                print()
188                for result in data['results']:
189                    print(format_result(result, domain))
190                    print()
191    else:
192        result = search(args.query, args.domain, args.max_results)
193 
194        if args.json:
195            print(json.dumps(result, indent=2))
196        else:
197            if result.get("error"):
198                print(f"Error: {result['error']}")
199                return
200 
201            print(f"Domain: {result['domain']}")
202            print(f"Query: {result['query']}")
203            print(f"File: {result['file']}")
204            print(f"Results: {result['count']}")
205            print()
206 
207            if result['count'] == 0:
208                print("No matching results found.")
209                return
210 
211            for i, item in enumerate(result['results'], 1):
212                print(f"--- Result {i} ---")
213                print(format_result(item, result['domain']))
214                print()
215 
216 
217if __name__ == "__main__":
218    main()
219