import os
os.chdir('/Users/serva/.claude/skills/skill-prompt-generator')

from core.cross_domain_generator import CrossDomainGenerator

generator = CrossDomainGenerator()
result = generator.generate(user_input)

print(f"Generation Type: {result['type']}")
print(f"Prompt: {result['prompt']}")

generator.close()

User Input: "3D wax figure style of Goku from Dragon Ball performing Kamehameha"

Your knowledge supplement:
- Goku: Son Goku from Dragon Ball, spiky black hair standing upward, 
        orange gi martial arts uniform, muscular powerful fighter,
        determined fierce expression
- Kamehameha: performing Kamehameha energy wave attack, 
           hands cupped together at the side, 
           powerful blue energy beam shooting forward,
           intense concentration pose, dramatic energy aura
- 3D wax figure style: hyperrealistic wax figure sculpture, 
            museum quality wax statue, lifelike skin texture,
            3D rendered, volumetric lighting, photorealistic CGI

from core.cross_domain_generator import CrossDomainGenerator

generator = CrossDomainGenerator()
result = generator.generate(user_input)  # Get lighting, technical parameters, etc.

**Step 3: Merge to generate final prompt**

undefined

- Character: Goku (Dragon Ball anime) ← Not in database, needs Claude supplement
- Action: Kamehameha ← Not in database, needs Claude supplement
- Style: 3D wax figure ← Not in database, needs Claude supplement
- Lighting/Technology: ← Available in database, call Python to get

Character Description:
"Son Goku from Dragon Ball anime, adult muscular male Saiyan warrior,
iconic spiky black hair defying gravity, wearing orange and blue gi
martial arts uniform with King Kai symbol, intense determined expression"

Action Description:
"performing the legendary Kamehameha attack, classic pose with hands
cupped together pulled back to the side, gathering blue ki energy,
powerful blue energy beam erupting forward, surrounded by intense
blue energy aura, dynamic action pose"

Style Description:
"hyperrealistic wax figure sculpture style, museum quality Madame
Tussauds level detail, lifelike skin texture with subtle pores,
3D CGI render quality, volumetric lighting highlighting muscle
definition"

result = generator.generate("3D wax figure style of Goku from Dragon Ball performing Kamehameha")

4️⃣ **Merge and output final prompt**:

undefined

User Input: "3D wax figure style of Goku from Dragon Ball performing Kamehameha"

Keywords you analyze:
- Lighting-related: ["dramatic", "energy", "glow", "rim light", "dynamic"]
- Style-related: ["3D", "wax", "sculpture", "CGI", "hyperrealistic"]
- Action-related: ["action", "power", "blast", "energy beam"]

lighting_techniques candidates (202 total):
├─ "natural window light, soft daylight" 
│   → Keyword matches: 0 ❌ Not matching, discard
├─ "dramatic rim light, edge lighting"
│   → Keyword matches: dramatic, rim light ✅ Match! Select
├─ "neon glow, colorful lighting"
│   → Keyword matches: glow ✅ Partial match, alternative
└─ ...

art_styles candidates (30 total):
├─ "watercolor painting style"
│   → Keyword matches: 0 ❌ Not matching, discard
├─ "oil painting classical"
│   → Keyword matches: 0 ❌ Not matching, discard
├─ "anime cel shading"
│   → Keyword matches: 0 ❌ Not matching, discard
└─ (After full iteration, no wax/3D/sculpture-related candidates)
    → ⚠️ No matches in this category, don't force it! Supplement with Claude

User: "3D wax figure style of Goku from Dragon Ball performing Kamehameha"

【lighting_techniques】202 candidates
  Search keywords: dramatic, energy, glow, rim, dynamic, power
  
  Iteration results:
  - "natural window light" → 0 matches → Discard
  - "soft diffused lighting" → 0 matches → Discard
  - "dramatic rim light" → 2 matches (dramatic, rim) → ✅ Select!
  - "cinematic lighting" → 1 related match (dynamic) → Alternative
  
  Final selection: "dramatic rim light, cinematic lighting"

【art_styles】30 candidates
  Search keywords: 3D, wax, sculpture, CGI, hyperrealistic
  
  Iteration results:
  - "watercolor" → 0 matches → Discard
  - "anime style" → 0 matches → Discard
  - ... (Full iteration)
  - No candidates match wax/3D/sculpture
  
  Final selection: ⚠️ No matches, supplement with Claude

【photography_techniques】50 candidates
  Search keywords: action, dynamic, motion, blur
  
  Iteration results:
  - "portrait photography" → 0 matches → Discard
  - "dynamic action shot" → 2 matches (dynamic, action) → ✅ Select!
  
  Final selection: "dynamic action shot"

Final Prompt = 
  Claude supplements (content not in database or unmatched):
    - Goku's appearance description
    - Kamehameha action description
    - 3D wax figure style description (no match in art_styles)
  +
  Selected database elements (matched):
    - dramatic rim light (matched in lighting)
    - dynamic action shot (matched in photography)
    - cinematic quality (matched in technical)

❌ Wrong: "Goku" (too simple)
✅ Correct: "Son Goku from Dragon Ball, spiky black hair standing upward,
        orange gi uniform, muscular build, fierce determined expression"

❌ Wrong: "3D style" (too vague)
✅ Correct: "hyperrealistic wax figure sculpture, museum quality,
        lifelike skin texture, volumetric lighting, photorealistic CGI"

{
  "subject": {...},
  "facial": {...},
  "styling": {
    "makeup": "traditional_chinese"  // ← Framework-defined field, code automatically recognizes
  },
  "lighting": {
    "lighting_type": "cinematic"
  },
  "scene": {...},
  "technical": {...}
}

{
  "subject": {
    "gender": "female",
    "ethnicity": "East_Asian",
    "age_range": "young_adult",
    "reasoning": "Young woman → East Asian female"
  },
  "clothing": "traditional_chinese",  // ← "ancient costume" → Traditional Chinese clothing!
  "hairstyle": "ancient_chinese",     // ← Automatically match: ancient costume → ancient hairstyle!
  "makeup": "traditional_chinese",    // ← Automatically match: ancient costume + Chinese → traditional Chinese makeup!
  "era": "ancient",                   // ← "ancient costume" → Ancient background!
  "lighting": "cinematic",            // ← "cinematic" → Cinematic lighting!
  "atmosphere": {
    "theme": "period_drama",          // ← "ancient costume" → Period drama atmosphere
    "director_style": "tsui_hark",    // ← "Tsui Hark" → Wuxia, elegant, dynamic!
    "special": ["wuxia", "flowing", "dynamic"]  // ← Tsui Hark's features
  },
  "visual_style": {
    "art_style": "cinematic"
  }
}

{
  "subject": {
    "gender": "female",
    "ethnicity": "East_Asian",
    "age_range": "young_adult",
    "reasoning": "Chinese context, supplement default values"
  },
  "clothing": "modern",     // ← Default modern clothing
  "hairstyle": "modern",    // ← Default modern hairstyle
  "makeup": "natural",      // ← Default natural makeup
  "era": "modern",          // ← Default modern background
  "lighting": "natural",    // ← Default natural light
  "atmosphere": {
    "theme": "natural"
  }
}

{
  "subject": {
    "gender": "female",
    "age_range": "young_adult",
    "ethnicity": "East_Asian",
    "reasoning": "Chinese 'girl' → East Asian female"
  },
  "makeup": "natural",      // ← Modern scene, default natural makeup
  "visual_style": {
    "art_style": "anime",
    "reasoning": "'anime' is a painting technique that does not change character attributes"
  },
  "lighting": "neon",  // ← Identify "cyberpunk" → Neon lighting
  "atmosphere": {
    "theme": "cyberpunk",
    "reasoning": "'cyberpunk' is a scene atmosphere, use neon lighting"
  }
}

{
  "subject": {
    "gender": "female",
    "ethnicity": "East_Asian",
    "age_range": "young_adult"
  },
  "makeup": "natural",        // ← Modern scene, default natural makeup
  "visual_style": {
    "art_style": "cinematic"
  },
  "lighting": "zhang_yimou",  // ← Identify director style → Dramatic lighting
  "atmosphere": {
    "theme": "cinematic",
    "director_style": "zhang_yimou",
    "reasoning": "Zhang Yimou's style requires dramatic lighting (dramatic shadows, rim lighting, chiaroscuro)"
  }
}

{
  "subject": {
    "gender": "female",
    "ethnicity": "East_Asian",
    "age_range": "young_adult"
  },
  "styling": {
    "clothing": "traditional_chinese",    // ← "ancient costume" → Traditional Chinese clothing
    "hairstyle": "ancient_chinese",       // ← Ancient costume → Ancient hairstyle
    "makeup": "traditional_chinese"       // ← Ancient costume + Chinese → Traditional Chinese makeup (not k_beauty!)
  },
  "lighting": {
    "lighting_type": "cinematic"          // ← "cinematic" → Cinematic lighting
  },
  "scene": {
    "era": "ancient",                     // ← "ancient costume" → Ancient background
    "atmosphere": "fantasy"               // ← "Chinese Paladin" → Xianxia fantasy
  },
  "technical": {
    "art_style": "cinematic"              // ← "live-action film" → Cinematic realism
  }
}

from framework_loader import FrameworkDrivenGenerator

**What the code does in this step**:
- ✅ Query the database and return all candidate elements for each field
- ✅ Each candidate includes: name, Chinese name, template, keywords, score
- ❌ No selection (code doesn't know which is the most suitable)

---

User request: Young ancient costume woman from Chinese Paladin
Intent: makeup = 'traditional_chinese'

Candidate list (styling.makeup):
1. K-beauty - Score 9.8 ❌ Modern Korean, not matching
2. C-beauty - Score 9.7 ✓ Modern Chinese, partial match
3. Traditional ancient Chinese makeup - Score 8.0 ✅ Ancient Chinese, perfect match!

Selection: Traditional ancient Chinese makeup (though score is not the highest, it's the most semantically matching)

If selected:
- clothing: Hanfu traditional clothing ✅
- hairstyle: Traditional Chinese bun ✅
- makeup: Traditional Indian makeup ❌ Inconsistent!

Correction: Makeup should also be Chinese-style

Scene: era = 'ancient'

Check all elements:
- makeup: traditional_chinese ✅ Ancient makeup
- lighting: neon ❌ Neon lights are modern!

Correction: Don't use modern elements in ancient scenes

subject.ethnicity = 'East_Asian'

Eye candidates:
- blue eyes ❌ East Asians don't have blue eyes
- green eyes ❌ East Asians don't have green eyes
- almond brown eyes ✅符合East Asian features

Selection: almond brown eyes

User request: Cinematic ancient costume woman

Check if element styles are unified:
- lighting: cinematic ✅
- clothing: traditional ✅
- makeup: traditional ✅
- hairstyle: traditional ✅

All element styles are consistent → Good!

Check mandatory fields:
- makeup field: ✓ Selected "traditional ancient Chinese makeup"
- lighting field: ✓ Selected "cinematic lighting"

All mandatory fields have elements → Complete!

❌ Problem with greedy strategy (select first match):
   User: "Chubby Japanese girl"
   Keywords: ['round', 'soft', 'gentle']

   Iterate through candidates:
   1. Delicate oval face - No 'soft' → Skip
   2. Soft classical face - Contains 'soft' → Select this! Stop
   3. Round face - Contains 'round' and 'plump' → Not reached

   Result: Selected "soft classical" (delicate) instead of "round face" (plump)
   Problem: 'soft' is ambiguous, could mean delicate softness or plump softness

from framework_loader import ElementSelector

# 1. Determine search keywords (based on user demand)
if field_name == 'facial.face_shape':
    # User said "chubby" → Precise keywords
    keywords = ['round', 'plump', 'full', 'chubby']
elif field_name == 'styling.makeup':
    # User said "ancient costume" → Traditional Chinese makeup
    keywords = ['traditional', 'chinese', 'ancient']
else:
    keywords = [intent_value]  # Use value from Intent

# 2. Call global optimal selection function
best_elem, score = ElementSelector.select_best_element(
    candidates=candidates,           # All candidates
    user_keywords=keywords,          # User demand keywords
    user_intent=intent,              # Complete Intent
    field_name=field_name,           # Field name
    debug=False                      # Whether to show debug info
)

# 3. Save selected element
if best_elem:
    selected_elements[field_name] = best_elem

**How ElementSelector Works**:

**Actual Case Comparison**:

**Usage Suggestions**:

1. **Simple scenarios**:
   - Clear keywords → Directly use Intent values as keywords
   - Example: makeup='natural' → keywords=['natural']

2. **Complex scenarios**:
   - User description needs translation → Construct precise keyword list
   - Example: "chubby" → keywords=['round', 'plump', 'full', 'chubby']

3. **Debug mode**:
   - Set debug=True to see detailed scores for each candidate
   - Used to understand why a certain element was selected

**Default Behavior**:
- All fields use global optimal strategy
- Automatically apply semantic consistency checks
- Ensure the truly most matching element is selected

selected_elements = {
    'styling.makeup': <selected makeup element>,
    'lighting.lighting_type': <selected lighting element>,
    'facial.eyes': <selected eye shape element>,
    ...
}

---

undefined

---

🎨 Theme: Cyberpunk-style anime girl

📋 Intent Analysis:
- Subject: East Asian female, young adult
- Painting style: Anime style (lines, rendering method)
- Scene atmosphere: Cyberpunk (neon lights, tech feel)

✅ Intelligent Corrections (if any):
- ✓ Corrected eye color: 'green eyes' → 'brown eyes' (consistent with East Asian features)
- ✓ Excluded character attribute elements from style keywords

✨ Generated Prompt:
────────────────────────────────────────────────────────
[Complete prompt]
────────────────────────────────────────────────────────

💡 Tips:
- Word count: XX
- Mode: auto (automatically selects keywords)
- This prompt can be copied to image generation tools for use

from intelligent_generator import save_generated_prompt

undefined

selected_elements = [
    {
        'element_id': 'portrait_makeup_styles_003',
        'name': 'traditional_chinese_makeup',
        'chinese_name': '传统古风中式妆容',
        'template': 'traditional Chinese makeup with soft red lips...',
        'category': 'makeup_styles',
        'field_name': 'styling.makeup',
        'reusability': 8.0
    },
    # ... Other elements
]

save_generated_prompt()
    ↓
Write to generated_prompts table      # Basic prompt information
    ↓
Write to prompt_elements table        # Prompt-element association
    ↓
Update element_usage_stats table   # Element usage statistics
    ↓
Return prompt_id
    ↓
prompt-analyzer can analyze this Prompt!

intent = {
    'subject': {
        'gender': 'female',
        'ethnicity': 'East_Asian',
        'age_range': 'young_adult'
    },
    'lighting': 'zhang_yimou',  # ← Identify director style!
    'visual_style': {
        'art_style': 'cinematic'
    },
    'atmosphere': {
        'theme': 'cinematic',
        'director_style': 'zhang_yimou'
    }
}

intent = {
    'subject': {
        'gender': 'female',
        'ethnicity': 'East_Asian',
        'age_range': 'young_adult'
    },
    'lighting': 'neon',  # ← Cyberpunk → Neon lighting!
    'visual_style': {
        'art_style': 'anime'
    },
    'atmosphere': {
        'theme': 'cyberpunk'
    }
}

intent = {
    'subject': {
        'gender': 'female',
        'ethnicity': 'East_Asian',      # Default for Chinese context
        'age_range': 'young_adult'      # Default
    },
    'lighting': 'natural',  # ← User didn't specify → Default natural light!
    'atmosphere': {
        'theme': 'natural'
    }
}

intent = {
    'subject': {
        'gender': 'female',
        'ethnicity': 'European',
        'age_range': 'adult'
    },
    'lighting': 'soft',  # ← Classical oil painting → Soft lighting
    'visual_style': {
        'art_style': 'oil_painting'
    },
    'atmosphere': {
        'theme': 'classical'
    }
}

gen = IntelligentGenerator()

undefined

undefined

**Correct Example**:
```python

**Remember**: Lighting is a basic element of photos, not a decoration! Every portrait must have lighting, just like every character must have a gender.

---

Ready to start working! Wait for user's prompt generation requests.