キャラクター一貫性

import os
import sys
from datetime import datetime
from dotenv import load_dotenv
from google import genai
from google.genai import types
from PIL import Image
import io

def generate_image(prompt, output_filename=None, aspect_ratio="1:1"):
    """プロンプトから画像を生成する

    Args:
        prompt: 画像生成プロンプト
        output_filename: 出力ファイル名（省略時はタイムスタンプ）
        aspect_ratio: アスペクト比（"1:1", "16:9", "9:16", "3:4", "4:3"）
    """

    load_dotenv()
    client = genai.Client(api_key=os.getenv("GOOGLE_API_KEY"))

    print(f"プロンプト: {prompt[:50]}..." if len(prompt) > 50 else f"プロンプト: {prompt}")
    print("生成中...")

    response = client.models.generate_content(
        model="gemini-2.5-flash-image",
        contents=prompt,
        config=types.GenerateContentConfig(
            response_modalities=['IMAGE'],
            image_config=types.ImageConfig(aspect_ratio=aspect_ratio)
        )
    )

    for part in response.candidates[0].content.parts:
        if hasattr(part, 'inline_data'):
            image = Image.open(io.BytesIO(part.inline_data.data))
            if output_filename is None:
                output_filename = f"generated_{datetime.now().strftime('%Y%m%d_%H%M%S')}.png"
            image.save(output_filename)
            print(f"画像を保存しました: {output_filename}")
            return image
    return None

def generate_with_reference(prompt, reference_images, output_filename=None, aspect_ratio="1:1"):
    """参照画像を使用して画像を生成する

    Args:
        prompt: 画像生成プロンプト
        reference_images: ファイルパスのリスト（例: ["anchor.png", "panel1.png"]）
        output_filename: 出力ファイル名（省略時はタイムスタンプ）
        aspect_ratio: アスペクト比
    """

    load_dotenv()
    client = genai.Client(api_key=os.getenv("GOOGLE_API_KEY"))

    # 参照画像を読み込み
    images = []
    for ref in reference_images:
        if isinstance(ref, str):
            if not os.path.exists(ref):
                print(f"エラー: 参照画像 '{ref}' が見つかりません。")
                print(f"ヒント: まずアンカー画像を生成してください。例:")
                print(f"  python generate_image.py \"Character sheet of ...\" anchor.png")
                return None
            images.append(Image.open(ref))
        else:
            images.append(ref)

    print(f"プロンプト: {prompt[:50]}..." if len(prompt) > 50 else f"プロンプト: {prompt}")
    print(f"参照画像: {len(images)}枚")
    print("生成中...")

    # contentsを構築（プロンプト + 画像）
    contents = [prompt] + images

    response = client.models.generate_content(
        model="gemini-2.5-flash-image",
        contents=contents,
        config=types.GenerateContentConfig(
            response_modalities=['IMAGE'],
            image_config=types.ImageConfig(aspect_ratio=aspect_ratio)
        )
    )

    for part in response.candidates[0].content.parts:
        if hasattr(part, 'inline_data'):
            image = Image.open(io.BytesIO(part.inline_data.data))
            if output_filename is None:
                output_filename = f"generated_{datetime.now().strftime('%Y%m%d_%H%M%S')}.png"
            image.save(output_filename)
            print(f"画像を保存しました: {output_filename}")
            return image
    return None

if __name__ == "__main__":
    if len(sys.argv) < 2:
        print("使い方:")
        print("  テキストのみ: python generate_image.py \"プロンプト\" [出力ファイル名] [アスペクト比]")
        print("  参照画像あり: python generate_image.py \"プロンプト\" --ref 参照画像1.png [参照画像2.png ...] [--out 出力ファイル名] [--ratio アスペクト比]")
        print("")
        print("例:")
        print("  python generate_image.py \"A cute cat\" cat.png")
        print("  python generate_image.py \"Same character in a new scene\" --ref anchor.png panel1.png --out panel2.png --ratio 3:4")
        sys.exit(1)

    prompt = sys.argv[1]

    # --ref オプションがあるか確認
    if "--ref" in sys.argv:
        ref_index = sys.argv.index("--ref")

        # 参照画像を収集
        reference_images = []
        for i in range(ref_index + 1, len(sys.argv)):
            if sys.argv[i].startswith("--"):
                break
            reference_images.append(sys.argv[i])

        # オプション解析
        output = None
        ratio = "1:1"
        if "--out" in sys.argv:
            output = sys.argv[sys.argv.index("--out") + 1]
        if "--ratio" in sys.argv:
            ratio = sys.argv[sys.argv.index("--ratio") + 1]

        generate_with_reference(prompt, reference_images, output, ratio)
    else:
        # 従来のテキストのみ生成
        output = sys.argv[2] if len(sys.argv) > 2 else None
        ratio = sys.argv[3] if len(sys.argv) > 3 else "1:1"
        generate_image(prompt, output, ratio)

python generate_image.py "A cute cat" cat.png

# アンカー画像1枚を参照
python generate_image.py "Same character in a new scene" --ref anchor.png --out panel1.png

# 複数の参照画像を使用
python generate_image.py "Continuing the story" --ref anchor.png panel1.png --out panel2.png --ratio 3:4

python generate_image.py

使い方:
  テキストのみ: python generate_image.py "プロンプト" [出力ファイル名] [アスペクト比]
  参照画像あり: python generate_image.py "プロンプト" --ref 参照画像1.png [参照画像2.png ...] [--out 出力ファイル名] [--ratio アスペクト比]

例:
  python generate_image.py "A cute cat" cat.png
  python generate_image.py "Same character in a new scene" --ref anchor.png panel1.png --out panel2.png --ratio 3:4

Step 1: アンカーシート生成
        「A close up portrait on the left and a full body view on the right」
        → 顔詳細 + 全身デザインを1枚で
        ↓
Step 2: 参照画像として使用
        新シーン生成時に画像をアップロード
        ↓
Step 3: 明示的な参照
        「featuring the same character shown in the reference image」
        ↓
Step 4: 特徴のロックイン
        重要な特徴（目の色、髪型、特徴的なマーク）を繰り返し指定

Create a character sheet of [CHARACTER_NAME]:
4 angles (front, 3/4, side, back),
consistent facial features and outfit;
flat neutral background;
even soft lighting;
label panels [blank labels only].

Create a character sheet of Yuki, a young anime girl:
4 angles (front, 3/4, side, back view),
consistent facial features and outfit.

Character details:
- Age: 16 years old
- Long silver hair with a slight wave reaching her waist
- Large violet eyes with a gentle expression
- Fair skin with a small beauty mark under her left eye
- Wearing a dark blue school uniform with white sailor collar
- Red ribbon bow tie
- Black knee-high socks and brown loafers

Flat neutral gray background.
Even soft lighting from the front.
Clean line art, anime style.

Anime character reference sheet, front view, side view, and back view of
a female elven knight. She has long silver braided hair and sharp pointed ears,
wearing polished silver plate armor with gold filigree and a blue cape,
holding a rapier. Flat colors, white background, detailed accessories,
clean line art.

Create a professional character design sheet of a friendly robot companion
as a 3D animated character in Pixar/Disney style.

Include:
- Front view
- Side profile
- 3/4 view
- Back view
- Close-up of the face showing expression

The robot has:
- Round head with large expressive LED eyes (blue)
- White and orange body color scheme
- Small antenna on top
- Friendly, approachable design
- Slightly worn/weathered look for character

Use clean cel-shading, soft shadows.
White background.

Create a sheet of 6 chibi-style stickers of an anime girl named Miku
with different expressions and poses:

1. Happy with peace sign
2. Angry with steam coming out of ears
3. Tired with coffee mug
4. Excited with sparkles around her
5. Thinking with hand on chin
6. Laughing with tears of joy

Character features (consistent across all):
- Twin-tails with turquoise hair
- Large cyan eyes
- Wearing a futuristic black and gray outfit

Kawaii style with bold black outlines and flat colors.
White background, arranged in a 2x3 grid.

キャラクターシート（アンカー）
        ↓ 参照画像として使用
     Panel 1
        ↓ アンカー + Panel 1 を参照
     Panel 2
        ↓ アンカー + Panel 2 を参照
     Panel 3
        ...

Character sheet of Takeshi, a young Japanese martial artist:
Front view and 3/4 view side by side.

Character details:
- 20 years old male
- Spiky black hair
- Determined green eyes with a small scar on left cheek
- Athletic build
- Wearing a worn red martial arts gi with black belt

White background, anime style, clean lines.

Using the character from the reference image (maintain exact facial features:
spiky black hair, green eyes, scar on left cheek, red gi).

A high-impact manga panel, black and white ink style.
The character is shown in a medium shot, standing at the edge of a cliff.
Expression: Determined, eyes focused forward.
Wind blowing through his hair.
Dramatic cloudy sky with light rays breaking through.
Bold inking, dynamic composition.

Continuing from the previous scene.
Using the same character from Image 1 (maintain exact facial features:
spiky black hair, green eyes, scar on left cheek, red gi).

Medium shot, character now turning to face the viewer.
Expression: Surprised, eyes wide.
A glowing object appears in the distance.
Maintain the same manga ink style with bold inking.
This scene continues from Image 2.

Using the same character (maintain exact features:
- Spiky black hair
- Green eyes
- Scar on left cheek
- Red martial arts gi
- Athletic build)

A high-impact manga panel, [スタイル記述].
The character from the reference image (maintain exact [特徴リスト])
is shown in a [ショットタイプ].

Expression: [表情].
Action: [アクション].
Background: [背景].

Style: [アートスタイル].

Character sheet of Hikari, a magical girl:
Front view, side view, and action pose.

Character details:
- Long pink twin-tails hair with black ribbons
- Large amber eyes
- Petite build
- Magical girl costume: white and pink dress with star motifs
- Star-shaped wand

Cute anime style, clean lines, pastel colors.
White background.

Using the character from the reference image.
Maintain exact features:
- Long pink twin-tails hair with black ribbons
- Large amber eyes
- Petite build
- Magical girl costume: white and pink dress with star motifs
- Star-shaped wand

Manga panel, shoujo style with sparkles and soft tones.
Hikari is walking to school on a sunny morning.
Expression: Cheerful, humming a tune.
Background: Cherry blossom-lined street.

Continuing from the previous scene.
Same character from Image 1, maintain exact features:
- Long pink twin-tails hair with black ribbons
- Large amber eyes
- Petite build
- Magical girl costume

Manga panel, dramatic angle.
Hikari notices something strange in the sky.
Expression: Surprised, eyes wide.
Action: Looking up, hand shading her eyes.
Background: Ominous dark clouds forming.
Speed lines for dramatic effect.

Continuing the story.
Same character from Image 1, maintain exact features:
- Long pink twin-tails hair with black ribbons
- Large amber eyes
- Petite build
- Magical girl costume: white and pink dress with star motifs
- Star-shaped wand

Manga panel, dynamic transformation sequence.
Hikari holds up her wand, magical energy swirling.
Expression: Determined, confident.
Action: Transformation pose with dramatic lighting.
Background: Magical circles and sparkles.
Bold impact lines radiating outward.

python generate_image.py "プロンプト" --ref 4koma_layout.png --out 4koma_output.png --ratio 2:3

A 4-panel manga strip (4-koma) layout, vertical arrangement, slice of life genre.

Panel 1: A cute chibi girl with twin-tails sitting at a desk, looking bored.
Panel 2: She notices a cat outside the window, expression changes to curious.
         Eyes sparkling with interest.
Panel 3: She rushes to the window, but the cat has disappeared.
         Expression: Disappointed, shoulders slumped.
Panel 4: The cat is now sitting on her desk behind her.
         Expression: Shocked surprise, sweat drop.
         The cat looks smug.

Character consistency: Same girl in all panels with:
- Short brown twin-tails
- Large brown eyes
- Pink casual outfit
- Expressive chibi proportions

Style: Cute Japanese manga, clean lines, light screentone shading.
Panel borders clearly defined.

A 4-panel manga strip (4-koma), vertical arrangement, office comedy genre.

Panel 1: A tired businessman reaches for his coffee mug on his desk.
         Morning sunlight through office windows.
Panel 2: He tilts his mug toward himself and stares at the completely dry,
         coffee-stained bottom - not a single drop remains inside.
         Question marks floating above his head.
Panel 3: He turns to see his coworker frozen mid-sip, holding a steaming full cup.
         She looks guilty with sweat drops.
Panel 4: Wide shot revealing the coworker surrounded by 3 drained cups.
         The main character slumps in defeat.
         The coworker gives a nervous peace sign.

Character consistency:
- Main character: tired-looking man with messy hair and loose tie, NO coffee in his mug
- Coworker: cheerful woman with short hair, SHE has all the coffee

Style: Clean manga art, expressive reactions, office background.

Using the character from the reference image (silver-haired anime girl in school uniform),
create a 4-panel manga strip (4-koma), vertical arrangement.

Panel 1: The character stands at her locker, finding a pink envelope inside.
         She looks curious, holding the letter up.
Panel 2: She reads the letter - it says "I like you" with a heart.
         Her face glows with happiness, sparkles around her.
Panel 3: Close-up of her blushing face, cheeks bright red, hands on cheeks.
         Hearts floating around her head.
Panel 4: She notices the envelope says "To: Tanaka-san" (not her name).
         She slumps in despair, dark rain clouds above her head.

Maintain exact character features from the reference (silver hair, violet eyes, school uniform).
Style: Cute manga art, clean lines, expressive reactions.
Panel borders clearly defined with equal spacing.

Wearing the same outfit as in the reference:
- Dark blue school uniform
- White sailor collar
- Red ribbon bow tie

Panel borders clearly defined with thin black lines.
Clear white gutters between panels.
Each panel is a distinct scene.