@oriro/orirocli 0.1.11 → 0.1.12

package/skills/craft/uipm-design/scripts/logo/generate.py

@@ -1,362 +1,362 @@
-#!/usr/bin/env python3
-# -*- coding: utf-8 -*-
-"""
-Logo Generation Script using Gemini Nano Banana API
-Uses Gemini 2.5 Flash Image and Gemini 3 Pro Image Preview models
-
-Models:
-- Nano Banana (default): gemini-2.5-flash-image - fast, high-volume, low-latency
-- Nano Banana Pro (--pro): gemini-3-pro-image-preview - professional quality, advanced reasoning
-
-Usage:
-    python generate.py --prompt "tech startup logo minimalist blue"
-    python generate.py --prompt "coffee shop vintage badge" --style vintage --output logo.png
-    python generate.py --brand "TechFlow" --industry tech --style minimalist
-    python generate.py --brand "TechFlow" --pro  # Use Nano Banana Pro model
-
-Batch mode (generates multiple variants):
-    python generate.py --brand "Unikorn" --batch 9 --output-dir ./logos --pro
-"""
-
-import argparse
-import os
-import sys
-import time
-from pathlib import Path
-from datetime import datetime
-
-# Load environment variables
-def load_env():
-    """Load .env files in priority order"""
-    env_paths = [
-        Path(__file__).parent.parent.parent / ".env",
-        Path.home() / ".claude" / "skills" / ".env",
-        Path.home() / ".claude" / ".env"
-    ]
-
-    for env_path in env_paths:
-        if env_path.exists():
-            with open(env_path) as f:
-                for line in f:
-                    line = line.strip()
-                    if line and not line.startswith('#') and '=' in line:
-                        key, value = line.split('=', 1)
-                        if key not in os.environ:
-                            os.environ[key] = value.strip('"\'')
-
-load_env()
-
-try:
-    from google import genai
-    from google.genai import types
-except ImportError:
-    print("Error: google-genai package not installed.")
-    print("Install with: pip install google-genai")
-    sys.exit(1)
-
-
-# ============ CONFIGURATION ============
-GEMINI_API_KEY = os.environ.get("GEMINI_API_KEY")
-
-# Gemini "Nano Banana" model configurations for image generation
-GEMINI_FLASH = "gemini-2.5-flash-image"  # Nano Banana: fast, high-volume, low-latency
-GEMINI_PRO = "gemini-3-pro-image-preview"  # Nano Banana Pro: professional quality, advanced reasoning
-
-# Supported aspect ratios
-ASPECT_RATIOS = ["1:1", "16:9", "9:16", "4:3", "3:4"]
-DEFAULT_ASPECT_RATIO = "1:1"  # Square is ideal for logos
-
-# Logo-specific prompt templates
-LOGO_PROMPT_TEMPLATE = """Generate a professional logo image: {prompt}
-
-Style requirements:
-- Clean vector-style illustration suitable for a logo
-- Simple, scalable design that works at any size
-- Clear silhouette and recognizable shape
-- Professional quality suitable for business use
-- Centered composition on plain white or transparent background
-- No text unless specifically requested
-- High contrast and clear edges
-- Square format, perfectly centered
-- Output as a clean, high-quality logo image
-"""
-
-STYLE_MODIFIERS = {
-    "minimalist": "minimalist, simple geometric shapes, clean lines, lots of white space, single color or limited palette",
-    "vintage": "vintage, retro, badge style, distressed texture, heritage feel, warm earth tones",
-    "modern": "modern, sleek, gradient colors, tech-forward, innovative feel",
-    "luxury": "luxury, elegant, gold accents, refined, premium feel, serif typography",
-    "playful": "playful, fun, colorful, friendly, approachable, rounded shapes",
-    "corporate": "corporate, professional, trustworthy, stable, conservative colors",
-    "organic": "organic, natural, flowing lines, earth tones, sustainable feel",
-    "geometric": "geometric, abstract, mathematical precision, symmetrical",
-    "hand-drawn": "hand-drawn, artisan, sketch-like, authentic, imperfect lines",
-    "3d": "3D, dimensional, depth, shadows, isometric perspective",
-    "abstract": "abstract mark, conceptual, symbolic, non-literal representation, artistic interpretation",
-    "lettermark": "lettermark, single letter or initials, typographic, monogram style, distinctive character",
-    "wordmark": "wordmark, logotype, custom typography, brand name as logo, distinctive lettering",
-    "emblem": "emblem, badge, crest style, enclosed design, traditional, authoritative feel",
-    "mascot": "mascot, character, friendly face, personified, memorable figure",
-    "gradient": "gradient, color transition, vibrant, modern digital feel, smooth color flow",
-    "lineart": "line art, single stroke, continuous line, elegant simplicity, wire-frame style",
-    "negative-space": "negative space, clever use of white space, hidden meaning, dual imagery, optical illusion"
-}
-
-INDUSTRY_PROMPTS = {
-    "tech": "technology company, digital, innovative, modern, circuit-like elements",
-    "healthcare": "healthcare, medical, caring, trust, cross or heart symbol",
-    "finance": "financial services, stable, trustworthy, growth, upward elements",
-    "food": "food and beverage, appetizing, warm colors, welcoming",
-    "fashion": "fashion brand, elegant, stylish, refined, artistic",
-    "fitness": "fitness and sports, dynamic, energetic, powerful, movement",
-    "eco": "eco-friendly, sustainable, natural, green, leaf or earth elements",
-    "education": "education, knowledge, growth, learning, book or cap symbol",
-    "real-estate": "real estate, property, home, roof or building silhouette",
-    "creative": "creative agency, artistic, unique, expressive, colorful"
-}
-
-
-def enhance_prompt(base_prompt, style=None, industry=None, brand_name=None):
-    """Enhance the logo prompt with style and industry modifiers"""
-    prompt_parts = [base_prompt]
-
-    if style and style in STYLE_MODIFIERS:
-        prompt_parts.append(STYLE_MODIFIERS[style])
-
-    if industry and industry in INDUSTRY_PROMPTS:
-        prompt_parts.append(INDUSTRY_PROMPTS[industry])
-
-    if brand_name:
-        prompt_parts.insert(0, f"Logo for '{brand_name}':")
-
-    combined = ", ".join(prompt_parts)
-    return LOGO_PROMPT_TEMPLATE.format(prompt=combined)
-
-
-def generate_logo(prompt, style=None, industry=None, brand_name=None,
-                  output_path=None, use_pro=False, aspect_ratio=None):
-    """Generate a logo using Gemini models with image generation
-
-    Args:
-        aspect_ratio: Image aspect ratio. Options: "1:1", "16:9", "9:16", "4:3", "3:4"
-                      Default is "1:1" (square) for logos.
-    """
-
-    if not GEMINI_API_KEY:
-        print("Error: GEMINI_API_KEY not set")
-        print("Set it with: export GEMINI_API_KEY='your-key'")
-        return None
-
-    # Initialize client
-    client = genai.Client(api_key=GEMINI_API_KEY)
-
-    # Enhance the prompt
-    full_prompt = enhance_prompt(prompt, style, industry, brand_name)
-
-    # Select model
-    model = GEMINI_PRO if use_pro else GEMINI_FLASH
-    model_label = "Nano Banana Pro (gemini-3-pro-image-preview)" if use_pro else "Nano Banana (gemini-2.5-flash-image)"
-
-    # Set aspect ratio (default to 1:1 for logos)
-    ratio = aspect_ratio if aspect_ratio in ASPECT_RATIOS else DEFAULT_ASPECT_RATIO
-
-    print(f"Generating logo with {model_label}...")
-    print(f"Aspect ratio: {ratio}")
-    print(f"Prompt: {full_prompt[:150]}...")
-    print()
-
-    try:
-        # Generate image using Gemini with image generation capability
-        response = client.models.generate_content(
-            model=model,
-            contents=full_prompt,
-            config=types.GenerateContentConfig(
-                response_modalities=["IMAGE", "TEXT"],
-                image_config=types.ImageConfig(
-                    aspect_ratio=ratio
-                ),
-                safety_settings=[
-                    types.SafetySetting(
-                        category="HARM_CATEGORY_HATE_SPEECH",
-                        threshold="BLOCK_LOW_AND_ABOVE"
-                    ),
-                    types.SafetySetting(
-                        category="HARM_CATEGORY_DANGEROUS_CONTENT",
-                        threshold="BLOCK_LOW_AND_ABOVE"
-                    ),
-                    types.SafetySetting(
-                        category="HARM_CATEGORY_SEXUALLY_EXPLICIT",
-                        threshold="BLOCK_LOW_AND_ABOVE"
-                    ),
-                    types.SafetySetting(
-                        category="HARM_CATEGORY_HARASSMENT",
-                        threshold="BLOCK_LOW_AND_ABOVE"
-                    ),
-                ]
-            )
-        )
-
-        # Extract image from response
-        image_data = None
-        for part in response.candidates[0].content.parts:
-            if hasattr(part, 'inline_data') and part.inline_data:
-                if part.inline_data.mime_type.startswith('image/'):
-                    image_data = part.inline_data.data
-                    break
-
-        if not image_data:
-            print("No image generated. The model may not have produced an image.")
-            print("Try a different prompt or check if the model supports image generation.")
-            return None
-
-        # Determine output path
-        if output_path is None:
-            timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
-            brand_slug = brand_name.lower().replace(" ", "_") if brand_name else "logo"
-            output_path = f"{brand_slug}_{timestamp}.png"
-
-        # Save image
-        with open(output_path, "wb") as f:
-            f.write(image_data)
-
-        print(f"Logo saved to: {output_path}")
-        return output_path
-
-    except Exception as e:
-        print(f"Error generating logo: {e}")
-        return None
-
-
-def generate_batch(prompt, brand_name, count, output_dir, use_pro=False, brand_context=None, aspect_ratio=None):
-    """Generate multiple logo variants with different styles"""
-
-    # Select appropriate styles for batch generation
-    batch_styles = [
-        ("minimalist", "Clean, simple geometric shape with minimal details"),
-        ("modern", "Sleek gradient with tech-forward aesthetic"),
-        ("geometric", "Abstract geometric patterns, mathematical precision"),
-        ("gradient", "Vibrant color transitions, modern digital feel"),
-        ("abstract", "Conceptual symbolic representation"),
-        ("lettermark", "Stylized letter 'U' as monogram"),
-        ("negative-space", "Clever use of negative space, hidden meaning"),
-        ("lineart", "Single stroke continuous line design"),
-        ("3d", "Dimensional design with depth and shadows"),
-    ]
-
-    # Ensure output directory exists
-    os.makedirs(output_dir, exist_ok=True)
-
-    results = []
-    model_label = "Pro" if use_pro else "Flash"
-    ratio = aspect_ratio if aspect_ratio in ASPECT_RATIOS else DEFAULT_ASPECT_RATIO
-
-    print(f"\n{'='*60}")
-    print(f"  BATCH LOGO GENERATION: {brand_name}")
-    print(f"  Model: Nano Banana {model_label}")
-    print(f"  Aspect Ratio: {ratio}")
-    print(f"  Variants: {count}")
-    print(f"  Output: {output_dir}")
-    print(f"{'='*60}\n")
-
-    for i in range(min(count, len(batch_styles))):
-        style_key, style_desc = batch_styles[i]
-
-        # Build enhanced prompt with brand context
-        enhanced_prompt = f"{prompt}, {style_desc}"
-        if brand_context:
-            enhanced_prompt = f"{brand_context}, {enhanced_prompt}"
-
-        # Generate filename
-        filename = f"{brand_name.lower().replace(' ', '_')}_{style_key}_{i+1:02d}.png"
-        output_path = os.path.join(output_dir, filename)
-
-        print(f"[{i+1}/{count}] Generating {style_key} variant...")
-
-        result = generate_logo(
-            prompt=enhanced_prompt,
-            style=style_key,
-            industry="tech",
-            brand_name=brand_name,
-            output_path=output_path,
-            use_pro=use_pro,
-            aspect_ratio=aspect_ratio
-        )
-
-        if result:
-            results.append(result)
-            print(f"  ✓ Saved: {filename}\n")
-        else:
-            print(f"  ✗ Failed: {style_key}\n")
-
-        # Rate limiting between requests
-        if i < count - 1:
-            time.sleep(2)
-
-    print(f"\n{'='*60}")
-    print(f"  BATCH COMPLETE: {len(results)}/{count} logos generated")
-    print(f"{'='*60}\n")
-
-    return results
-
-
-def main():
-    parser = argparse.ArgumentParser(description="Generate logos using Gemini Nano Banana models")
-    parser.add_argument("--prompt", "-p", type=str, help="Logo description prompt")
-    parser.add_argument("--brand", "-b", type=str, help="Brand name")
-    parser.add_argument("--style", "-s", choices=list(STYLE_MODIFIERS.keys()), help="Logo style")
-    parser.add_argument("--industry", "-i", choices=list(INDUSTRY_PROMPTS.keys()), help="Industry type")
-    parser.add_argument("--output", "-o", type=str, help="Output file path")
-    parser.add_argument("--output-dir", type=str, help="Output directory for batch generation")
-    parser.add_argument("--batch", type=int, help="Number of logo variants to generate (batch mode)")
-    parser.add_argument("--brand-context", type=str, help="Additional brand context for prompts")
-    parser.add_argument("--pro", action="store_true", help="Use Nano Banana Pro (gemini-3-pro-image-preview) for professional quality")
-    parser.add_argument("--aspect-ratio", "-r", choices=ASPECT_RATIOS, default=DEFAULT_ASPECT_RATIO,
-                        help=f"Image aspect ratio (default: {DEFAULT_ASPECT_RATIO} for logos)")
-    parser.add_argument("--list-styles", action="store_true", help="List available styles")
-    parser.add_argument("--list-industries", action="store_true", help="List available industries")
-
-    args = parser.parse_args()
-
-    if args.list_styles:
-        print("Available styles:")
-        for style, desc in STYLE_MODIFIERS.items():
-            print(f"  {style}: {desc[:60]}...")
-        return
-
-    if args.list_industries:
-        print("Available industries:")
-        for industry, desc in INDUSTRY_PROMPTS.items():
-            print(f"  {industry}: {desc[:60]}...")
-        return
-
-    if not args.prompt and not args.brand:
-        parser.error("Either --prompt or --brand is required")
-
-    prompt = args.prompt or "professional logo"
-
-    # Batch mode
-    if args.batch:
-        output_dir = args.output_dir or f"./{args.brand.lower().replace(' ', '_')}_logos"
-        generate_batch(
-            prompt=prompt,
-            brand_name=args.brand or "Logo",
-            count=args.batch,
-            output_dir=output_dir,
-            use_pro=args.pro,
-            brand_context=args.brand_context,
-            aspect_ratio=args.aspect_ratio
-        )
-    else:
-        generate_logo(
-            prompt=prompt,
-            style=args.style,
-            industry=args.industry,
-            brand_name=args.brand,
-            output_path=args.output,
-            use_pro=args.pro,
-            aspect_ratio=args.aspect_ratio
-        )
-
-
-if __name__ == "__main__":
-    main()
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""
+Logo Generation Script using Gemini Nano Banana API
+Uses Gemini 2.5 Flash Image and Gemini 3 Pro Image Preview models
+
+Models:
+- Nano Banana (default): gemini-2.5-flash-image - fast, high-volume, low-latency
+- Nano Banana Pro (--pro): gemini-3-pro-image-preview - professional quality, advanced reasoning
+
+Usage:
+    python generate.py --prompt "tech startup logo minimalist blue"
+    python generate.py --prompt "coffee shop vintage badge" --style vintage --output logo.png
+    python generate.py --brand "TechFlow" --industry tech --style minimalist
+    python generate.py --brand "TechFlow" --pro  # Use Nano Banana Pro model
+
+Batch mode (generates multiple variants):
+    python generate.py --brand "Unikorn" --batch 9 --output-dir ./logos --pro
+"""
+
+import argparse
+import os
+import sys
+import time
+from pathlib import Path
+from datetime import datetime
+
+# Load environment variables
+def load_env():
+    """Load .env files in priority order"""
+    env_paths = [
+        Path(__file__).parent.parent.parent / ".env",
+        Path.home() / ".claude" / "skills" / ".env",
+        Path.home() / ".claude" / ".env"
+    ]
+
+    for env_path in env_paths:
+        if env_path.exists():
+            with open(env_path) as f:
+                for line in f:
+                    line = line.strip()
+                    if line and not line.startswith('#') and '=' in line:
+                        key, value = line.split('=', 1)
+                        if key not in os.environ:
+                            os.environ[key] = value.strip('"\'')
+
+load_env()
+
+try:
+    from google import genai
+    from google.genai import types
+except ImportError:
+    print("Error: google-genai package not installed.")
+    print("Install with: pip install google-genai")
+    sys.exit(1)
+
+
+# ============ CONFIGURATION ============
+GEMINI_API_KEY = os.environ.get("GEMINI_API_KEY")
+
+# Gemini "Nano Banana" model configurations for image generation
+GEMINI_FLASH = "gemini-2.5-flash-image"  # Nano Banana: fast, high-volume, low-latency
+GEMINI_PRO = "gemini-3-pro-image-preview"  # Nano Banana Pro: professional quality, advanced reasoning
+
+# Supported aspect ratios
+ASPECT_RATIOS = ["1:1", "16:9", "9:16", "4:3", "3:4"]
+DEFAULT_ASPECT_RATIO = "1:1"  # Square is ideal for logos
+
+# Logo-specific prompt templates
+LOGO_PROMPT_TEMPLATE = """Generate a professional logo image: {prompt}
+
+Style requirements:
+- Clean vector-style illustration suitable for a logo
+- Simple, scalable design that works at any size
+- Clear silhouette and recognizable shape
+- Professional quality suitable for business use
+- Centered composition on plain white or transparent background
+- No text unless specifically requested
+- High contrast and clear edges
+- Square format, perfectly centered
+- Output as a clean, high-quality logo image
+"""
+
+STYLE_MODIFIERS = {
+    "minimalist": "minimalist, simple geometric shapes, clean lines, lots of white space, single color or limited palette",
+    "vintage": "vintage, retro, badge style, distressed texture, heritage feel, warm earth tones",
+    "modern": "modern, sleek, gradient colors, tech-forward, innovative feel",
+    "luxury": "luxury, elegant, gold accents, refined, premium feel, serif typography",
+    "playful": "playful, fun, colorful, friendly, approachable, rounded shapes",
+    "corporate": "corporate, professional, trustworthy, stable, conservative colors",
+    "organic": "organic, natural, flowing lines, earth tones, sustainable feel",
+    "geometric": "geometric, abstract, mathematical precision, symmetrical",
+    "hand-drawn": "hand-drawn, artisan, sketch-like, authentic, imperfect lines",
+    "3d": "3D, dimensional, depth, shadows, isometric perspective",
+    "abstract": "abstract mark, conceptual, symbolic, non-literal representation, artistic interpretation",
+    "lettermark": "lettermark, single letter or initials, typographic, monogram style, distinctive character",
+    "wordmark": "wordmark, logotype, custom typography, brand name as logo, distinctive lettering",
+    "emblem": "emblem, badge, crest style, enclosed design, traditional, authoritative feel",
+    "mascot": "mascot, character, friendly face, personified, memorable figure",
+    "gradient": "gradient, color transition, vibrant, modern digital feel, smooth color flow",
+    "lineart": "line art, single stroke, continuous line, elegant simplicity, wire-frame style",
+    "negative-space": "negative space, clever use of white space, hidden meaning, dual imagery, optical illusion"
+}
+
+INDUSTRY_PROMPTS = {
+    "tech": "technology company, digital, innovative, modern, circuit-like elements",
+    "healthcare": "healthcare, medical, caring, trust, cross or heart symbol",
+    "finance": "financial services, stable, trustworthy, growth, upward elements",
+    "food": "food and beverage, appetizing, warm colors, welcoming",
+    "fashion": "fashion brand, elegant, stylish, refined, artistic",
+    "fitness": "fitness and sports, dynamic, energetic, powerful, movement",
+    "eco": "eco-friendly, sustainable, natural, green, leaf or earth elements",
+    "education": "education, knowledge, growth, learning, book or cap symbol",
+    "real-estate": "real estate, property, home, roof or building silhouette",
+    "creative": "creative agency, artistic, unique, expressive, colorful"
+}
+
+
+def enhance_prompt(base_prompt, style=None, industry=None, brand_name=None):
+    """Enhance the logo prompt with style and industry modifiers"""
+    prompt_parts = [base_prompt]
+
+    if style and style in STYLE_MODIFIERS:
+        prompt_parts.append(STYLE_MODIFIERS[style])
+
+    if industry and industry in INDUSTRY_PROMPTS:
+        prompt_parts.append(INDUSTRY_PROMPTS[industry])
+
+    if brand_name:
+        prompt_parts.insert(0, f"Logo for '{brand_name}':")
+
+    combined = ", ".join(prompt_parts)
+    return LOGO_PROMPT_TEMPLATE.format(prompt=combined)
+
+
+def generate_logo(prompt, style=None, industry=None, brand_name=None,
+                  output_path=None, use_pro=False, aspect_ratio=None):
+    """Generate a logo using Gemini models with image generation
+
+    Args:
+        aspect_ratio: Image aspect ratio. Options: "1:1", "16:9", "9:16", "4:3", "3:4"
+                      Default is "1:1" (square) for logos.
+    """
+
+    if not GEMINI_API_KEY:
+        print("Error: GEMINI_API_KEY not set")
+        print("Set it with: export GEMINI_API_KEY='your-key'")
+        return None
+
+    # Initialize client
+    client = genai.Client(api_key=GEMINI_API_KEY)
+
+    # Enhance the prompt
+    full_prompt = enhance_prompt(prompt, style, industry, brand_name)
+
+    # Select model
+    model = GEMINI_PRO if use_pro else GEMINI_FLASH
+    model_label = "Nano Banana Pro (gemini-3-pro-image-preview)" if use_pro else "Nano Banana (gemini-2.5-flash-image)"
+
+    # Set aspect ratio (default to 1:1 for logos)
+    ratio = aspect_ratio if aspect_ratio in ASPECT_RATIOS else DEFAULT_ASPECT_RATIO
+
+    print(f"Generating logo with {model_label}...")
+    print(f"Aspect ratio: {ratio}")
+    print(f"Prompt: {full_prompt[:150]}...")
+    print()
+
+    try:
+        # Generate image using Gemini with image generation capability
+        response = client.models.generate_content(
+            model=model,
+            contents=full_prompt,
+            config=types.GenerateContentConfig(
+                response_modalities=["IMAGE", "TEXT"],
+                image_config=types.ImageConfig(
+                    aspect_ratio=ratio
+                ),
+                safety_settings=[
+                    types.SafetySetting(
+                        category="HARM_CATEGORY_HATE_SPEECH",
+                        threshold="BLOCK_LOW_AND_ABOVE"
+                    ),
+                    types.SafetySetting(
+                        category="HARM_CATEGORY_DANGEROUS_CONTENT",
+                        threshold="BLOCK_LOW_AND_ABOVE"
+                    ),
+                    types.SafetySetting(
+                        category="HARM_CATEGORY_SEXUALLY_EXPLICIT",
+                        threshold="BLOCK_LOW_AND_ABOVE"
+                    ),
+                    types.SafetySetting(
+                        category="HARM_CATEGORY_HARASSMENT",
+                        threshold="BLOCK_LOW_AND_ABOVE"
+                    ),
+                ]
+            )
+        )
+
+        # Extract image from response
+        image_data = None
+        for part in response.candidates[0].content.parts:
+            if hasattr(part, 'inline_data') and part.inline_data:
+                if part.inline_data.mime_type.startswith('image/'):
+                    image_data = part.inline_data.data
+                    break
+
+        if not image_data:
+            print("No image generated. The model may not have produced an image.")
+            print("Try a different prompt or check if the model supports image generation.")
+            return None
+
+        # Determine output path
+        if output_path is None:
+            timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+            brand_slug = brand_name.lower().replace(" ", "_") if brand_name else "logo"
+            output_path = f"{brand_slug}_{timestamp}.png"
+
+        # Save image
+        with open(output_path, "wb") as f:
+            f.write(image_data)
+
+        print(f"Logo saved to: {output_path}")
+        return output_path
+
+    except Exception as e:
+        print(f"Error generating logo: {e}")
+        return None
+
+
+def generate_batch(prompt, brand_name, count, output_dir, use_pro=False, brand_context=None, aspect_ratio=None):
+    """Generate multiple logo variants with different styles"""
+
+    # Select appropriate styles for batch generation
+    batch_styles = [
+        ("minimalist", "Clean, simple geometric shape with minimal details"),
+        ("modern", "Sleek gradient with tech-forward aesthetic"),
+        ("geometric", "Abstract geometric patterns, mathematical precision"),
+        ("gradient", "Vibrant color transitions, modern digital feel"),
+        ("abstract", "Conceptual symbolic representation"),
+        ("lettermark", "Stylized letter 'U' as monogram"),
+        ("negative-space", "Clever use of negative space, hidden meaning"),
+        ("lineart", "Single stroke continuous line design"),
+        ("3d", "Dimensional design with depth and shadows"),
+    ]
+
+    # Ensure output directory exists
+    os.makedirs(output_dir, exist_ok=True)
+
+    results = []
+    model_label = "Pro" if use_pro else "Flash"
+    ratio = aspect_ratio if aspect_ratio in ASPECT_RATIOS else DEFAULT_ASPECT_RATIO
+
+    print(f"\n{'='*60}")
+    print(f"  BATCH LOGO GENERATION: {brand_name}")
+    print(f"  Model: Nano Banana {model_label}")
+    print(f"  Aspect Ratio: {ratio}")
+    print(f"  Variants: {count}")
+    print(f"  Output: {output_dir}")
+    print(f"{'='*60}\n")
+
+    for i in range(min(count, len(batch_styles))):
+        style_key, style_desc = batch_styles[i]
+
+        # Build enhanced prompt with brand context
+        enhanced_prompt = f"{prompt}, {style_desc}"
+        if brand_context:
+            enhanced_prompt = f"{brand_context}, {enhanced_prompt}"
+
+        # Generate filename
+        filename = f"{brand_name.lower().replace(' ', '_')}_{style_key}_{i+1:02d}.png"
+        output_path = os.path.join(output_dir, filename)
+
+        print(f"[{i+1}/{count}] Generating {style_key} variant...")
+
+        result = generate_logo(
+            prompt=enhanced_prompt,
+            style=style_key,
+            industry="tech",
+            brand_name=brand_name,
+            output_path=output_path,
+            use_pro=use_pro,
+            aspect_ratio=aspect_ratio
+        )
+
+        if result:
+            results.append(result)
+            print(f"  ✓ Saved: {filename}\n")
+        else:
+            print(f"  ✗ Failed: {style_key}\n")
+
+        # Rate limiting between requests
+        if i < count - 1:
+            time.sleep(2)
+
+    print(f"\n{'='*60}")
+    print(f"  BATCH COMPLETE: {len(results)}/{count} logos generated")
+    print(f"{'='*60}\n")
+
+    return results
+
+
+def main():
+    parser = argparse.ArgumentParser(description="Generate logos using Gemini Nano Banana models")
+    parser.add_argument("--prompt", "-p", type=str, help="Logo description prompt")
+    parser.add_argument("--brand", "-b", type=str, help="Brand name")
+    parser.add_argument("--style", "-s", choices=list(STYLE_MODIFIERS.keys()), help="Logo style")
+    parser.add_argument("--industry", "-i", choices=list(INDUSTRY_PROMPTS.keys()), help="Industry type")
+    parser.add_argument("--output", "-o", type=str, help="Output file path")
+    parser.add_argument("--output-dir", type=str, help="Output directory for batch generation")
+    parser.add_argument("--batch", type=int, help="Number of logo variants to generate (batch mode)")
+    parser.add_argument("--brand-context", type=str, help="Additional brand context for prompts")
+    parser.add_argument("--pro", action="store_true", help="Use Nano Banana Pro (gemini-3-pro-image-preview) for professional quality")
+    parser.add_argument("--aspect-ratio", "-r", choices=ASPECT_RATIOS, default=DEFAULT_ASPECT_RATIO,
+                        help=f"Image aspect ratio (default: {DEFAULT_ASPECT_RATIO} for logos)")
+    parser.add_argument("--list-styles", action="store_true", help="List available styles")
+    parser.add_argument("--list-industries", action="store_true", help="List available industries")
+
+    args = parser.parse_args()
+
+    if args.list_styles:
+        print("Available styles:")
+        for style, desc in STYLE_MODIFIERS.items():
+            print(f"  {style}: {desc[:60]}...")
+        return
+
+    if args.list_industries:
+        print("Available industries:")
+        for industry, desc in INDUSTRY_PROMPTS.items():
+            print(f"  {industry}: {desc[:60]}...")
+        return
+
+    if not args.prompt and not args.brand:
+        parser.error("Either --prompt or --brand is required")
+
+    prompt = args.prompt or "professional logo"
+
+    # Batch mode
+    if args.batch:
+        output_dir = args.output_dir or f"./{args.brand.lower().replace(' ', '_')}_logos"
+        generate_batch(
+            prompt=prompt,
+            brand_name=args.brand or "Logo",
+            count=args.batch,
+            output_dir=output_dir,
+            use_pro=args.pro,
+            brand_context=args.brand_context,
+            aspect_ratio=args.aspect_ratio
+        )
+    else:
+        generate_logo(
+            prompt=prompt,
+            style=args.style,
+            industry=args.industry,
+            brand_name=args.brand,
+            output_path=args.output,
+            use_pro=args.pro,
+            aspect_ratio=args.aspect_ratio
+        )
+
+
+if __name__ == "__main__":
+    main()