244 lines
8.1 KiB
Python
244 lines
8.1 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
GRAVITY BATCH GENERATOR - DolinaSmrti Asset Production
|
|
Uses ONLY Gravity (generate_image) - NO ComfyUI/Ufi
|
|
Style: Gritty Noir Hand-Drawn 2D Stylized Indie
|
|
"""
|
|
|
|
import os
|
|
import time
|
|
from pathlib import Path
|
|
|
|
# === CONFIGURATION ===
|
|
BASE_DIR = Path("/Users/davidkotnik/repos/novafarma")
|
|
ASSETS_DIR = BASE_DIR / "assets" / "images"
|
|
|
|
# === GRITTY NOIR STYLE ===
|
|
STYLE_PROMPT = """
|
|
(bold black outlines:1.4), dark hand-drawn stylized indie game asset,
|
|
(gritty noir aesthetic:1.2), flat colors, muted color palette,
|
|
exaggerated features, warped perspective, wonky proportions,
|
|
mature post-apocalyptic vibe, Don't Starve inspired,
|
|
high contrast noir elements, transparent background PNG
|
|
"""
|
|
|
|
NEGATIVE_PROMPT = """
|
|
pixel art, pixels, grainy, blurry, 3D rendering, realistic photo,
|
|
shading gradients, Disney style, cute kawaii, bright colors,
|
|
background elements, text, watermark
|
|
"""
|
|
|
|
# === ASSET REGISTRY ===
|
|
# Based on DODATNA_VSEBINA_REGISTRY_V1.14.md
|
|
|
|
ASSET_CATEGORIES = {
|
|
"buildings": [
|
|
"church_complete", "church_unfinished", "church_ruined",
|
|
"workshop_complete", "workshop_ruined",
|
|
"barn_complete", "barn_damaged",
|
|
"farmhouse_complete", "farmhouse_ruined",
|
|
"store_complete", "store_damaged",
|
|
"windmill", "watchtower", "greenhouse",
|
|
"laboratory", "vape_lab", "furnace_building",
|
|
"mint_building", "tailoring_shop",
|
|
],
|
|
|
|
"workstations": [
|
|
"furnace", "campfire", "tailoring_table",
|
|
"vape_lab_station", "mint_station",
|
|
"basic_sprinkler", "quality_sprinkler", "iridium_sprinkler",
|
|
"enchanting_table", "crafting_bench",
|
|
],
|
|
|
|
"items": [
|
|
# Weapons
|
|
"wooden_bow", "crystal_bow", "dragon_bow", "alpha_rainbow_bow",
|
|
"fire_arrow", "ice_arrow", "lightning_arrow", "bomb_arrow",
|
|
"poison_arrow", "holy_arrow", "silver_arrow",
|
|
|
|
# Tools
|
|
"wooden_axe", "iron_axe", "gold_axe", "diamond_axe",
|
|
"wooden_pickaxe", "iron_pickaxe", "gold_pickaxe", "diamond_pickaxe",
|
|
"wooden_hoe", "iron_hoe", "gold_hoe",
|
|
"watering_can", "fishing_rod",
|
|
|
|
# Resources
|
|
"iron_ore", "gold_ore", "silver_ore", "diamond",
|
|
"iron_bar", "gold_bar", "silver_bar",
|
|
"wood_log", "stone", "coal",
|
|
"hemp_fiber", "leather", "spider_silk",
|
|
|
|
# Food
|
|
"bread", "cheese", "milk", "beef",
|
|
"chicken_egg", "truffle", "ham",
|
|
|
|
# Magic
|
|
"slime_gel_green", "slime_gel_blue", "slime_gel_red",
|
|
"rainbow_vape_liquid", "menthol_vape_liquid",
|
|
"magic_mushroom",
|
|
],
|
|
|
|
"zivali": [ # Animals
|
|
# Livestock
|
|
"sheep_normal", "sheep_fire", "sheep_golden_fleece",
|
|
"cow_normal", "cow_mutant",
|
|
"chicken_normal", "chicken_three_headed",
|
|
"pig_normal", "pig_giant",
|
|
"horse_normal", "horse_undead",
|
|
|
|
# Wildlife
|
|
"fox", "deer", "rabbit", "hedgehog",
|
|
"bear", "wolf", "wild_boar",
|
|
"owl", "bat",
|
|
|
|
# Marine
|
|
"fish_bass", "fish_trout", "fish_salmon", "fish_tuna", "fish_piranha",
|
|
"golden_fish", "sea_dragon", "loch_ness_monster",
|
|
"shark", "jellyfish",
|
|
|
|
# Dogs
|
|
"dog_retriever", "dog_shepherd", "dog_husky", "dog_corgi", "dog_dalmatian",
|
|
"dachshund_susi",
|
|
|
|
# Slimes
|
|
"slime_green", "slime_blue", "slime_red", "slime_yellow",
|
|
"slime_purple", "slime_black", "slime_rainbow",
|
|
],
|
|
|
|
"mutanti": [
|
|
"zombie_brown_dreads", "zombie_bald", "zombie_pink_dreads",
|
|
"zombie_portrait_closeup",
|
|
"werewolf_normal", "werewolf_transformed",
|
|
"skeleton_warrior", "ghost",
|
|
"mutant_monkey", "mutant_beast",
|
|
"griffin", "pterodactyl", "hippogriff",
|
|
"bigfoot", "yeti",
|
|
],
|
|
|
|
"bosses": [
|
|
"mutant_king", "zombie_horde_leader", "ancient_tree",
|
|
"giant_troll_king", "ice_titan", "fire_dragon",
|
|
"king_slime",
|
|
],
|
|
|
|
"environment": [
|
|
# Crops
|
|
"wheat_planted", "wheat_growing", "wheat_harvest",
|
|
"corn_planted", "corn_growing", "corn_harvest",
|
|
"tomato_planted", "tomato_growing", "tomato_harvest",
|
|
|
|
# Trees
|
|
"cherry_blossom_tree", "oak_tree", "pine_tree",
|
|
"dead_tree", "burned_tree",
|
|
|
|
# Decorations
|
|
"grave_stone", "grave_cross", "grave_angel",
|
|
"rock_small", "rock_large", "boulder",
|
|
"bush_green", "bush_dead",
|
|
|
|
# Special
|
|
"carnivorous_plant_seedling", "carnivorous_plant_giant",
|
|
"portal_normal", "portal_activated",
|
|
],
|
|
|
|
"ui": [
|
|
"health_bar", "stamina_bar", "mana_bar",
|
|
"inventory_slot", "inventory_slot_selected",
|
|
"coin_icon", "heart_icon", "star_icon",
|
|
"button_play", "button_pause", "button_settings",
|
|
],
|
|
}
|
|
|
|
|
|
def generate_prompt(category: str, asset_name: str) -> str:
|
|
"""Generate full prompt for Gravity image generation"""
|
|
|
|
# Clean asset name for description
|
|
clean_name = asset_name.replace("_", " ")
|
|
|
|
# Category-specific additions
|
|
category_context = {
|
|
"buildings": "game building asset, isometric view, architectural structure",
|
|
"workstations": "crafting station, work table, game object",
|
|
"items": "game item icon, inventory object, collectible",
|
|
"zivali": "animal character, creature sprite, wildlife",
|
|
"mutanti": "mutant creature, zombie character, monster enemy",
|
|
"bosses": "boss enemy, large creature, intimidating monster",
|
|
"environment": "environment object, natural element, decoration",
|
|
"ui": "UI element, game interface icon, HUD component",
|
|
}
|
|
|
|
context = category_context.get(category, "game asset")
|
|
|
|
full_prompt = f"""
|
|
{clean_name}, {context}, {STYLE_PROMPT}
|
|
""".strip()
|
|
|
|
return full_prompt
|
|
|
|
|
|
def main():
|
|
"""Main batch generation loop"""
|
|
|
|
print("=" * 60)
|
|
print("🎨 GRAVITY BATCH GENERATOR - DolinaSmrti")
|
|
print("=" * 60)
|
|
print(f"Style: Gritty Noir Hand-Drawn 2D Stylized Indie")
|
|
print(f"Output: {ASSETS_DIR}")
|
|
print("=" * 60)
|
|
|
|
total_assets = sum(len(assets) for assets in ASSET_CATEGORIES.values())
|
|
print(f"\n📊 Total assets to generate: {total_assets}")
|
|
|
|
generated_count = 0
|
|
|
|
for category, assets in ASSET_CATEGORIES.items():
|
|
print(f"\n{'='*60}")
|
|
print(f"📁 Category: {category.upper()} ({len(assets)} assets)")
|
|
print(f"{'='*60}")
|
|
|
|
# Create category directory
|
|
category_dir = ASSETS_DIR / category
|
|
category_dir.mkdir(parents=True, exist_ok=True)
|
|
|
|
for i, asset_name in enumerate(assets, 1):
|
|
output_path = category_dir / f"{asset_name}.png"
|
|
|
|
# Skip if already exists
|
|
if output_path.exists():
|
|
print(f" ⏭️ [{i}/{len(assets)}] SKIP: {asset_name} (already exists)")
|
|
generated_count += 1
|
|
continue
|
|
|
|
# Generate prompt
|
|
prompt = generate_prompt(category, asset_name)
|
|
|
|
print(f"\n 🎨 [{i}/{len(assets)}] Generating: {asset_name}")
|
|
print(f" Prompt: {prompt[:80]}...")
|
|
|
|
# === THIS IS WHERE GRAVITY INTEGRATION HAPPENS ===
|
|
# In actual usage, Antigravity will call generate_image here
|
|
# For now, this is a placeholder that will be replaced
|
|
|
|
print(f" ⚠️ PLACEHOLDER: Call generate_image('{prompt}', '{asset_name}')")
|
|
print(f" 💾 Save to: {output_path}")
|
|
|
|
generated_count += 1
|
|
|
|
# Progress update
|
|
progress = (generated_count / total_assets) * 100
|
|
print(f"\n 📊 Overall Progress: {generated_count}/{total_assets} ({progress:.1f}%)")
|
|
|
|
# Respectful delay (avoid API hammering)
|
|
time.sleep(2)
|
|
|
|
print("\n" + "=" * 60)
|
|
print("✅ GENERATION COMPLETE!")
|
|
print("=" * 60)
|
|
print(f"Total generated: {generated_count}/{total_assets}")
|
|
print(f"Output directory: {ASSETS_DIR}")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|