better gen

src/lib/components/MonsterGenerator/MonsterGenerator.svelte

@@ -8,6 +8,7 @@
   import { saveMonster } from '$lib/db/monsters';
   import { extractPicletMetadata } from '$lib/services/picletMetadata';
   import { savePicletInstance } from '$lib/db/piclets';
+  import { PicletType, TYPE_DATA } from '$lib/types/picletTypes';
   
   interface Props extends MonsterGeneratorProps {}
   
@@ -300,30 +301,54 @@ Focus on: colors, body shape, eyes, limbs, mouth, and key visual features. Omit
     // Default tier (will be set from the generated stats)
     let tier: 'low' | 'medium' | 'high' | 'legendary' = 'medium';
     
-    // Extract monster name from the concept
-    const nameMatch = state.monsterConcept.match(/^#\s+(.+)$/m);
-    const monsterName = nameMatch ? nameMatch[1] : 'Unknown Monster';
+    // Extract object description from the Object Caption section
+    const objectMatch = state.monsterConcept.match(/# Object Caption\s*\n([\s\S]*?)(?=^# )/m);
+    const objectDescription = objectMatch ? objectMatch[1].trim() : '';
+    
+    // Extract monster name - now it's the third heading after Object Caption and Transformation Brainstorm
+    const monsterNameMatch = state.monsterConcept.match(/^# (?!Object Caption|Transformation Brainstorm)(.+)$/m);
+    const monsterName = monsterNameMatch ? monsterNameMatch[1].trim() : 'Unknown Monster';
     
     // Extract only the Monster Lore section for description
     const loreMatch = state.monsterConcept.match(/## Monster Lore\s*\n([\s\S]*?)(?=##|$)/);
     const monsterLore = loreMatch ? loreMatch[1].trim() : '';
     
+    // Debug logging
+    console.log('Extracted object description:', objectDescription);
+    console.log('Extracted monster name:', monsterName);
+    console.log('Extracted monster lore:', monsterLore);
+    
     // Create stats prompt - only ask for battle-related stats
     const statsPrompt = `Based on this monster concept, generate a JSON object with battle stats and abilities:
 "${state.monsterConcept}"
 
+The original object is described as: "${objectDescription}"
+
 First, assess the rarity of the original object based on real-world availability and value:
 • COMMON: Everyday items everyone has (stationery, grass, rocks, basic furniture, common tools)
 • UNCOMMON: Items that cost money but are widely available (electronics, appliances, vehicles, branded items)  
 • RARE: Expensive or specialized items (luxury goods, professional equipment, gold jewelry, antiques)
 • LEGENDARY: Priceless or one-of-a-kind items (crown jewels, world wonders, famous artifacts, masterpiece art)
 
+Next, determine the monster's type based on its concept and appearance. Choose the most appropriate type from these options:
+• BEAST: Vertebrate wildlife — mammals, birds, reptiles. Raw physicality, instincts, region-based variants.
+• BUG: Arthropods — butterflies, beetles, mantises. Agile swarms, precision strikes, metamorph evolutions.
+• AQUATIC: Life that swims, dives, sloshes, seeps — fish, octopus, sentient puddles. Tides, mist, pressure.
+• FLORA: Plants and fungi — blooming or decaying. Growth, spores, vines, seasonal shifts.
+• MINERAL: Stones, crystals, metals — earth's depths. Durability, reflective armor, seismic shocks.
+• SPACE: Stars, moon, cosmic objects — not of this world. Celestial energy and cosmic forces.
+• MACHINA: Engineered devices — gadgets to machinery. Gears, circuits, drones, power surges.
+• STRUCTURE: Buildings, bridges, monuments — architectural titans. Fortification, terrain shaping.
+• CULTURE: Art, fashion, toys, symbols — creative expressions. Buffs, debuffs, illusion, stories.
+• CUISINE: Dishes, drinks, culinary art — flavors and aromas. Temperature, restorative, spicy offense.
+
 The output should be formatted as a JSON instance that conforms to the JSON schema below.
 
 \`\`\`json
 {
   "properties": {
     "rarity": {"type": "string", "enum": ["common", "uncommon", "rare", "legendary"], "description": "Rarity of the original object based on real-world availability and value"},
+    "picletType": {"type": "string", "enum": ["beast", "bug", "aquatic", "flora", "mineral", "space", "machina", "structure", "culture", "cuisine"], "description": "The type that best matches this monster's concept, appearance, and nature"},
     "height": {"type": "number", "minimum": 0.1, "maximum": 50.0, "description": "Height of the piclet in meters (e.g., 1.2, 0.5, 10.0)"},
     "weight": {"type": "number", "minimum": 0.1, "maximum": 10000.0, "description": "Weight of the piclet in kilograms (e.g., 25.4, 150.0, 0.8)"},
     "HP": {"type": "integer", "minimum": 0, "maximum": 100, "description": "Health/vitality stat (0=fragile, 100=incredibly tanky)"},
@@ -340,7 +365,7 @@ The output should be formatted as a JSON instance that conforms to the JSON sche
     "specialActionName": {"type": "string", "description": "Name of the monster's ultimate move (one use per battle)"},
     "specialActionDescription": {"type": "string", "description": "Describe this powerful finishing move and its dramatic effects in battle"}
   },
-  "required": ["rarity", "height", "weight", "HP", "defence", "attack", "speed", "specialPassiveTraitDescription", "attackActionName", "attackActionDescription", "buffActionName", "buffActionDescription", "debuffActionName", "debuffActionDescription", "specialActionName", "specialActionDescription"]
+  "required": ["rarity", "picletType", "height", "weight", "HP", "defence", "attack", "speed", "specialPassiveTraitDescription", "attackActionName", "attackActionDescription", "buffActionName", "buffActionDescription", "debuffActionName", "debuffActionDescription", "specialActionName", "specialActionDescription"]
 }
 \`\`\`
 
@@ -393,7 +418,7 @@ Write your response within \`\`\`json\`\`\``;
         const parsedStats = JSON.parse(cleanJson.trim());
         
         // Remove any extra fields not in our schema
-        const allowedFields = ['rarity', 'height', 'weight', 'HP', 'defence', 'attack', 'speed',
+        const allowedFields = ['rarity', 'picletType', 'height', 'weight', 'HP', 'defence', 'attack', 'speed',
           'specialPassiveTraitDescription', 'attackActionName', 'attackActionDescription',
           'buffActionName', 'buffActionDescription', 'debuffActionName', 'debuffActionDescription',
           'specialActionName', 'specialActionDescription', 'boostActionName', 'boostActionDescription',

src/lib/components/MonsterGenerator/MonsterResult.svelte

@@ -1,6 +1,8 @@
 <script lang="ts">
   import type { MonsterWorkflowState } from '$lib/types';
   import { saveMonster } from '$lib/db/monsters';
+  import { TYPE_DATA, PicletType } from '$lib/types/picletTypes';
+  import TypeBadge from '$lib/components/UI/TypeBadge.svelte';
   
   interface Props {
     workflowState: MonsterWorkflowState;
@@ -118,6 +120,12 @@
             <div class="tier-badge {workflowState.monsterStats.tier}">{workflowState.monsterStats.tier.toUpperCase()}</div>
             <span class="stat-label">Tier</span>
           </div>
+          {#if (workflowState.monsterStats as any).picletType}
+            <div class="stat-item">
+              <TypeBadge type={(workflowState.monsterStats as any).picletType as PicletType} showLabel={true} size="medium" />
+              <span class="stat-label">Type</span>
+            </div>
+          {/if}
           <div class="stat-item">
             <div class="stat-bar" style="width: {workflowState.monsterStats.HP}%; background-color: {getStatColor(workflowState.monsterStats.HP)}20"></div>
             <span class="stat-label">HP</span>

src/lib/db/piclets.ts

@@ -26,8 +26,16 @@ export async function monsterToPicletInstance(monster: Monster, level: number =
   
   const maxHp = calculateHp(baseHp, level);
   
-  // Determine primary type based on concept and caption
-  const primaryType = getTypeFromConcept(monster.concept, monster.imageCaption);
+  // Determine primary type - use picletType from stats if available, otherwise auto-detect
+  let primaryType: PicletType;
+  if (monster.stats && (monster.stats as any).picletType) {
+    // Use the type determined during stat generation
+    const statsType = (monster.stats as any).picletType as string;
+    primaryType = statsType as PicletType;
+  } else {
+    // Fallback to concept-based detection
+    primaryType = getTypeFromConcept(monster.concept, monster.imageCaption);
+  }
   
   // Create moves based on monster's abilities
   const moves: BattleMove[] = [