1#!/usr/bin/env python3
2# -*- coding: utf-8 -*-
3"""
4Logo Design Search - CLI for searching logo design guidelines
5Usage: python search.py "<query>" [--domain <domain>] [--max-results 3]
6       python search.py "<query>" --design-brief [-p "Brand Name"]
7 
8Domains: style, color, industry
9"""
10 
11import argparse
12from core import CSV_CONFIG, MAX_RESULTS, search, search_all
13 
14 
15def format_output(result):
16    """Format results for Claude consumption (token-optimized)"""
17    if "error" in result:
18        return f"Error: {result['error']}"
19 
20    output = []
21    output.append(f"## Logo Design Search Results")
22    output.append(f"**Domain:** {result['domain']} | **Query:** {result['query']}")
23    output.append(f"**Source:** {result['file']} | **Found:** {result['count']} results\n")
24 
25    for i, row in enumerate(result['results'], 1):
26        output.append(f"### Result {i}")
27        for key, value in row.items():
28            value_str = str(value)
29            if len(value_str) > 300:
30                value_str = value_str[:300] + "..."
31            output.append(f"- **{key}:** {value_str}")
32        output.append("")
33 
34    return "\n".join(output)
35 
36 
37def generate_design_brief(query, brand_name=None):
38    """Generate a comprehensive logo design brief based on query"""
39    results = search_all(query, max_results=2)
40 
41    output = []
42    output.append("=" * 60)
43    if brand_name:
44        output.append(f"  LOGO DESIGN BRIEF: {brand_name.upper()}")
45    else:
46        output.append("  LOGO DESIGN BRIEF")
47    output.append("=" * 60)
48    output.append(f"  Query: {query}")
49    output.append("=" * 60)
50    output.append("")
51 
52    # Industry recommendations
53    if "industry" in results:
54        output.append("## INDUSTRY ANALYSIS")
55        for r in results["industry"]:
56            output.append(f"**Industry:** {r.get('Industry', 'N/A')}")
57            output.append(f"- Recommended Styles: {r.get('Recommended Styles', 'N/A')}")
58            output.append(f"- Colors: {r.get('Primary Colors', 'N/A')}")
59            output.append(f"- Typography: {r.get('Typography', 'N/A')}")
60            output.append(f"- Symbols: {r.get('Common Symbols', 'N/A')}")
61            output.append(f"- Mood: {r.get('Mood', 'N/A')}")
62            output.append(f"- Best Practices: {r.get('Best Practices', 'N/A')}")
63            output.append(f"- Avoid: {r.get('Avoid', 'N/A')}")
64            output.append("")
65 
66    # Style recommendations
67    if "style" in results:
68        output.append("## STYLE RECOMMENDATIONS")
69        for r in results["style"]:
70            output.append(f"**{r.get('Style Name', 'N/A')}** ({r.get('Category', 'N/A')})")
71            output.append(f"- Colors: {r.get('Primary Colors', 'N/A')} | {r.get('Secondary Colors', 'N/A')}")
72            output.append(f"- Typography: {r.get('Typography', 'N/A')}")
73            output.append(f"- Effects: {r.get('Effects', 'N/A')}")
74            output.append(f"- Best For: {r.get('Best For', 'N/A')}")
75            output.append(f"- Complexity: {r.get('Complexity', 'N/A')}")
76            output.append("")
77 
78    # Color recommendations
79    if "color" in results:
80        output.append("## COLOR PALETTE OPTIONS")
81        for r in results["color"]:
82            output.append(f"**{r.get('Palette Name', 'N/A')}**")
83            output.append(f"- Primary: {r.get('Primary Hex', 'N/A')}")
84            output.append(f"- Secondary: {r.get('Secondary Hex', 'N/A')}")
85            output.append(f"- Accent: {r.get('Accent Hex', 'N/A')}")
86            output.append(f"- Background: {r.get('Background Hex', 'N/A')}")
87            output.append(f"- Psychology: {r.get('Psychology', 'N/A')}")
88            output.append("")
89 
90    output.append("=" * 60)
91    return "\n".join(output)
92 
93 
94if __name__ == "__main__":
95    parser = argparse.ArgumentParser(description="Logo Design Search")
96    parser.add_argument("query", help="Search query")
97    parser.add_argument("--domain", "-d", choices=list(CSV_CONFIG.keys()), help="Search domain")
98    parser.add_argument("--max-results", "-n", type=int, default=MAX_RESULTS, help="Max results (default: 3)")
99    parser.add_argument("--json", action="store_true", help="Output as JSON")
100    parser.add_argument("--design-brief", "-db", action="store_true", help="Generate comprehensive design brief")
101    parser.add_argument("--brand-name", "-p", type=str, default=None, help="Brand name for design brief")
102 
103    args = parser.parse_args()
104 
105    if args.design_brief:
106        result = generate_design_brief(args.query, args.brand_name)
107        print(result)
108    else:
109        result = search(args.query, args.domain, args.max_results)
110        if args.json:
111            import json
112            print(json.dumps(result, indent=2, ensure_ascii=False))
113        else:
114            print(format_output(result))
115