1#!/usr/bin/env python3
2# -*- coding: utf-8 -*-
3"""
4Logo Generation Script using Gemini Nano Banana API
5Uses Gemini 2.5 Flash Image and Gemini 3 Pro Image Preview models
6 
7Models:
8- Nano Banana (default): gemini-2.5-flash-image - fast, high-volume, low-latency
9- Nano Banana Pro (--pro): gemini-3-pro-image-preview - professional quality, advanced reasoning
10 
11Usage:
12    python generate.py --prompt "tech startup logo minimalist blue"
13    python generate.py --prompt "coffee shop vintage badge" --style vintage --output logo.png
14    python generate.py --brand "TechFlow" --industry tech --style minimalist
15    python generate.py --brand "TechFlow" --pro  # Use Nano Banana Pro model
16 
17Batch mode (generates multiple variants):
18    python generate.py --brand "Unikorn" --batch 9 --output-dir ./logos --pro
19"""
20 
21import argparse
22import os
23import sys
24import time
25from pathlib import Path
26from datetime import datetime
27 
28# Load environment variables
29def load_env():
30    """Load .env files in priority order"""
31    env_paths = [
32        Path(__file__).parent.parent.parent / ".env",
33        Path.home() / ".claude" / "skills" / ".env",
34        Path.home() / ".claude" / ".env"
35    ]
36 
37    for env_path in env_paths:
38        if env_path.exists():
39            with open(env_path) as f:
40                for line in f:
41                    line = line.strip()
42                    if line and not line.startswith('#') and '=' in line:
43                        key, value = line.split('=', 1)
44                        if key not in os.environ:
45                            os.environ[key] = value.strip('"\'')
46 
47load_env()
48 
49try:
50    from google import genai
51    from google.genai import types
52except ImportError:
53    print("Error: google-genai package not installed.")
54    print("Install with: pip install google-genai")
55    sys.exit(1)
56 
57 
58# ============ CONFIGURATION ============
59GEMINI_API_KEY = os.environ.get("GEMINI_API_KEY")
60 
61# Gemini "Nano Banana" model configurations for image generation
62GEMINI_FLASH = "gemini-2.5-flash-image"  # Nano Banana: fast, high-volume, low-latency
63GEMINI_PRO = "gemini-3-pro-image-preview"  # Nano Banana Pro: professional quality, advanced reasoning
64 
65# Supported aspect ratios
66ASPECT_RATIOS = ["1:1", "16:9", "9:16", "4:3", "3:4"]
67DEFAULT_ASPECT_RATIO = "1:1"  # Square is ideal for logos
68 
69# Logo-specific prompt templates
70LOGO_PROMPT_TEMPLATE = """Generate a professional logo image: {prompt}
71 
72Style requirements:
73- Clean vector-style illustration suitable for a logo
74- Simple, scalable design that works at any size
75- Clear silhouette and recognizable shape
76- Professional quality suitable for business use
77- Centered composition on plain white or transparent background
78- No text unless specifically requested
79- High contrast and clear edges
80- Square format, perfectly centered
81- Output as a clean, high-quality logo image
82"""
83 
84STYLE_MODIFIERS = {
85    "minimalist": "minimalist, simple geometric shapes, clean lines, lots of white space, single color or limited palette",
86    "vintage": "vintage, retro, badge style, distressed texture, heritage feel, warm earth tones",
87    "modern": "modern, sleek, gradient colors, tech-forward, innovative feel",
88    "luxury": "luxury, elegant, gold accents, refined, premium feel, serif typography",
89    "playful": "playful, fun, colorful, friendly, approachable, rounded shapes",
90    "corporate": "corporate, professional, trustworthy, stable, conservative colors",
91    "organic": "organic, natural, flowing lines, earth tones, sustainable feel",
92    "geometric": "geometric, abstract, mathematical precision, symmetrical",
93    "hand-drawn": "hand-drawn, artisan, sketch-like, authentic, imperfect lines",
94    "3d": "3D, dimensional, depth, shadows, isometric perspective",
95    "abstract": "abstract mark, conceptual, symbolic, non-literal representation, artistic interpretation",
96    "lettermark": "lettermark, single letter or initials, typographic, monogram style, distinctive character",
97    "wordmark": "wordmark, logotype, custom typography, brand name as logo, distinctive lettering",
98    "emblem": "emblem, badge, crest style, enclosed design, traditional, authoritative feel",
99    "mascot": "mascot, character, friendly face, personified, memorable figure",
100    "gradient": "gradient, color transition, vibrant, modern digital feel, smooth color flow",
101    "lineart": "line art, single stroke, continuous line, elegant simplicity, wire-frame style",
102    "negative-space": "negative space, clever use of white space, hidden meaning, dual imagery, optical illusion"
103}
104 
105INDUSTRY_PROMPTS = {
106    "tech": "technology company, digital, innovative, modern, circuit-like elements",
107    "healthcare": "healthcare, medical, caring, trust, cross or heart symbol",
108    "finance": "financial services, stable, trustworthy, growth, upward elements",
109    "food": "food and beverage, appetizing, warm colors, welcoming",
110    "fashion": "fashion brand, elegant, stylish, refined, artistic",
111    "fitness": "fitness and sports, dynamic, energetic, powerful, movement",
112    "eco": "eco-friendly, sustainable, natural, green, leaf or earth elements",
113    "education": "education, knowledge, growth, learning, book or cap symbol",
114    "real-estate": "real estate, property, home, roof or building silhouette",
115    "creative": "creative agency, artistic, unique, expressive, colorful"
116}
117 
118 
119def enhance_prompt(base_prompt, style=None, industry=None, brand_name=None):
120    """Enhance the logo prompt with style and industry modifiers"""
121    prompt_parts = [base_prompt]
122 
123    if style and style in STYLE_MODIFIERS:
124        prompt_parts.append(STYLE_MODIFIERS[style])
125 
126    if industry and industry in INDUSTRY_PROMPTS:
127        prompt_parts.append(INDUSTRY_PROMPTS[industry])
128 
129    if brand_name:
130        prompt_parts.insert(0, f"Logo for '{brand_name}':")
131 
132    combined = ", ".join(prompt_parts)
133    return LOGO_PROMPT_TEMPLATE.format(prompt=combined)
134 
135 
136def generate_logo(prompt, style=None, industry=None, brand_name=None,
137                  output_path=None, use_pro=False, aspect_ratio=None):
138    """Generate a logo using Gemini models with image generation
139 
140    Args:
141        aspect_ratio: Image aspect ratio. Options: "1:1", "16:9", "9:16", "4:3", "3:4"
142                      Default is "1:1" (square) for logos.
143    """
144 
145    if not GEMINI_API_KEY:
146        print("Error: GEMINI_API_KEY not set")
147        print("Set it with: export GEMINI_API_KEY='your-key'")
148        return None
149 
150    # Initialize client
151    client = genai.Client(api_key=GEMINI_API_KEY)
152 
153    # Enhance the prompt
154    full_prompt = enhance_prompt(prompt, style, industry, brand_name)
155 
156    # Select model
157    model = GEMINI_PRO if use_pro else GEMINI_FLASH
158    model_label = "Nano Banana Pro (gemini-3-pro-image-preview)" if use_pro else "Nano Banana (gemini-2.5-flash-image)"
159 
160    # Set aspect ratio (default to 1:1 for logos)
161    ratio = aspect_ratio if aspect_ratio in ASPECT_RATIOS else DEFAULT_ASPECT_RATIO
162 
163    print(f"Generating logo with {model_label}...")
164    print(f"Aspect ratio: {ratio}")
165    print(f"Prompt: {full_prompt[:150]}...")
166    print()
167 
168    try:
169        # Generate image using Gemini with image generation capability
170        response = client.models.generate_content(
171            model=model,
172            contents=full_prompt,
173            config=types.GenerateContentConfig(
174                response_modalities=["IMAGE", "TEXT"],
175                image_config=types.ImageConfig(
176                    aspect_ratio=ratio
177                ),
178                safety_settings=[
179                    types.SafetySetting(
180                        category="HARM_CATEGORY_HATE_SPEECH",
181                        threshold="BLOCK_LOW_AND_ABOVE"
182                    ),
183                    types.SafetySetting(
184                        category="HARM_CATEGORY_DANGEROUS_CONTENT",
185                        threshold="BLOCK_LOW_AND_ABOVE"
186                    ),
187                    types.SafetySetting(
188                        category="HARM_CATEGORY_SEXUALLY_EXPLICIT",
189                        threshold="BLOCK_LOW_AND_ABOVE"
190                    ),
191                    types.SafetySetting(
192                        category="HARM_CATEGORY_HARASSMENT",
193                        threshold="BLOCK_LOW_AND_ABOVE"
194                    ),
195                ]
196            )
197        )
198 
199        # Extract image from response
200        image_data = None
201        for part in response.candidates[0].content.parts:
202            if hasattr(part, 'inline_data') and part.inline_data:
203                if part.inline_data.mime_type.startswith('image/'):
204                    image_data = part.inline_data.data
205                    break
206 
207        if not image_data:
208            print("No image generated. The model may not have produced an image.")
209            print("Try a different prompt or check if the model supports image generation.")
210            return None
211 
212        # Determine output path
213        if output_path is None:
214            timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
215            brand_slug = brand_name.lower().replace(" ", "_") if brand_name else "logo"
216            output_path = f"{brand_slug}_{timestamp}.png"
217 
218        # Save image
219        with open(output_path, "wb") as f:
220            f.write(image_data)
221 
222        print(f"Logo saved to: {output_path}")
223        return output_path
224 
225    except Exception as e:
226        print(f"Error generating logo: {e}")
227        return None
228 
229 
230def generate_batch(prompt, brand_name, count, output_dir, use_pro=False, brand_context=None, aspect_ratio=None):
231    """Generate multiple logo variants with different styles"""
232 
233    # Select appropriate styles for batch generation
234    batch_styles = [
235        ("minimalist", "Clean, simple geometric shape with minimal details"),
236        ("modern", "Sleek gradient with tech-forward aesthetic"),
237        ("geometric", "Abstract geometric patterns, mathematical precision"),
238        ("gradient", "Vibrant color transitions, modern digital feel"),
239        ("abstract", "Conceptual symbolic representation"),
240        ("lettermark", "Stylized letter 'U' as monogram"),
241        ("negative-space", "Clever use of negative space, hidden meaning"),
242        ("lineart", "Single stroke continuous line design"),
243        ("3d", "Dimensional design with depth and shadows"),
244    ]
245 
246    # Ensure output directory exists
247    os.makedirs(output_dir, exist_ok=True)
248 
249    results = []
250    model_label = "Pro" if use_pro else "Flash"
251    ratio = aspect_ratio if aspect_ratio in ASPECT_RATIOS else DEFAULT_ASPECT_RATIO
252 
253    print(f"\n{'='*60}")
254    print(f"  BATCH LOGO GENERATION: {brand_name}")
255    print(f"  Model: Nano Banana {model_label}")
256    print(f"  Aspect Ratio: {ratio}")
257    print(f"  Variants: {count}")
258    print(f"  Output: {output_dir}")
259    print(f"{'='*60}\n")
260 
261    for i in range(min(count, len(batch_styles))):
262        style_key, style_desc = batch_styles[i]
263 
264        # Build enhanced prompt with brand context
265        enhanced_prompt = f"{prompt}, {style_desc}"
266        if brand_context:
267            enhanced_prompt = f"{brand_context}, {enhanced_prompt}"
268 
269        # Generate filename
270        filename = f"{brand_name.lower().replace(' ', '_')}_{style_key}_{i+1:02d}.png"
271        output_path = os.path.join(output_dir, filename)
272 
273        print(f"[{i+1}/{count}] Generating {style_key} variant...")
274 
275        result = generate_logo(
276            prompt=enhanced_prompt,
277            style=style_key,
278            industry="tech",
279            brand_name=brand_name,
280            output_path=output_path,
281            use_pro=use_pro,
282            aspect_ratio=aspect_ratio
283        )
284 
285        if result:
286            results.append(result)
287            print(f"  ✓ Saved: {filename}\n")
288        else:
289            print(f"  ✗ Failed: {style_key}\n")
290 
291        # Rate limiting between requests
292        if i < count - 1:
293            time.sleep(2)
294 
295    print(f"\n{'='*60}")
296    print(f"  BATCH COMPLETE: {len(results)}/{count} logos generated")
297    print(f"{'='*60}\n")
298 
299    return results
300 
301 
302def main():
303    parser = argparse.ArgumentParser(description="Generate logos using Gemini Nano Banana models")
304    parser.add_argument("--prompt", "-p", type=str, help="Logo description prompt")
305    parser.add_argument("--brand", "-b", type=str, help="Brand name")
306    parser.add_argument("--style", "-s", choices=list(STYLE_MODIFIERS.keys()), help="Logo style")
307    parser.add_argument("--industry", "-i", choices=list(INDUSTRY_PROMPTS.keys()), help="Industry type")
308    parser.add_argument("--output", "-o", type=str, help="Output file path")
309    parser.add_argument("--output-dir", type=str, help="Output directory for batch generation")
310    parser.add_argument("--batch", type=int, help="Number of logo variants to generate (batch mode)")
311    parser.add_argument("--brand-context", type=str, help="Additional brand context for prompts")
312    parser.add_argument("--pro", action="store_true", help="Use Nano Banana Pro (gemini-3-pro-image-preview) for professional quality")
313    parser.add_argument("--aspect-ratio", "-r", choices=ASPECT_RATIOS, default=DEFAULT_ASPECT_RATIO,
314                        help=f"Image aspect ratio (default: {DEFAULT_ASPECT_RATIO} for logos)")
315    parser.add_argument("--list-styles", action="store_true", help="List available styles")
316    parser.add_argument("--list-industries", action="store_true", help="List available industries")
317 
318    args = parser.parse_args()
319 
320    if args.list_styles:
321        print("Available styles:")
322        for style, desc in STYLE_MODIFIERS.items():
323            print(f"  {style}: {desc[:60]}...")
324        return
325 
326    if args.list_industries:
327        print("Available industries:")
328        for industry, desc in INDUSTRY_PROMPTS.items():
329            print(f"  {industry}: {desc[:60]}...")
330        return
331 
332    if not args.prompt and not args.brand:
333        parser.error("Either --prompt or --brand is required")
334 
335    prompt = args.prompt or "professional logo"
336 
337    # Batch mode
338    if args.batch:
339        output_dir = args.output_dir or f"./{args.brand.lower().replace(' ', '_')}_logos"
340        generate_batch(
341            prompt=prompt,
342            brand_name=args.brand or "Logo",
343            count=args.batch,
344            output_dir=output_dir,
345            use_pro=args.pro,
346            brand_context=args.brand_context,
347            aspect_ratio=args.aspect_ratio
348        )
349    else:
350        generate_logo(
351            prompt=prompt,
352            style=args.style,
353            industry=args.industry,
354            brand_name=args.brand,
355            output_path=args.output,
356            use_pro=args.pro,
357            aspect_ratio=args.aspect_ratio
358        )
359 
360 
361if __name__ == "__main__":
362    main()
363