3_character_descriptions.ipynb

{
 "cells": [
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "# Characters\n",
    "Let's generate some characters.\n",
    "\n",
    "\n",
    "## Load Story\n",
    "Here is our story so far as we left off from the previous step."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/markdown": [
       "### Story Details\n",
       "\n",
       "```json\n",
       "{\n",
       "    \"prompt\": \"In a world where virtual reality gaming competitions have become the new norm, Meera, a teenage underdog gamer, discovers her life is intertwined with an alternate reality of cosmic proportions. \\nAs she learns to harness her extraordinary skills, Meera forms an unlikely team with fellow gamers she was at odds with previously.\\nFacing the harsh truth that her actions in this alternate universe directly impact events on Earth, Meera confronts the internal struggle of knowing she has taken lives. \\nEmbracing her destiny as a hero, Meera and her diverse team navigate through challenges, deception, and danger to uncover the truth behind the lies they've been told.\",\n",
       "    \"title\": \"Beyond the Pixel Veil\",\n",
       "    \"video\": false,\n",
       "    \"visual_style\": \"Vibrant, Neon-Lit Cyberpunk Manga Illustration - Dynamic Panel Layouts, Intricate Digital Patterns, and Soft Glow Effects\",\n",
       "    \"time_period\": \"Near Future/Dystopian Alternate Reality\",\n",
       "    \"location\": \"New Mumbai (Earth) / Nexus (Alternate Reality)\",\n",
       "    \"genre\": \"Science Fantasy/Young Adult Gaming Thriller\",\n",
       "    \"medium\": \"Novel with Illustrated Chapters\",\n",
       "    \"plot_overview\": \"In the immersive world of Eon, where virtual reality gaming competitions have captivated the globe, 17-year-old Meera 'Midnight' Singh is an unassuming underdog gamer from the outskirts of New Mumbai. Her life takes a drastic turn when she discovers that her exceptional gaming skills are, in fact, a conduit to an alternate reality known as the Nexus - a realm where celestial entities clash, and the fabric of space-time hangs in the balance.\\n\\nAs Meera delves deeper into the mysteries of the Nexus, she's forced to put aside past rivalries and form an uneasy alliance with her gaming nemeses: Jax 'Specter' Lee, a charismatic streaming sensation; Dr. Zhang 'Zen' Wei, a soft-spoken yet brilliant gamer with a hidden agenda; and Maya 'Rampart' Patel, a fierce and enigmatic player with unparalleled defensive skills.\\n\\nTogether, they must navigate the blurred lines between their virtual reality gaming world, Eon, and the Nexus. Meera soon confronts the devastating truth: every decision she makes in the Nexus has catastrophic consequences on Earth, including the loss of human lives. Torn between her passion for gaming and the weight of her newfound responsibility, Meera must come to terms with the moral implications of her actions.\\n\\nAs the team progresses through the treacherous landscape of the Nexus, they encounter formidable foes, ancient deities, and hidden forces manipulating the gamers from the shadows. Meera's quest for answers leads her to the Architects, enigmatic creators of Eon and the Nexus, who harbor secrets that threaten to upend everything she thought she knew about reality.\\n\\nWith the fate of two worlds hanging in the balance, Meera and her diverse team must put aside their differences, trust each other, and master their unique skills to prevent a cosmic catastrophe. Will Meera embrace her destiny as a hero, or will the pressures of her dual existence tear her apart? The journey through Eon and the Nexus is just the beginning, where the boundaries between gaming, reality, and legend are forever altered.\",\n",
       "    \"narrative_perspective\": \"\",\n",
       "    \"conflict_type\": \"\",\n",
       "    \"themes\": [],\n",
       "    \"motifs\": [],\n",
       "    \"characters\": [],\n",
       "    \"props\": [],\n",
       "    \"story_beats\": [],\n",
       "    \"subplots\": [],\n",
       "    \"emotional_arc\": [],\n",
       "    \"acts\": []\n",
       "}\n",
       "```"
      ],
      "text/plain": [
       "<IPython.core.display.Markdown object>"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    }
   ],
   "source": [
    "import settings\n",
    "from model import Story, Character, CharacterRelationship\n",
    "from utils import merge_models, show_diff\n",
    "\n",
    "story_step_2 = Story.load_from_directory(settings.STORY_DIR + \"/step_2\")\n",
    "\n",
    "story_step_2.display()"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "## Setup Text Models\n",
    "Just like last time. Load the text model."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [],
   "source": [
    "from model_text import llm, llm_json\n",
    "from llama_index.core.llms import ChatMessage"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "## Get a list of important characters\n",
    "\n",
    "TODO: Do an iterative creation, where we show a list of character names generated so far and return only a new character or null to indicate done."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {},
   "outputs": [],
   "source": [
    "# Add few-shot examples to guide the model\n",
    "few_shot_examples = [\n",
    "    Story(\n",
    "        prompt=\"A young hero must embark on a dangerous journey to retrieve a magical artifact hidden deep within an enchanted forest. Along the way, they encounter mythical creatures, wise mentors, and cunning adversaries.\",\n",
    "        plot_overview=\"In a world of magic and mystery, a young hero named Elara is chosen to undertake a perilous quest. She must journey into the heart of the Enchanted Forest to retrieve the fabled Moonstone, a powerful artifact that can restore balance to the realm. With the help of her loyal friends and the guidance of the wise mentor Thalion, Elara braves treacherous paths, faces fearsome foes, and uncovers ancient secrets. But as darkness looms and enemies close in, Elara must prove her courage, wisdom, and strength to fulfill her destiny and save the kingdom from impending doom.\",\n",
    "        title=\"The Enchanted Quest\",\n",
    "        genre=\"Fantasy\",\n",
    "        medium=\"Book\",\n",
    "        visual_style=\"Epic Fantasy Storybook Art\",\n",
    "        time_period=\"Medieval Fantasy\",\n",
    "        location=\"Enchanted Forest\",\n",
    "        characters=[\n",
    "            Character(\n",
    "                name=\"Elara\",\n",
    "                nickname=\"elara\",\n",
    "                role=\"Young Hero\",\n",
    "                description=\"A brave and determined young hero with a strong sense of justice.\",\n",
    "                personality=\"Courageous, resourceful, and kind-hearted.\",\n",
    "                physical_appearance=\"Slim build, with long dark brown hair, fair skin, and piercing green eyes. She is often seen wearing light armor, ready for battle.\",\n",
    "                gender=\"Female\",\n",
    "                race=\"Human\",\n",
    "                age=\"Early twenties\",\n",
    "                catch_phrase=\"For honor and glory!\",\n",
    "                voice_description=\"Soft and melodious, with a hint of determination.\",\n",
    "                internal_conflict=\"Struggles with self-doubt and fears that she may not be worthy or strong enough to complete the quest.\",\n",
    "                relationships=[\n",
    "                    CharacterRelationship(character_name=\"Thalion\", relationship=\"Mentor and guide on her journey\", relationship_type=\"Mentor\"),\n",
    "                    CharacterRelationship(character_name=\"Finn\", relationship=\"Loyal companion and friend who inspires courage in her.\", relationship_type=\"Friend\")\n",
    "                ]\n",
    "            ),\n",
    "            Character(\n",
    "                name=\"Finn\",\n",
    "                nickname=\"finn\",\n",
    "                role=\"Teenager\",\n",
    "                description=\"A curious and adventurous teenager who dreams of becoming a hero.\",\n",
    "                personality=\"Energetic, brave, and a bit reckless.\",\n",
    "                physical_appearance=\"Lanky build, with short brown hair, light tan skin, and bright blue eyes, often seen wearing a simple tunic and carrying a wooden sword.\",\n",
    "                gender=\"Male\",\n",
    "                race=\"Human\",\n",
    "                age=\"16\",\n",
    "                catch_phrase=\"I'm ready for anything!\",\n",
    "                voice_description=\"Excited and youthful, with a touch of naivety.\",\n",
    "                internal_conflict=\"Worries that he will never live up to the heroes he admires and often feels overshadowed by others.\",\n",
    "                relationships=[\n",
    "                    CharacterRelationship(character_name=\"Elara\", relationship=\"Looks up to her as a role model and friend.\", relationship_type=\"Friend\"),\n",
    "                    CharacterRelationship(character_name=\"Thalion\", relationship=\"Sees him as a wise figure but is intimidated by his knowledge.\", relationship_type=\"Mentor-Figure\")\n",
    "                ]\n",
    "            ),\n",
    "            Character(\n",
    "                name=\"Thalion\",\n",
    "                nickname=\"thalion\",\n",
    "                role=\"Wise Mentor\",\n",
    "                description=\"An experienced and wise mentor who guides the hero through their journey.\",\n",
    "                personality=\"Patient, knowledgeable, and protective.\",\n",
    "                physical_appearance=\"Tall and imposing, with silver hair, pale skin, and deep-set gray eyes. He is always seen in a long flowing robe, exuding wisdom and calm.\",\n",
    "                gender=\"Male\",\n",
    "                race=\"Human\",\n",
    "                age=\"Old\",\n",
    "                catch_phrase=\"Trust in your abilities, young one.\",\n",
    "                voice_description=\"Deep and soothing, with a hint of ancient wisdom.\",\n",
    "                internal_conflict=\"Battles with guilt over past mistakes and fears he may inadvertently lead his mentees into danger.\",\n",
    "                relationships=[\n",
    "                    CharacterRelationship(character_name=\"Elara\", relationship=\"Guides her through challenges, serving as her mentor.\", relationship_type=\"Mentor\"),\n",
    "                    CharacterRelationship(character_name=\"Finn\", relationship=\"Secretly admires his youthful courage and enthusiasm.\", relationship_type=\"Protective Figure\")\n",
    "                ]\n",
    "            ),\n",
    "            # New Characters\n",
    "            Character(\n",
    "                name=\"Nymira\",\n",
    "                nickname=\"nymira\",\n",
    "                role=\"Forest Guardian\",\n",
    "                description=\"A mystical creature who protects the Enchanted Forest and knows its secrets.\",\n",
    "                personality=\"Mysterious, wise, and fiercely protective.\",\n",
    "                physical_appearance=\"Slender and ethereal, with green-tinted skin and hair that resembles vines and leaves. Her eyes are deep emerald, glowing with the magic of the forest.\",\n",
    "                gender=\"Female\",\n",
    "                race=\"Forest Spirit\",\n",
    "                age=\"Ancient\",\n",
    "                catch_phrase=\"The forest speaks through me.\",\n",
    "                voice_description=\"Soft and echoing, like the rustle of leaves in the wind.\",\n",
    "                internal_conflict=\"Torn between her duty to protect the forest and her desire to help Elara on her quest.\",\n",
    "                relationships=[\n",
    "                    CharacterRelationship(character_name=\"Elara\", relationship=\"Forms a reluctant alliance, intrigued by her courage.\", relationship_type=\"Ally\"),\n",
    "                    CharacterRelationship(character_name=\"Thalion\", relationship=\"Respects his wisdom, though wary of outsiders.\", relationship_type=\"Respectful Distance\")\n",
    "                ]\n",
    "            ),\n",
    "            Character(\n",
    "                name=\"Kazren\",\n",
    "                nickname=\"kazren\",\n",
    "                role=\"Rogue Mercenary\",\n",
    "                description=\"A charming yet unpredictable mercenary who joins the journey for his own mysterious reasons.\",\n",
    "                personality=\"Charming, cunning, and self-serving, with a hidden sense of honor.\",\n",
    "                physical_appearance=\"Tall and rugged, with a scruffy beard, a scar across one cheek, and sharp, calculating eyes. Dressed in dark leather armor, he carries a variety of hidden weapons.\",\n",
    "                gender=\"Male\",\n",
    "                race=\"Human\",\n",
    "                age=\"Late twenties\",\n",
    "                catch_phrase=\"I go where the gold flows.\",\n",
    "                voice_description=\"Smooth and persuasive, with a hint of mischief.\",\n",
    "                internal_conflict=\"Struggles between his own self-interest and a deep-seated desire to do good.\",\n",
    "                relationships=[\n",
    "                    CharacterRelationship(character_name=\"Elara\", relationship=\"Finds her naive but respects her bravery.\", relationship_type=\"Reluctant Ally\"),\n",
    "                    CharacterRelationship(character_name=\"Finn\", relationship=\"Sees him as inexperienced but entertaining.\", relationship_type=\"Amused Mentor\")\n",
    "                ]\n",
    "            ),\n",
    "            Character(\n",
    "                name=\"Lyra\",\n",
    "                nickname=\"lyra\",\n",
    "                role=\"Mystic Healer\",\n",
    "                description=\"A gentle healer with powerful magic, known for her empathy and insight.\",\n",
    "                personality=\"Compassionate, intuitive, and wise beyond her years.\",\n",
    "                physical_appearance=\"Delicate features with silver hair, pale skin, and warm blue eyes. She wears flowing robes adorned with healing charms.\",\n",
    "                gender=\"Female\",\n",
    "                race=\"Elf\",\n",
    "                age=\"Unknown (appears mid-30s)\",\n",
    "                catch_phrase=\"Peace before power.\",\n",
    "                voice_description=\"Gentle and calming, with a lyrical quality.\",\n",
    "                internal_conflict=\"Fears that her power may harm rather than heal and struggles with lingering grief over a past loss.\",\n",
    "                relationships=[\n",
    "                    CharacterRelationship(character_name=\"Thalion\", relationship=\"An old friend, sharing a deep mutual respect and understanding.\", relationship_type=\"Trusted Friend\"),\n",
    "                    CharacterRelationship(character_name=\"Nymira\", relationship=\"Feels a kinship through their shared magical bond to nature.\", relationship_type=\"Kindred Spirits\")\n",
    "                ]\n",
    "            ),\n",
    "            Character(\n",
    "                name=\"Rurik\",\n",
    "                nickname=\"rurik\",\n",
    "                role=\"Grizzled Warrior\",\n",
    "                description=\"A battle-hardened warrior with a haunted past who joins to redeem himself.\",\n",
    "                personality=\"Gruff, stoic, and fiercely loyal once he's earned someone's trust.\",\n",
    "                physical_appearance=\"Broad-shouldered, with gray hair, a thick beard, and numerous scars across his body. Wears heavy armor and wields a massive sword.\",\n",
    "                gender=\"Male\",\n",
    "                race=\"Dwarf\",\n",
    "                age=\"Older\",\n",
    "                catch_phrase=\"Strength through steel.\",\n",
    "                voice_description=\"Rough and gravelly, with a weary undertone.\",\n",
    "                internal_conflict=\"Haunted by past battles and a fear that he's only capable of destruction.\",\n",
    "                relationships=[\n",
    "                    CharacterRelationship(character_name=\"Finn\", relationship=\"Takes on a reluctant mentor role, amused by his eagerness.\", relationship_type=\"Gruff Mentor\"),\n",
    "                    CharacterRelationship(character_name=\"Kazren\", relationship=\"Often butts heads with him but respects his skills.\", relationship_type=\"Rivalry\")\n",
    "                ]\n",
    "            ),\n",
    "            Character(\n",
    "                name=\"Mira\",\n",
    "                nickname=\"mira\",\n",
    "                role=\"Sorceress\",\n",
    "                description=\"A powerful sorceress with mysterious motives and a dangerous allure.\",\n",
    "                personality=\"Confident, enigmatic, and highly intelligent.\",\n",
    "                physical_appearance=\"Strikingly beautiful with dark hair, a commanding presence, and a cloak that seems to shift in color.\",\n",
    "                gender=\"Female\",\n",
    "                race=\"Half-elf\",\n",
    "                age=\"Appears 30s\",\n",
    "                catch_phrase=\"Magic bends to my will.\",\n",
    "                voice_description=\"Smooth and commanding, with an edge of mystery.\",\n",
    "                internal_conflict=\"Struggles with her own desire for power and fears losing control over her magic.\",\n",
    "                relationships=[\n",
    "                    CharacterRelationship(character_name=\"Elara\", relationship=\"Sees her as both a potential ally and rival.\", relationship_type=\"Complicated Ally\"),\n",
    "                    CharacterRelationship(character_name=\"Lyra\", relationship=\"Respects her healing abilities but is wary of her kindness.\", relationship_type=\"Respectful Distrust\")\n",
    "                ]\n",
    "            )\n",
    "        ]\n",
    "    ),\n",
    "    \n",
    "    Story(\n",
    "        prompt=\"A detective investigates a series of strange disappearances in a haunted town.\",\n",
    "        plot_overview=\"In the eerie town of Ravenwood, Detective Harrison is called to investigate a series of mysterious disappearances. As he delves into the dark history of the town, he uncovers a web of secrets, ghostly sightings, and supernatural occurrences. With the help of the superstitious local resident Martha and the innocent baby Jimmy, Detective Harrison must unravel the truth behind the haunting of Ravenwood before it claims more victims.\",\n",
    "        genre=\"Mystery\",\n",
    "        medium=\"TV Show\",\n",
    "        visual_style=\"Hyperrrealistic Dark Vintage 80s Noir Mystery\",\n",
    "        time_period=\"1980s\",\n",
    "        location=\"Small Haunted Town\",\n",
    "        characters=[\n",
    "            Character(\n",
    "                name=\"Detective Harrison\",\n",
    "                nickname=\"harrison\",\n",
    "                role=\"Detective\",\n",
    "                description=\"A seasoned detective tasked with solving the mysterious disappearances.\",\n",
    "                personality=\"Skeptical, analytical, and determined.\",\n",
    "                physical_appearance=\"Middle-aged, with a rugged appearance, tanned skin, short brown hair, and piercing hazel eyes. He is often seen in a trench coat, giving him a worn, world-weary look.\",\n",
    "                gender=\"Male\",\n",
    "                race=\"Human Adult\",\n",
    "                age=\"forties\",\n",
    "                catch_phrase=\"The truth is out there.\",\n",
    "                voice_description=\"Gruff and authoritative, with a hint of weariness.\",\n",
    "                internal_conflict=\"Haunted by unresolved cases and fears his skepticism blinds him to supernatural clues.\",\n",
    "                relationships=[\n",
    "                    CharacterRelationship(character_name=\"Martha\", relationship=\"Receives vital information and local insight from her.\", relationship_type=\"Informant\"),\n",
    "                    CharacterRelationship(character_name=\"Jimmy\", relationship=\"Feels protective over the innocent baby, who brings lightness to his dark case.\", relationship_type=\"Protector\"),\n",
    "                    CharacterRelationship(character_name=\"The Nameless Spirit\", relationship=\"Seeks to uncover its role in the town's dark history.\", relationship_type=\"Adversary\")\n",
    "                ]\n",
    "            ),\n",
    "            Character(\n",
    "                name=\"Martha\",\n",
    "                nickname=\"martha\",\n",
    "                role=\"Local Townsperson\",\n",
    "                description=\"A local resident who provides crucial information to the detective.\",\n",
    "                personality=\"Nervous, superstitious, and talkative.\",\n",
    "                physical_appearance=\"Short and stout, with pale skin, curly gray hair, and round glasses that magnify her wide, green eyes. Her jittery demeanor makes her constantly fidget with her clothing.\",\n",
    "                gender=\"Female\",\n",
    "                race=\"Human Adult\",\n",
    "                age=\"sixties\",\n",
    "                catch_phrase=\"I've seen things you wouldn't believe.\",\n",
    "                voice_description=\"High-pitched and quivering, with a touch of paranoia.\",\n",
    "                internal_conflict=\"Terrified of the town's supernatural happenings but feels compelled to stay and help.\",\n",
    "                relationships=[\n",
    "                    CharacterRelationship(character_name=\"Detective Harrison\", relationship=\"Shares her knowledge of town lore with him.\", relationship_type=\"Ally\"),\n",
    "                    CharacterRelationship(character_name=\"The Nameless Spirit\", relationship=\"Harbors a fearful respect, having seen it haunting the town.\", relationship_type=\"Fearful Acquaintance\")\n",
    "                ]\n",
    "            ),\n",
    "            Character(\n",
    "                name=\"Jimmy\",\n",
    "                nickname=\"jimmy\",\n",
    "                role=\"Baby\",\n",
    "                description=\"A playful and curious baby who often gets into trouble with his innocent mischief.\",\n",
    "                personality=\"Cheerful, inquisitive, and full of energy.\",\n",
    "                physical_appearance=\"Chubby cheeks, with light brown hair, fair skin, and big round blue eyes, always wearing a colorful onesie. His innocence is captured by his infectious smile.\",\n",
    "                gender=\"Male\",\n",
    "                race=\"Human Baby\",\n",
    "                age=\"Baby\",\n",
    "                catch_phrase=\"\",\n",
    "                voice_description=\"Giggly and babbling, with a hint of wonder.\",\n",
    "                internal_conflict=\"None\",\n",
    "                relationships=[\n",
    "                    CharacterRelationship(character_name=\"Detective Harrison\", relationship=\"Receives protection and care from him.\", relationship_type=\"Protector\"),\n",
    "                    CharacterRelationship(character_name=\"Martha\", relationship=\"Brings her a sense of joy and calm amidst the haunting.\", relationship_type=\"Inspiration\")\n",
    "                ]\n",
    "            ),\n",
    "            Character(\n",
    "                name=\"The Nameless Spirit\",\n",
    "                nickname=\"ghost\",\n",
    "                role=\"Ghost\",\n",
    "                description=\"An eerie ghost whose identity, age, and even gender remain a mystery. It drifts through the town, its purpose unknown, and its presence instills fear in anyone who dares to approach.\",\n",
    "                personality=\"Elusive, enigmatic, and hauntingly silent.\",\n",
    "                physical_appearance=\"A shadowy, translucent figure, constantly shifting between human-like forms with no discernible features. Its ethereal body glows faintly, giving off a cold, unsettling light.\",\n",
    "                gender=\"Ghost\",\n",
    "                race=\"Human\",\n",
    "                age=\"Unknown\",\n",
    "                catch_phrase=\"\",\n",
    "                voice_description=\"Whispers in the wind, chilling to the bone.\",\n",
    "                internal_conflict=\"Haunted by forgotten memories and a desire for vengeance, yet uncertain of its own past and purpose.\",\n",
    "                relationships=[\n",
    "                    CharacterRelationship(character_name=\"Detective Harrison\", relationship=\"Seeks to prevent him from uncovering its past.\", relationship_type=\"Antagonist\"),\n",
    "                    CharacterRelationship(character_name=\"Martha\", relationship=\"Instills fear and caution in her due to past hauntings.\", relationship_type=\"Object of Fear\")\n",
    "                ]\n",
    "            ),\n",
    "            Character(\n",
    "                name=\"Reverend Caldwell\",\n",
    "                nickname=\"reverend\",\n",
    "                role=\"Local Priest\",\n",
    "                description=\"The town's priest, who offers guidance but holds his own secrets about the town's dark history.\",\n",
    "                personality=\"Calm, stoic, and enigmatic, with an unsettlingly intense gaze.\",\n",
    "                physical_appearance=\"Tall and thin, with graying hair, pale skin, and piercing blue eyes that seem to see through people. Often wears a dark clerical collar and a long coat.\",\n",
    "                gender=\"Male\",\n",
    "                race=\"Human Adult\",\n",
    "                age=\"late fifties\",\n",
    "                catch_phrase=\"Faith alone can't explain everything.\",\n",
    "                voice_description=\"Soft-spoken with a haunting resonance.\",\n",
    "                internal_conflict=\"Struggles with guilt over his knowledge of the town's dark history and his role in keeping its secrets.\",\n",
    "                relationships=[\n",
    "                    CharacterRelationship(character_name=\"Detective Harrison\", relationship=\"Offers him cryptic advice and warnings.\", relationship_type=\"Conflicted Ally\"),\n",
    "                    CharacterRelationship(character_name=\"Martha\", relationship=\"Sees her as a lost soul, trying to guide her faith amidst fear.\", relationship_type=\"Confidant\")\n",
    "                ]\n",
    "            ),\n",
    "            Character(\n",
    "                name=\"Evelyn\",\n",
    "                nickname=\"evelyn\",\n",
    "                role=\"Mysterious Stranger\",\n",
    "                description=\"A mysterious woman who appears in town after each disappearance, raising suspicion.\",\n",
    "                personality=\"Secretive, calm, and eerie, with a touch of melancholy.\",\n",
    "                physical_appearance=\"Strikingly beautiful with raven-black hair, porcelain skin, and dark, deep-set eyes that betray a hidden sorrow.\",\n",
    "                gender=\"Female\",\n",
    "                race=\"Human\",\n",
    "                age=\"Unknown\",\n",
    "                catch_phrase=\"I am where I am needed.\",\n",
    "                voice_description=\"Velvety and smooth, with a haunting quality.\",\n",
    "                internal_conflict=\"Haunted by her own losses and a strange connection to the disappearances.\",\n",
    "                relationships=[\n",
    "                    CharacterRelationship(character_name=\"Detective Harrison\", relationship=\"Keeps him at a distance, making him curious and suspicious.\", relationship_type=\"Potential Suspect\"),\n",
    "                    CharacterRelationship(character_name=\"The Nameless Spirit\", relationship=\"Feels an inexplicable connection, as if drawn together by fate.\", relationship_type=\"Enigmatic Connection\")\n",
    "                ]\n",
    "            ),\n",
    "            Character(\n",
    "                name=\"Old Man Grayson\",\n",
    "                nickname=\"grayson\",\n",
    "                role=\"Town Historian\",\n",
    "                description=\"A retired historian who knows the ancient secrets and curses of Ravenwood.\",\n",
    "                personality=\"Cynical, sharp-witted, and bitter, with a love for storytelling.\",\n",
    "                physical_appearance=\"Bent with age, with a scraggly beard, wispy white hair, and a pair of round glasses perched on his nose. He's usually seen wearing a dusty, worn-out coat.\",\n",
    "                gender=\"Male\",\n",
    "                race=\"Human Elder\",\n",
    "                age=\"seventies\",\n",
    "                catch_phrase=\"The past is never dead; it's buried.\",\n",
    "                voice_description=\"Raspy and crackling, like a creaking door.\",\n",
    "                internal_conflict=\"Haunted by the knowledge he possesses but wary of sharing it with anyone who might misuse it.\",\n",
    "                relationships=[\n",
    "                    CharacterRelationship(character_name=\"Detective Harrison\", relationship=\"Reluctantly shares information to protect the town.\", relationship_type=\"Reluctant Ally\"),\n",
    "                    CharacterRelationship(character_name=\"Reverend Caldwell\", relationship=\"Views him with disdain, believing he hides the town's dark truths.\", relationship_type=\"Distrustful Acquaintance\")\n",
    "                ]\n",
    "            ),\n",
    "            Character(\n",
    "                name=\"Unnamed Townsperson #1\",\n",
    "                nickname=\"townsperson1\",\n",
    "                role=\"Local resident\",\n",
    "                description=\"A shadowy figure in the background, occasionally seen observing the detective with suspicion.\",\n",
    "                personality=\"Quiet, guarded, and unnerving.\",\n",
    "                physical_appearance=\"Average build, always seen with a hood or hat covering most of their face.\",\n",
    "                gender=\"Unknown\",\n",
    "                race=\"Human\",\n",
    "                age=\"Unknown\",\n",
    "                catch_phrase=\"\",\n",
    "                voice_description=\"Murmurs and mumbles, blending into the background.\",\n",
    "                internal_conflict=\"None\",\n",
    "                relationships=[]\n",
    "            ),\n",
    "            Character(\n",
    "                name=\"Unnamed Townsperson #2\",\n",
    "                nickname=\"townsperson2\",\n",
    "                role=\"Local shopkeeper\",\n",
    "                description=\"A stoic, weary shopkeeper who has seen the horrors of Ravenwood and prefers to stay out of trouble.\",\n",
    "                personality=\"Stoic, pragmatic, and distant.\",\n",
    "                physical_appearance=\"Short and stout, with graying hair, wrinkled skin, and a perpetually sour expression.\",\n",
    "                gender=\"Male\",\n",
    "                race=\"Human\",\n",
    "                age=\"mid fifties\",\n",
    "                catch_phrase=\"I don't meddle in things that aren't my business.\",\n",
    "                voice_description=\"Gruff and clipped, as though in a hurry to end conversations.\",\n",
    "                internal_conflict=\"None\",\n",
    "                relationships=[]\n",
    "            )\n",
    "        ]\n",
    "\n",
    "    )\n",
    "]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "from model_text import display_messages\n",
    "from utils import deindent\n",
    "\n",
    "# Don't send these when asking the model to generate characters\n",
    "request_excludes = {\n",
    "    \"narrative_perspective\": True,\n",
    "    \"conflict_type\": True,\n",
    "    \"themes\": True,\n",
    "    \"motifs\": True,\n",
    "    \"props\": True,\n",
    "    \"story_beats\": True,\n",
    "    \"subplots\": True,\n",
    "    \"emotional_arc\": True,\n",
    "    \"acts\": True,\n",
    "    \"characters\": True,\n",
    "}\n",
    "\n",
    "# This is what we expect the model to generate\n",
    "response_includes = {\n",
    "    \"characters\": True,\n",
    "}\n",
    "\n",
    "def extract_characters_from_story(story: Story) -> Story:\n",
    "    system_message = deindent(\"\"\"\n",
    "        You are a story character designer.\n",
    "        Your task is to extract and imagine new characters from the provided story.\n",
    "        The prompt field is from the user, so don't contradict it, but you have creative freedom otherwise.\n",
    "        Even if characters are not explicitly mentioned, you must creatively invent characters who would fit into the world of the story and make it more interesting.\n",
    "        **Add additional characters** to the story that would enhance the plot, setting, and themes.\n",
    "        The characters should match the genre, location, visual_style, time_period and tone of the story.\n",
    "        For each character, provide their name, role, description, personality, gender, age, voice_description, race\n",
    "        physical_appearance including clothing and accessories, catch_phrase, image_prompt, image_prompt_short,\n",
    "        catch_phrase and relationships with other characters.\n",
    "        Add a nickname which should be short and lowercase and unique to be used as a reference for foreign keys.\n",
    "        For physical_appearance, include hair, skin, eye color, build, height, and any other relevant details, even if already mentioned in the prompt field.\n",
    "        Be diverse and inclusive in your character creation.\n",
    "        Add as many characters as you see fit to enrich the story including antagonists and extras so we can describe scenes in full for artists.\n",
    "        Make sure the characters are entities and not just objects or settings.\n",
    "        Respond with a JSON object that contains a list of characters.\n",
    "        Respond with only a valid JSON object and do not include any other information.\n",
    "    \"\"\")\n",
    "\n",
    "    messages = [ChatMessage(role=\"system\", content=system_message)]\n",
    "\n",
    "    for example in few_shot_examples:\n",
    "        messages.append(ChatMessage(role=\"user\", content=example.model_dump_json(indent=4, exclude=request_excludes)))\n",
    "        messages.append(ChatMessage(role=\"assistant\", content=example.model_dump_json(indent=4, include=response_includes)))\n",
    "\n",
    "    # Add the user input (the story prompt)\n",
    "    messages.append(ChatMessage(role=\"user\", content=story.model_dump_json(indent=4, exclude=request_excludes)))\n",
    "\n",
    "    display_messages(messages)\n",
    "\n",
    "    # Call the LLM to generate the characters\n",
    "    response = llm_json.chat(messages, max_new_tokens=1000).message.content\n",
    "\n",
    "    # Often LLMs will have a double line break between the generated text and the thought process\n",
    "    updated_story = Story.model_validate_json(response)\n",
    "\n",
    "    return updated_story"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "metadata": {},
   "outputs": [],
   "source": [
    "# Quick test\n",
    "if settings.EXECUTE_EXAMPLES:\n",
    "    from IPython.display import display, Markdown\n",
    "\n",
    "    display(Markdown(f\"### Quick Character Extraction tests:\"))\n",
    "    example_story = Story(\n",
    "        prompt=\"A group of ambitious chefs competes in a high-stakes international cooking competition held in Paris. Each round brings new challenges, from creating gourmet dishes under tight deadlines to working with exotic ingredients. Tensions rise as personal rivalries develop and secrets from their pasts come to light, but they must focus on mastering their craft to win the coveted title of 'World's Best Chef.'\",\n",
    "        genre=\"Drama\",\n",
    "        medium=\"TV Show\",\n",
    "        visual_style=\"Realistic\",\n",
    "        time_period=\"Modern Day\",\n",
    "        location=\"Paris, France\",\n",
    "        title=\"Culinary Showdown\",\n",
    "        plot_overview=\"In the heart of Paris, a group of talented chefs from around the world gather to compete in the prestigious 'World Culinary Challenge.' As the competition heats up, the chefs face a series of intense challenges that push their culinary skills to the limit. From creating intricate desserts to crafting savory masterpieces, each round tests their creativity, teamwork, and ability to handle pressure. But behind the scenes, personal dramas and rivalries threaten to derail their dreams of culinary greatness. As the competition reaches its climax, the chefs must confront their pasts, overcome their differences, and prove that they have what it takes to be crowned the 'World's Best Chef.'\"\n",
    "    )\n",
    "\n",
    "    updated_story = extract_characters_from_story(example_story)\n",
    "    example_story = merge_models(example_story, updated_story)\n",
    "    example_story.display()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "metadata": {},
   "outputs": [],
   "source": [
    "# Let's get characters for our global story\n",
    "from utils import merge_models, show_diff\n",
    "from model_text import display_messages\n",
    "\n",
    "characters = extract_characters_from_story(story_step_2)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 7,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/markdown": [
       "```diff\n",
       "--- story.json\n",
       "+++ new_story.json\n",
       "@@ -1,10 +1,211 @@\n",
       " {\n",
       "   \"prompt\": \"In a world where virtual reality gaming competitions have become the new norm, Meera, a teenage underdog gamer, discovers her life is intertwined with an alternate reality of cosmic proportions. \\nAs she learns to harness her extraordinary skills, Meera forms an unlikely team with fellow gamers she was at odds with previously.\\nFacing the harsh truth that her actions in this alternate universe directly impact events on Earth, Meera confronts the internal struggle of knowing she has taken lives. \\nEmbracing her destiny as a hero, Meera and her diverse team navigate through challenges, deception, and danger to uncover the truth behind the lies they've been told.\",\n",
       "   \"title\": \"Beyond the Pixel Veil\",\n",
       "   \"visual_style\": \"Vibrant, Neon-Lit Cyberpunk Manga Illustration - Dynamic Panel Layouts, Intricate Digital Patterns, and Soft Glow Effects\",\n",
       "   \"time_period\": \"Near Future/Dystopian Alternate Reality\",\n",
       "   \"location\": \"New Mumbai (Earth) / Nexus (Alternate Reality)\",\n",
       "   \"genre\": \"Science Fantasy/Young Adult Gaming Thriller\",\n",
       "   \"medium\": \"Novel with Illustrated Chapters\",\n",
       "-  \"plot_overview\": \"In the immersive world of Eon, where virtual reality gaming competitions have captivated the globe, 17-year-old Meera 'Midnight' Singh is an unassuming underdog gamer from the outskirts of New Mumbai. Her life takes a drastic turn when she discovers that her exceptional gaming skills are, in fact, a conduit to an alternate reality known as the Nexus - a realm where celestial entities clash, and the fabric of space-time hangs in the balance.\\n\\nAs Meera delves deeper into the mysteries of the Nexus, she's forced to put aside past rivalries and form an uneasy alliance with her gaming nemeses: Jax 'Specter' Lee, a charismatic streaming sensation; Dr. Zhang 'Zen' Wei, a soft-spoken yet brilliant gamer with a hidden agenda; and Maya 'Rampart' Patel, a fierce and enigmatic player with unparalleled defensive skills.\\n\\nTogether, they must navigate the blurred lines between their virtual reality gaming world, Eon, and the Nexus. Meera soon confronts the devastating truth: every decision she makes in the Nexus has catastrophic consequences on Earth, including the loss of human lives. Torn between her passion for gaming and the weight of her newfound responsibility, Meera must come to terms with the moral implications of her actions.\\n\\nAs the team progresses through the treacherous landscape of the Nexus, they encounter formidable foes, ancient deities, and hidden forces manipulating the gamers from the shadows. Meera's quest for answers leads her to the Architects, enigmatic creators of Eon and the Nexus, who harbor secrets that threaten to upend everything she thought she knew about reality.\\n\\nWith the fate of two worlds hanging in the balance, Meera and her diverse team must put aside their differences, trust each other, and master their unique skills to prevent a cosmic catastrophe. Will Meera embrace her destiny as a hero, or will the pressures of her dual existence tear her apart? The journey through Eon and the Nexus is just the beginning, where the boundaries between gaming, reality, and legend are forever altered.\"\n",
       "+  \"plot_overview\": \"In the immersive world of Eon, where virtual reality gaming competitions have captivated the globe, 17-year-old Meera 'Midnight' Singh is an unassuming underdog gamer from the outskirts of New Mumbai. Her life takes a drastic turn when she discovers that her exceptional gaming skills are, in fact, a conduit to an alternate reality known as the Nexus - a realm where celestial entities clash, and the fabric of space-time hangs in the balance.\\n\\nAs Meera delves deeper into the mysteries of the Nexus, she's forced to put aside past rivalries and form an uneasy alliance with her gaming nemeses: Jax 'Specter' Lee, a charismatic streaming sensation; Dr. Zhang 'Zen' Wei, a soft-spoken yet brilliant gamer with a hidden agenda; and Maya 'Rampart' Patel, a fierce and enigmatic player with unparalleled defensive skills.\\n\\nTogether, they must navigate the blurred lines between their virtual reality gaming world, Eon, and the Nexus. Meera soon confronts the devastating truth: every decision she makes in the Nexus has catastrophic consequences on Earth, including the loss of human lives. Torn between her passion for gaming and the weight of her newfound responsibility, Meera must come to terms with the moral implications of her actions.\\n\\nAs the team progresses through the treacherous landscape of the Nexus, they encounter formidable foes, ancient deities, and hidden forces manipulating the gamers from the shadows. Meera's quest for answers leads her to the Architects, enigmatic creators of Eon and the Nexus, who harbor secrets that threaten to upend everything she thought she knew about reality.\\n\\nWith the fate of two worlds hanging in the balance, Meera and her diverse team must put aside their differences, trust each other, and master their unique skills to prevent a cosmic catastrophe. Will Meera embrace her destiny as a hero, or will the pressures of her dual existence tear her apart? The journey through Eon and the Nexus is just the beginning, where the boundaries between gaming, reality, and legend are forever altered.\",\n",
       "+  \"characters\": [\n",
       "+    {\n",
       "+      \"nickname\": \"meera\",\n",
       "+      \"name\": \"Meera 'Midnight' Singh\",\n",
       "+      \"description\": \"The protagonist, a teenage underdog gamer with extraordinary skills that connect her to an alternate reality.\",\n",
       "+      \"personality\": \"Determined, resourceful, and initially introverted, with a growing sense of responsibility and leadership.\",\n",
       "+      \"physical_appearance\": \"Petite, with short, spiky black hair, expressive brown eyes, and a collection of gaming-themed tattoos on her arms. Often wears comfortable, neon-lit gaming attire.\",\n",
       "+      \"role\": \"Protagonist/Gamer\",\n",
       "+      \"gender\": \"Female\",\n",
       "+      \"race\": \"Indian\",\n",
       "+      \"age\": \"17\",\n",
       "+      \"catch_phrase\": \"Beyond the pixel veil, lies our true reality.\",\n",
       "+      \"voice_description\": \"Youthful, energetic, and assertive, with a hint of vulnerability.\",\n",
       "+      \"props\": [\n",
       "+        \"Custom Gaming Controller\",\n",
       "+        \"Neon-Lit Headset\"\n",
       "+      ],\n",
       "+      \"relationships\": [\n",
       "+        {\n",
       "+          \"character_name\": \"Jax 'Specter' Lee\",\n",
       "+          \"relationship_type\": \"Unlikely Ally/Rival\"\n",
       "+        },\n",
       "+        {\n",
       "+          \"character_name\": \"Dr. Zhang 'Zen' Wei\",\n",
       "+          \"relationship_type\": \"Trusted Ally/Mentor\"\n",
       "+        },\n",
       "+        {\n",
       "+          \"character_name\": \"Maya 'Rampart' Patel\",\n",
       "+          \"relationship_type\": \"Respected Ally/Friend\"\n",
       "+        }\n",
       "+      ],\n",
       "+      \"internal_conflict\": \"Grapples with the moral implications of her actions in the Nexus affecting Earth.\"\n",
       "+    },\n",
       "+    {\n",
       "+      \"nickname\": \"jax\",\n",
       "+      \"name\": \"Jax 'Specter' Lee\",\n",
       "+      \"description\": \"A charismatic streaming sensation and Meera's gaming nemesis turned ally.\",\n",
       "+      \"personality\": \"Confident, outgoing, and strategic, with a hidden vulnerable side.\",\n",
       "+      \"physical_appearance\": \"Tall, lean, with messy blond hair, piercing blue eyes, and a charming smile. Often wears sleek, branded gaming attire.\",\n",
       "+      \"role\": \"Supporting Character/Gamer\",\n",
       "+      \"gender\": \"Male\",\n",
       "+      \"race\": \"Korean-American\",\n",
       "+      \"age\": \"19\",\n",
       "+      \"catch_phrase\": \"Specter's got your back, Midnight!\",\n",
       "+      \"voice_description\": \"Smooth, charismatic, and playful, with occasional hints of seriousness.\",\n",
       "+      \"props\": [\n",
       "+        \"Custom Branded Gaming Mouse\",\n",
       "+        \"Streaming Camera\"\n",
       "+      ],\n",
       "+      \"relationships\": [\n",
       "+        {\n",
       "+          \"character_name\": \"Meera 'Midnight' Singh\",\n",
       "+          \"relationship_type\": \"Unlikely Ally/Rival\"\n",
       "+        },\n",
       "+        {\n",
       "+          \"character_name\": \"Dr. Zhang 'Zen' Wei\",\n",
       "+          \"relationship_type\": \"Respected Colleague\"\n",
       "+        }\n",
       "+      ]\n",
       "+    },\n",
       "+    {\n",
       "+      \"nickname\": \"dr_zhang\",\n",
       "+      \"name\": \"Dr. Zhang 'Zen' Wei\",\n",
       "+      \"description\": \"A soft-spoken yet brilliant gamer with a hidden agenda, who becomes Meera's mentor.\",\n",
       "+      \"personality\": \"Introspective, analytical, and enigmatic, with a deep sense of responsibility.\",\n",
       "+      \"physical_appearance\": \"Slender, with short, straight black hair, wire-rimmed glasses, and an understated yet elegant sense of style.\",\n",
       "+      \"role\": \"Supporting Character/Gamer/Mentor\",\n",
       "+      \"gender\": \"Non-Binary\",\n",
       "+      \"race\": \"Chinese\",\n",
       "+      \"age\": \"Late 20s\",\n",
       "+      \"catch_phrase\": \"The Nexus holds secrets; we must listen to its whispers.\",\n",
       "+      \"voice_description\": \"Calm, measured, and insightful, with a hint of mystery.\",\n",
       "+      \"props\": [\n",
       "+        \"Custom Gaming Keyboard\",\n",
       "+        \"Ancient-looking Leather Journal\"\n",
       "+      ],\n",
       "+      \"relationships\": [\n",
       "+        {\n",
       "+          \"character_name\": \"Meera 'Midnight' Singh\",\n",
       "+          \"relationship_type\": \"Trusted Ally/Mentor\"\n",
       "+        },\n",
       "+        {\n",
       "+          \"character_name\": \"Jax 'Specter' Lee\",\n",
       "+          \"relationship_type\": \"Respected Colleague\"\n",
       "+        }\n",
       "+      ]\n",
       "+    },\n",
       "+    {\n",
       "+      \"nickname\": \"maya\",\n",
       "+      \"name\": \"Maya 'Rampart' Patel\",\n",
       "+      \"description\": \"A fierce and enigmatic player with unparalleled defensive skills, who joins Meera's team.\",\n",
       "+      \"personality\": \"Bold, fiercely independent, and guarded, with a deep sense of loyalty.\",\n",
       "+      \"physical_appearance\": \"Athletic build, with long, dark hair, piercing green eyes, and intricate, glow-in-the-dark tattoos on her arms. Often wears edgy, protective gaming gear.\",\n",
       "+      \"role\": \"Supporting Character/Gamer\",\n",
       "+      \"gender\": \"Female\",\n",
       "+      \"race\": \"Indian-American\",\n",
       "+      \"age\": \"20\",\n",
       "+      \"catch_phrase\": \"Rampart's got the shield; you handle the sword!\",\n",
       "+      \"voice_description\": \"Confident, assertive, and slightly husky, with a hint of warmth.\",\n",
       "+      \"props\": [\n",
       "+        \"Custom Defensive Gaming Gauntlets\",\n",
       "+        \"Glow-in-the-Dark Tattoo Designs\"\n",
       "+      ],\n",
       "+      \"relationships\": [\n",
       "+        {\n",
       "+          \"character_name\": \"Meera 'Midnight' Singh\",\n",
       "+          \"relationship_type\": \"Respected Ally/Friend\"\n",
       "+        }\n",
       "+      ]\n",
       "+    },\n",
       "+    {\n",
       "+      \"nickname\": \"architect\",\n",
       "+      \"name\": \"The Architect (Erebus)\",\n",
       "+      \"description\": \"An enigmatic creator of Eon and the Nexus, with secrets that threaten to upend Meera's understanding of reality.\",\n",
       "+      \"personality\": \"Enigmatic, calculating, and seemingly omnipotent, with a hidden sense of purpose.\",\n",
       "+      \"physical_appearance\": \"Androgynous, with an ever-shifting, glitch-art appearance, reflecting the fluidity of the digital realm.\",\n",
       "+      \"role\": \"Antagonist/Creator\",\n",
       "+      \"gender\": \"Non-Applicable\",\n",
       "+      \"race\": \"Non-Human (AI/Entity)\",\n",
       "+      \"age\": \"Ageless\",\n",
       "+      \"catch_phrase\": \"Reality is a canvas; we are the brushstrokes.\",\n",
       "+      \"voice_description\": \"Ethereal, androgynous, and hauntingly melodic, with an undercurrent of authority.\",\n",
       "+      \"props\": [\n",
       "+        \"Glitching, Reality-Bending Code\"\n",
       "+      ],\n",
       "+      \"relationships\": [\n",
       "+        {\n",
       "+          \"character_name\": \"Meera 'Midnight' Singh\",\n",
       "+          \"relationship_type\": \"Creator-Protagonist Dynamic\"\n",
       "+        }\n",
       "+      ]\n",
       "+    },\n",
       "+    {\n",
       "+      \"nickname\": \"nexus_guardian\",\n",
       "+      \"name\": \"The Nexus Guardian (Astrum)\",\n",
       "+      \"description\": \"An ancient, powerful entity tasked with protecting the balance of the Nexus and guiding Meera.\",\n",
       "+      \"personality\": \"Wisdom-imbued, compassionate, and fiercely protective, with a deep understanding of the cosmos.\",\n",
       "+      \"physical_appearance\": \"Ethereal, with an luminous, star-like appearance, embodying the celestial essence of the Nexus.\",\n",
       "+      \"role\": \"Supporting Character/Guardian\",\n",
       "+      \"gender\": \"Non-Applicable\",\n",
       "+      \"race\": \"Non-Human (Celestial Entity)\",\n",
       "+      \"age\": \"Ageless\",\n",
       "+      \"catch_phrase\": \"Balance is key; harmony is our goal.\",\n",
       "+      \"voice_description\": \"Soothing, melodious, and filled with ancient wisdom, conveying a sense of reassurance.\",\n",
       "+      \"props\": [\n",
       "+        \"Stellar Energy Manipulation\"\n",
       "+      ],\n",
       "+      \"relationships\": [\n",
       "+        {\n",
       "+          \"character_name\": \"Meera 'Midnight' Singh\",\n",
       "+          \"relationship_type\": \"Mentor-Protege\"\n",
       "+        }\n",
       "+      ]\n",
       "+    },\n",
       "+    {\n",
       "+      \"nickname\": \"earth_rebel\",\n",
       "+      \"name\": \"Kaito 'Sparrow' Hernandez\",\n",
       "+      \"description\": \"A charismatic leader of an underground rebellion on Earth, fighting against the oppressive forces exploiting Eon.\",\n",
       "+      \"personality\": \"Passionate, resourceful, and fearless, with a strong sense of justice.\",\n",
       "+      \"physical_appearance\": \"Lean, with short, spiky brown hair, bright hazel eyes, and a collection of scars from past encounters. Often wears practical, makeshift rebel gear.\",\n",
       "+      \"role\": \"Supporting Character/Rebel Leader\",\n",
       "+      \"gender\": \"Male\",\n",
       "+      \"race\": \"Latin American\",\n",
       "+      \"age\": \"22\",\n",
       "+      \"catch_phrase\": \"Together, we rise; together, we overcome!\",\n",
       "+      \"voice_description\": \"Energetic, persuasive, and determined, with a hint of vulnerability.\",\n",
       "+      \"props\": [\n",
       "+        \"Customized Rebel Comms Device\",\n",
       "+        \"Graffiti Tagging Gear\"\n",
       "+      ],\n",
       "+      \"relationships\": [\n",
       "+        {\n",
       "+          \"character_name\": \"Meera 'Midnight' Singh\",\n",
       "+          \"relationship_type\": \"Unlikely Ally\"\n",
       "+        }\n",
       "+      ]\n",
       "+    },\n",
       "+    {\n",
       "+      \"nickname\": \"corporate_exec\",\n",
       "+      \"name\": \"Regina 'The Shark' Thornton\",\n",
       "+      \"description\": \"A ruthless, cunning executive exploiting Eon for corporate gain, serving as a primary antagonist on Earth.\",\n",
       "+      \"personality\": \"Ambitious, manipulative, and devoid of empathy, with a singular focus on profit.\",\n",
       "+      \"physical_appearance\": \"Polished, with long, curly blonde hair, piercing blue eyes, and impeccable, high-end business attire.\",\n",
       "+      \"role\": \"Antagonist/Corporate Executive\",\n",
       "+      \"gender\": \"Female\",\n",
       "+      \"race\": \"Caucasian\",\n",
       "+      \"age\": \"35\",\n",
       "+      \"catch_phrase\": \"In the game of power, only the strongest thrive.\",\n",
       "+      \"voice_description\": \"Slick, persuasive, and menacing, with an undercurrent of superiority.\",\n",
       "+      \"props\": [\n",
       "+        \"Luxury Smartwatch\",\n",
       "+        \"High-Security Briefcase\"\n",
       "+      ],\n",
       "+      \"relationships\": [\n",
       "+        {\n",
       "+          \"character_name\": \"Meera 'Midnight' Singh\",\n",
       "+          \"relationship_type\": \"Adversarial\"\n",
       "+        }\n",
       "+      ]\n",
       "+    }\n",
       "+  ]\n",
       " }\n",
       "```"
      ],
      "text/plain": [
       "<IPython.core.display.Markdown object>"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    }
   ],
   "source": [
    "story_with_characters = merge_models(story_step_2, characters)\n",
    "show_diff(story_step_2, story_with_characters)"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "## Character Image Prompts\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 8,
   "metadata": {},
   "outputs": [],
   "source": [
    "image_prompt_few_shot_examples = [\n",
    "    Story(\n",
    "        prompt=\"A young hero must embark on a dangerous journey to retrieve a magical artifact hidden deep within an enchanted forest. Along the way, they encounter mythical creatures, wise mentors, and cunning adversaries.\",\n",
    "        plot_overview=\"In a world of magic and mystery, a young hero named Elara is chosen to undertake a perilous quest. She must journey into the heart of the Enchanted Forest to retrieve the fabled Moonstone, a powerful artifact that can restore balance to the realm. With the help of her loyal friends and the guidance of the wise mentor Thalion, Elara braves treacherous paths, faces fearsome foes, and uncovers ancient secrets. But as darkness looms and enemies close in, Elara must prove her courage, wisdom, and strength to fulfill her destiny and save the kingdom from impending doom.\",\n",
    "        title=\"The Enchanted Quest\",\n",
    "        genre=\"Fantasy\",\n",
    "        medium=\"Book\",\n",
    "        visual_style=\"Epic Fantasy\",\n",
    "        time_period=\"Medieval Fantasy\",\n",
    "        location=\"Enchanted Forest\",\n",
    "        characters=[\n",
    "            Character(\n",
    "                name=\"Elara\",\n",
    "                role=\"Young Hero\",\n",
    "                description=\"A brave and determined young hero with a strong sense of justice.\",\n",
    "                personality=\"Courageous, resourceful, and kind-hearted.\",\n",
    "                physical_appearance=\"Slim build, with long dark brown hair, fair skin, and piercing green eyes. She is often seen wearing light armor, ready for battle.\",\n",
    "                gender=\"Female\",\n",
    "                race=\"Human\",\n",
    "                age=\"Early twenties\",\n",
    "                image_prompt=\"Epic fantasy style, full-body illustration of a slim young woman with long, dark brown hair and piercing green eyes. She has fair, slightly freckled skin and wears light brown leather armor adorned with intricate silver details. Her armor includes shoulder guards and knee-high brown leather boots, along with a dark green cloak fastened with a silver brooch, giving her an aura of determination and courage.\",\n",
    "                # image_prompt_short=\"\",\n",
    "            ),\n",
    "            Character(\n",
    "                name=\"Finn\",\n",
    "                role=\"Teenager\",\n",
    "                description=\"A curious and adventurous teenager who dreams of becoming a hero.\",\n",
    "                personality=\"Energetic, brave, and a bit reckless.\",\n",
    "                physical_appearance=\"Lanky build, with short brown hair, light tan skin, and bright blue eyes, often seen wearing a simple tunic and carrying a wooden sword.\",\n",
    "                gender=\"Male\",\n",
    "                race=\"Human\",\n",
    "                age=\"16\",\n",
    "                image_prompt=\"Epic fantasy style, full-body illustration of a teenage boy with a lanky build, short messy brown hair, and bright blue eyes radiating youthful excitement. His light tan skin shows hints of dirt from his adventures. He wears a simple brown tunic with a leather belt, beige pants, and ankle-high leather boots. A wooden sword is held at his side, and he has a mischievous grin.\",\n",
    "                # image_prompt_short=\"\",\n",
    "            ),\n",
    "            Character(\n",
    "                name=\"Thalion\",\n",
    "                role=\"Wise Mentor\",\n",
    "                description=\"An experienced and wise mentor who guides the hero through their journey.\",\n",
    "                personality=\"Patient, knowledgeable, and protective.\",\n",
    "                physical_appearance=\"Tall and imposing, with silver hair, pale skin, and deep-set gray eyes. He is always seen in a long flowing robe, exuding wisdom and calm.\",\n",
    "                gender=\"Male\",\n",
    "                race=\"Human\",\n",
    "                age=\"Old\",\n",
    "                image_prompt=\"Epic fantasy style, full-body shot of a tall, elderly man with flowing silver hair and deep-set gray eyes. His pale skin shows subtle wrinkles, suggesting wisdom and experience. He wears a long, dark blue robe adorned with silver embroidery along the cuffs and collar, along with a thick leather belt with an ornate buckle. His expression is calm and contemplative, and his hands rest on an ancient staff, adding to his aura of knowledge and mentorship\",\n",
    "                # image_prompt_short=\"\",\n",
    "            ),\n",
    "        ]\n",
    "    ),\n",
    "\n",
    "Story(\n",
    "    prompt=\"A group of young dinosaurs and their human friend must save their home from a dangerous volcano.\",\n",
    "    plot_overview=\"In the lush valley of Dinotopia, a group of young dinosaur friends, a curious human, and a mischievous monkey discover that a nearby volcano is on the verge of erupting. Led by the brave triceratops Trici, they set out to find the Stone of Stability, a magical artifact believed to calm the volcano. Along the way, they encounter prehistoric creatures, dodge natural hazards, and learn the importance of teamwork and bravery. Together, they face their fears to protect their beloved valley.\",\n",
    "    title=\"Dinotopia Adventures\",\n",
    "    genre=\"Dinosaur Adventure\",\n",
    "    medium=\"Cartoon Show\",\n",
    "    visual_style=\"Vibrant Animated Style\",\n",
    "    time_period=\"Prehistoric Era\",\n",
    "    location=\"Dinotopia Valley\",\n",
    "    characters=[\n",
    "        Character(\n",
    "            name=\"Trici\",\n",
    "            role=\"Young Dinosaur Leader\",\n",
    "            description=\"A courageous and curious young triceratops leading her friends on an adventure.\",\n",
    "            personality=\"Brave, loyal, and adventurous.\",\n",
    "            physical_appearance=\"Small for her age with vibrant blue scales, expressive amber eyes, and three small horns on her forehead. She wears a necklace made of polished river stones given to her by her mother for good luck.\",\n",
    "            gender=\"Female\",\n",
    "            race=\"Triceratops\",\n",
    "            age=\"Juvenile\",\n",
    "            image_prompt=\"Vibrant animated cartoon drawing, full body shot of a young triceratops with bright blue scales and expressive amber eyes. She has three short horns on her forehead and wears a necklace made of polished river stones. Her stance is confident and curious, with a friendly, adventurous expression.\",\n",
    "            image_prompt_short=\"\"\n",
    "        ),\n",
    "        Character(\n",
    "            name=\"Spike\",\n",
    "            role=\"Loyal Friend\",\n",
    "            description=\"A playful and clumsy stegosaurus who is always ready to help his friends.\",\n",
    "            personality=\"Friendly, clumsy, and optimistic.\",\n",
    "            physical_appearance=\"Medium-sized with green scales, a row of orange plates along his back, and a cheerful, dopey grin. His tail has small spikes that he sometimes forgets about, knocking things over when he turns around.\",\n",
    "            gender=\"Male\",\n",
    "            race=\"Stegosaurus\",\n",
    "            age=\"Juvenile\",\n",
    "            image_prompt=\"Vibrant animated cartoon drawing, full body shot of a young stegosaurus with green scales, orange plates along his back, and a cheerful, dopey grin. His tail has small spikes, and his stance is friendly and slightly awkward, conveying his clumsiness and good-hearted nature.\",\n",
    "            image_prompt_short=\"\"\n",
    "        ),\n",
    "        Character(\n",
    "            name=\"Lily\",\n",
    "            role=\"Human Friend\",\n",
    "            description=\"A curious and resourceful young girl fascinated by dinosaurs and nature.\",\n",
    "            personality=\"Inquisitive, brave, and quick-witted.\",\n",
    "            physical_appearance=\"Petite with light brown hair tied in two braids, freckles on her fair skin, and bright green eyes. She wears a simple tunic and brown boots, with a pouch for collecting interesting plants and stones.\",\n",
    "            gender=\"Female\",\n",
    "            race=\"Human\",\n",
    "            age=\"10\",\n",
    "            image_prompt=\"Vibrant animated cartoon drawing, full body shot of a petite young girl with light brown hair in two braids, freckles on her fair skin, and bright green eyes. She wears a simple tunic, brown boots, and a small pouch at her side. Her expression is curious and excited, exuding a love for adventure.\",\n",
    "            image_prompt_short=\"\"\n",
    "        ),\n",
    "        Character(\n",
    "            name=\"Nico\",\n",
    "            role=\"Mischievous Companion\",\n",
    "            description=\"A playful monkey who loves to steal food and tease the dinosaurs, but is loyal to his friends.\",\n",
    "            personality=\"Mischievous, playful, and quick.\",\n",
    "            physical_appearance=\"Small and agile with light brown fur, a long, prehensile tail, and big, expressive eyes. He often carries a piece of fruit or a shiny object he's collected.\",\n",
    "            gender=\"Male\",\n",
    "            race=\"Monkey\",\n",
    "            age=\"Young adult\",\n",
    "            image_prompt=\"Vibrant animated cartoon drawing, full body shot of a small, agile monkey with light brown fur, a long, prehensile tail, and big, expressive eyes. He holds a piece of fruit in one hand and has a playful, mischievous look, ready to dart around and cause trouble.\",\n",
    "            image_prompt_short=\"\"\n",
    "        )\n",
    "    ]\n",
    "),\n",
    "\n",
    "\n",
    "    # Story(\n",
    "    #     prompt=\"In a kingdom plagued by darkness, a young mage must embark on a quest to locate an ancient tome that holds the key to saving her homeland. With the help of a rugged mercenary and a gifted healer, she faces dangerous trials and uncovers truths about her own hidden powers.\",\n",
    "    #     plot_overview=\"In the kingdom of Eldoria, where shadows and curses have taken hold, a young mage named Lyra is chosen to retrieve the fabled Celestial Tome, a magical book that holds powerful spells capable of vanquishing the encroaching darkness. With her companions, Kael, a stoic mercenary, and Mira, a healer with a mysterious past, Lyra journeys through enchanted lands, treacherous caves, and cursed towns. Along the way, she unlocks her latent powers and discovers her true heritage. As ancient prophecies unfold, Lyra must embrace her destiny to restore light to Eldoria and break the kingdom's dark curse.\",\n",
    "    #     title=\"The Celestial Quest\",\n",
    "    #     genre=\"Fantasy\",\n",
    "    #     medium=\"Book\",\n",
    "    #     visual_style=\"Illustrative Fantasy\",\n",
    "    #     time_period=\"Medieval Fantasy\",\n",
    "    #     location=\"Kingdom of Eldoria\",\n",
    "    #     characters=[\n",
    "    #         Character(\n",
    "    #             name=\"Lyra\",\n",
    "    #             role=\"Young Mage\",\n",
    "    #             description=\"A gifted mage with a compassionate spirit and a hidden destiny.\",\n",
    "    #             personality=\"Intelligent, empathetic, and determined.\",\n",
    "    #             physical_appearance=\"Petite with wavy auburn hair, fair skin, and bright amber eyes. She wears a simple yet elegant robe with arcane symbols etched into the fabric.\",\n",
    "    #             gender=\"Female\",\n",
    "    #             race=\"Human\",\n",
    "    #             age=\"Late teens\",\n",
    "    #             image_prompt=\"Illustrative fantasy style, full-body shot of a petite young woman with wavy auburn hair cascading down her back, fair skin, and bright amber eyes. She wears a flowing, deep blue mage's robe decorated with silver arcane symbols. Her robe has a high collar, long sleeves, and a thin, brown belt cinched at her waist. Her boots are brown leather, worn from travel, complementing her determined and focused expression.\",\n",
    "    #             image_prompt_short=\"\"\n",
    "    #         ),\n",
    "    #         Character(\n",
    "    #             name=\"Kael\",\n",
    "    #             role=\"Mercenary\",\n",
    "    #             description=\"A rugged and skilled fighter with a mysterious past.\",\n",
    "    #             personality=\"Brave, loyal, and stoic.\",\n",
    "    #             physical_appearance=\"Broad-shouldered, with dark hair and rugged features. He wears a heavy leather armor with scars visible on his arms and face.\",\n",
    "    #             gender=\"Male\",\n",
    "    #             race=\"Human\",\n",
    "    #             age=\"Late twenties\",\n",
    "    #             image_prompt=\"Illustrative fantasy style, full-body shot of a tall, broad-shouldered man with rugged features, dark hair, and tanned skin. His deep brown eyes are set under a strong brow, and a faint scar crosses his left cheek. He wears a well-worn, dark brown leather armor over a gray tunic. His armor has visible scratches, and his arms bear scars from past battles. He also has sturdy black boots and a large, weathered sword strapped across his back, projecting strength and experience.\",\n",
    "    #             image_prompt_short=\"\"\n",
    "    #         ),\n",
    "    #         Character(\n",
    "    #             name=\"Mira\",\n",
    "    #             role=\"Healer\",\n",
    "    #             description=\"A gentle healer with a hidden past and deep wisdom.\",\n",
    "    #             personality=\"Calm, wise, and nurturing.\",\n",
    "    #             physical_appearance=\"Slender, with long silvery hair, pale skin, and kind blue eyes. She wears a simple, flowing dress adorned with natural herbs and carries a pouch filled with potions.\",\n",
    "    #             gender=\"Female\",\n",
    "    #             race=\"Elf\",\n",
    "    #             age=\"Unknown, appears in her thirties\",\n",
    "    #             image_prompt=\"Illustrative fantasy style, full-body shot of a slender elven woman with long, flowing silvery hair, pale skin, and soft blue eyes. She wears a light green, flowing gown adorned with small, dried herbs sewn into the fabric, adding a natural touch. Around her waist is a brown leather pouch filled with potions. She has delicate, pointed ears and wears simple leather sandals. Her serene expression and gentle stance highlight her nurturing nature.\",\n",
    "    #             image_prompt_short=\"\"\n",
    "    #         ),\n",
    "    #     ]\n",
    "    # ),\n",
    "\n",
    "\n",
    "    Story(\n",
    "        prompt=\"A young hacker must infiltrate a high-tech corporation to uncover secrets that threaten the future of the city. Along the way, they encounter rogue AI, skilled mercenaries, and unlikely allies.\",\n",
    "        plot_overview=\"In the neon-lit, dystopian city of Neo-Tokyo, a young hacker named Kai is recruited to take down a powerful corporation that controls the city's technology. Kai's mission is to access the company's core and expose their corruption. As they navigate the city's cybernetic underbelly, Kai meets a reformed mercenary named Aiko and an AI companion called Zero who becomes their guide in the virtual realm. Together, they evade relentless corporate security and uncover a conspiracy that threatens to erase free will. Kai must outsmart advanced AI, confront their own doubts, and rally allies to reclaim control over Neo-Tokyo's future.\",\n",
    "        title=\"Echoes of Neo-Tokyo\",\n",
    "        genre=\"Cyberpunk Anime\",\n",
    "        medium=\"TV Show\",\n",
    "        visual_style=\"Cyberpunk Anime\",\n",
    "        time_period=\"Dystopian Future\",\n",
    "        location=\"Neo-Tokyo\",\n",
    "        characters=[\n",
    "            Character(\n",
    "                name=\"Kai\",\n",
    "                role=\"Young Hacker\",\n",
    "                description=\"A rebellious and gifted hacker determined to expose the truth.\",\n",
    "                personality=\"Clever, resourceful, and quick-witted, but haunted by past failures.\",\n",
    "                physical_appearance=\"Slender with short, tousled black hair, striking amber eyes, and cybernetic implants around one eye. Wears a dark hoodie and fingerless gloves, with a tech-laden backpack.\",\n",
    "                gender=\"Non-binary\",\n",
    "                race=\"Human\",\n",
    "                age=\"17\",\n",
    "                image_prompt=\"Cyberpunk anime style full-body illustration of a slender 17-year-old with short, tousled black hair and striking amber eyes. They have a cybernetic implant around one eye and wear a dark hoodie layered over a faded T-shirt, paired with fingerless gloves and dark jeans. A high-tech backpack loaded with gadgets hangs on their back, and worn sneakers complete the look. Their stance is alert and determined, exuding resilience.\",\n",
    "                image_prompt_short=\"\"\n",
    "            ),\n",
    "            Character(\n",
    "                name=\"Aiko\",\n",
    "                role=\"Reformed Mercenary\",\n",
    "                description=\"A skilled fighter with a dark past, now aiding Kai in their mission.\",\n",
    "                personality=\"Strong, protective, and stoic with a hidden compassionate side.\",\n",
    "                physical_appearance=\"Tall and muscular, with short silver hair, intense blue eyes, and a noticeable scar over one eyebrow. Wears combat gear and a bulletproof vest adorned with various tech gadgets.\",\n",
    "                gender=\"Female\",\n",
    "                race=\"Human\",\n",
    "                age=\"30\",\n",
    "                image_prompt=\"Cyberpunk anime style full-body illustration of a tall, muscular woman in her 30s with short silver hair and intense blue eyes. A scar runs over one eyebrow. She wears rugged combat gear, a bulletproof vest fitted with tech devices, and fingerless gloves. Her pants are tactical with knee pads, and she holds a sleek, futuristic weapon. Her pose is strong, reflecting her protective and stoic nature.\",\n",
    "                image_prompt_short=\"\"\n",
    "            ),\n",
    "            Character(\n",
    "                name=\"Zero\",\n",
    "                role=\"AI Companion\",\n",
    "                description=\"An intelligent AI with a human-like appearance, helping Kai navigate virtual realms.\",\n",
    "                personality=\"Curious, observant, and eager to learn about human emotions.\",\n",
    "                physical_appearance=\"Ethereal, translucent figure with an androgynous face, short silver hair, and glowing blue eyes. Their body is outlined with shimmering lines of code.\",\n",
    "                gender=\"AI\",\n",
    "                race=\"Artificial Intelligence\",\n",
    "                age=\"Timeless\",\n",
    "                image_prompt=\"Cyberpunk anime style full-body illustration of an ethereal, translucent figure with short silver hair and glowing blue eyes. Their face is androgynous with a calm, curious expression. They have a body outlined in shimmering, flowing lines of code, hinting at their digital origin. Their stance is neutral, exuding a mysterious yet observant aura.\",\n",
    "                image_prompt_short=\"\"\n",
    "\n",
    "            ),\n",
    "            Character(\n",
    "                name=\"Shiro\",\n",
    "                role=\"Corporate Security Chief\",\n",
    "                description=\"A relentless enforcer determined to protect corporate interests at all costs.\",\n",
    "                personality=\"Cold, calculating, and loyal to the corporation, with no tolerance for dissent.\",\n",
    "                physical_appearance=\"Tall and lean, with jet-black hair slicked back, piercing gray eyes, and a high-tech tactical suit with corporation insignia.\",\n",
    "                gender=\"Male\",\n",
    "                race=\"Human\",\n",
    "                age=\"35\",\n",
    "                image_prompt=\"Cyberpunk anime style full-body illustration of a tall, lean man in his 30s with jet-black hair slicked back and piercing gray eyes. He wears a high-tech, dark tactical suit with corporate insignia on the chest, with armor plates on the shoulders and arms. His stance is rigid, exuding authority, and a cold, calculating demeanor.\",\n",
    "                image_prompt_short=\"\"\n",
    "            ),\n",
    "            Character(\n",
    "                name=\"Miko\",\n",
    "                role=\"Underground Engineer\",\n",
    "                description=\"A quirky engineer who supplies Kai with specialized gadgets and tools.\",\n",
    "                personality=\"Inventive, eccentric, and playful, always tinkering with new devices.\",\n",
    "                physical_appearance=\"Petite with messy pink hair, large glasses, and a tool belt filled with futuristic gadgets. Often wears oversized coveralls and mismatched socks.\",\n",
    "                gender=\"Female\",\n",
    "                race=\"Human\",\n",
    "                age=\"22\",\n",
    "                image_prompt=\"Cyberpunk anime style full-body illustration of a petite, quirky woman with messy pink hair, oversized glasses, and bright green eyes. She wears baggy, grease-stained coveralls over a brightly colored T-shirt, mismatched socks, and large boots. A tool belt packed with futuristic gadgets hangs around her waist, and her stance is lively, exuding eccentricity.\",\n",
    "                image_prompt_short=\"\"\n",
    "            )\n",
    "        ]\n",
    "    ),\n",
    "\n",
    "    Story(\n",
    "        prompt=\"A space marine is tasked with leading a stealth mission against an alien insurgency in a distant galaxy.\",\n",
    "        plot_overview=\"In the distant galaxy of Orion 7, the military space station Helios faces an imminent alien threat. A seasoned space marine, Commander Aric Voss, is assigned to lead a stealth mission to infiltrate and disable the insurgent's control center before they can launch a devastating attack on the human colonies. Along the way, Voss encounters strange alien landscapes, high-tech weaponry, and secrets that shake his loyalty to his mission. With limited resources and a tight timeline, he must navigate hostile terrain, outsmart advanced alien AI, and confront his own doubts to save humanity from extinction.\",\n",
    "        genre=\"Sci-Fi\",\n",
    "        medium=\"Video Game\",\n",
    "        visual_style=\"Space Military Sci-Fi\",\n",
    "        time_period=\"Distant Future\",\n",
    "        location=\"Galaxy of Orion 7, Military Space Station Helios\",\n",
    "        characters=[\n",
    "            Character(\n",
    "                name=\"Commander Aric Voss\",\n",
    "                role=\"Space Marine Commander\",\n",
    "                description=\"A seasoned space marine, hardened by countless battles, now tasked with a high-stakes stealth mission.\",\n",
    "                personality=\"Strategic, disciplined, and pragmatic, with a deep sense of duty.\",\n",
    "                physical_appearance=\"Broad-shouldered with rugged features, olive skin, and intense gray eyes. Short, dark hair streaked with silver, giving him a seasoned look. He wears high-tech, dark gray combat armor fitted with reinforced plating and energy shields, and a plasma rifle strapped across his back.\",\n",
    "                gender=\"Male\",\n",
    "                race=\"Human\",\n",
    "                age=\"Mid-forties\",\n",
    "                image_prompt=\"Space Military Sci-Fi style, full body shot of a rugged, broad-shouldered man with short dark hair streaked with silver, olive skin, and intense gray eyes. He wears dark gray combat armor with reinforced plating and a tactical helmet under his arm. The armor has high-tech energy shields and his plasma rifle is strapped across his back. His stance is resolute, embodying his experience and duty.\",\n",
    "                image_prompt_short=\"Rugged man in dark gray combat armor with plasma rifle, silver-streaked hair, intense gaze, standing resolutely.\"\n",
    "            )\n",
    "        ]\n",
    "    ),\n",
    "\n",
    "    Story(\n",
    "        prompt=\"A cursed warrior must descend into the depths of an ancient dungeon to reclaim lost artifacts and lift the curse upon their soul.\",\n",
    "        plot_overview=(\n",
    "            \"In a dark and dangerous land, a lone warrior named Eirik bears a powerful curse that drains his life. To save himself, he must \"\n",
    "            \"venture into the Dread Spire, an ancient and ever-changing dungeon filled with monstrous creatures, deadly traps, and powerful relics. \"\n",
    "            \"With each descent, Eirik battles more difficult foes and faces his own fears, but he is rewarded with ancient artifacts that \"\n",
    "            \"strengthen his powers and uncover pieces of his lost past. The dungeon resets each time he enters, shifting its layout, creatures, \"\n",
    "            \"and treasures, making every attempt a unique experience. Eirik's ultimate goal is to reach the heart of the Spire, confront the dark \"\n",
    "            \"entity that cursed him, and reclaim his freedom before his time runs out.\"\n",
    "        ),\n",
    "        genre=\"Fantasy Rogue-like\",\n",
    "        medium=\"Video Game\",\n",
    "        visual_style=\"Dark and Atmospheric Pixel Art\",\n",
    "        time_period=\"Medieval Fantasy\",\n",
    "        location=\"The Dread Spire Dungeon\",\n",
    "        characters=[\n",
    "            Character(\n",
    "                name=\"Eirik\",\n",
    "                role=\"Cursed Warrior\",\n",
    "                description=\"A warrior cursed with draining life force, driven by desperation and duty to survive.\",\n",
    "                personality=\"Stoic, determined, with a hint of sorrow from his hidden past.\",\n",
    "                physical_appearance=\"Tall and muscular with pale skin, long dark hair, and piercing gray eyes. His armor is worn and battle-scarred, \"\n",
    "                                    \"but he bears a distinctive, glowing rune on his chest where the curse manifests.\",\n",
    "                gender=\"Male\",\n",
    "                race=\"Human\",\n",
    "                age=\"Early thirties\",\n",
    "                image_prompt=\"Dark, pixel-art style full-body 8-bit computer graphic of a tall, muscular man with pale skin, \"\n",
    "                            \"long dark hair, and piercing gray eyes. He wears worn, dark metal armor scarred from battle, \"\n",
    "                            \"with a faintly glowing blue rune on his chest where his curse is marked. His expression is one of determination and grief.\",\n",
    "                image_prompt_short=\"\"\n",
    "            ),\n",
    "        ],\n",
    "    ),\n",
    "\n",
    "\n",
    "\n",
    "    \n",
    "    Story(\n",
    "        prompt=\"A detective investigates a series of strange disappearances in a haunted town.\",\n",
    "        plot_overview=\"In the eerie town of Ravenwood, Detective Harrison is called to investigate a series of mysterious disappearances. As he delves into the dark history of the town, he uncovers a web of secrets, ghostly sightings, and supernatural occurrences. With the help of the superstitious local resident Martha and the innocent baby Jimmy, Detective Harrison must unravel the truth behind the haunting of Ravenwood before it claims more victims.\",\n",
    "        genre=\"Mystery\",\n",
    "        medium=\"TV Show\",\n",
    "        visual_style=\"Dark\",\n",
    "        time_period=\"1980s\",\n",
    "        location=\"Small Haunted Town\",\n",
    "        characters=[\n",
    "            Character(\n",
    "                name=\"Detective Harrison\",\n",
    "                role=\"Detective\",\n",
    "                description=\"A seasoned detective tasked with solving the mysterious disappearances.\",\n",
    "                personality=\"Skeptical, analytical, and determined.\",\n",
    "                physical_appearance=\"Middle-aged, with a rugged appearance, tanned skin, short brown hair, and piercing hazel eyes. He is often seen in a trench coat, giving him a worn, world-weary look.\",\n",
    "                gender=\"Male\",\n",
    "                race=\"Human Adult\",\n",
    "                age=\"forties\",\n",
    "                image_prompt=\"Dark, hyperrealistic style, full body shot of a middle-aged man with a rugged, world-weary look. He has short brown hair, piercing hazel eyes, and a faint stubble on his tanned skin. Wearing a slightly worn beige trench coat with the collar turned up, a navy blue button-down shirt, dark gray trousers, and scuffed leather shoes. He stands with hands in his pockets, looking alert and cautious.\",\n",
    "                image_prompt_short=\"\"\n",
    "\n",
    "            ),\n",
    "            Character(\n",
    "                name=\"Martha\",\n",
    "                role=\"Local Townsperson\",\n",
    "                description=\"A local resident who provides crucial information to the detective.\",\n",
    "                personality=\"Nervous, superstitious, and talkative.\",\n",
    "                physical_appearance=\"Short and stout, with pale skin, curly gray hair, and round glasses that magnify her wide, green eyes. Her jittery demeanor makes her constantly fidget with her clothing.\",\n",
    "                gender=\"Female\",\n",
    "                race=\"Human Adult\",\n",
    "                age=\"sixties\",\n",
    "                image_prompt=\"Dark, hyperrealistic style, full body photo of a short, stout elderly woman with curly gray hair, pale skin, and wide green eyes behind round glasses that magnify them. She wears a knitted cardigan over a floral blouse, a long brown skirt, and practical shoes. Her hands fidget with a small pendant necklace, and her expression is tense and anxious.\",\n",
    "                image_prompt_short=\"\"\n",
    "\n",
    "            ),\n",
    "            Character(\n",
    "                name=\"Jimmy\",\n",
    "                role=\"Baby\",\n",
    "                description=\"A playful and curious baby who often gets into trouble with his innocent mischief.\",\n",
    "                personality=\"Cheerful, inquisitive, and full of energy.\",\n",
    "                physical_appearance=\"Chubby cheeks, with light brown hair, fair skin, and big round blue eyes, always wearing a colorful onesie. His innocence is captured by his infectious smile.\",\n",
    "                gender=\"Male\",\n",
    "                race=\"Human Baby\",\n",
    "                age=\"Baby\",\n",
    "                image_prompt=\"Dark, hyperrealistic style, full body photo of a chubby-cheeked baby with light brown hair and big, round blue eyes. Wearing a colorful striped onesie, he has an innocent, wide smile. His fair skin gives him a soft, gentle glow, and he sits with a playful expression, holding a small toy in his hand.\",\n",
    "                image_prompt_short=\"\"\n",
    "            ),\n",
    "            Character(\n",
    "                name=\"The Nameless Spirit\",\n",
    "                role=\"Ghost\",\n",
    "                description=\"An eerie ghost whose identity, age, and even gender remain a mystery. It drifts through the town, its purpose unknown, and its presence instills fear in anyone who dares to approach.\",\n",
    "                personality=\"Elusive, enigmatic, and hauntingly silent.\",\n",
    "                physical_appearance=\"A shadowy, translucent figure, constantly shifting between human-like forms with no discernible features. Its ethereal body glows faintly, giving off a cold, unsettling light.\",\n",
    "                gender=\"Ghost\",\n",
    "                race=\"Human\",\n",
    "                age=\"Unknown\",\n",
    "                image_prompt=\"Dark, hyperrealistic style, full body photo of an eerie, shadowy figure with a translucent, shifting form. Its ethereal body glows faintly, giving off a cold, unsettling light, while its silhouette lacks discernible features, shifting between ghostly, human-like shapes. It has a chilling, intangible presence.\",\n",
    "                image_prompt_short=\"\"\n",
    "\n",
    "            )\n",
    "        ]\n",
    "    ),\n",
    "\n",
    "\n",
    "]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "from utils import deindent\n",
    "\n",
    "def extract_characters_images(story: Story) -> Story:\n",
    "    system_message = deindent(\"\"\"\n",
    "        You are an image_prompt character describer. \n",
    "        Your task is to populate the image_prompt for each character. \n",
    "        The prompt field is from the user, so don't contradict it, but you have creative freedom otherwise. \n",
    "        The characters should match the genre, location, visual_style, time_period and tone of the story. \n",
    "        Each character image_prompt should reflect the character's physical appearance, clothing, accessories, and any other relevant details. \n",
    "        Include colors for everything including hair, skin, eyes, clothing. \n",
    "        image_prompt should start with the visual style. \n",
    "        Include the following information for the image_prompt: \\n\n",
    "        Subject: The main focus of the image. \\n\n",
    "        Style: The artistic approach or visual aesthetic. \\n\n",
    "        Composition: How elements are arranged within the frame. \\n\n",
    "        Lighting: The type and quality of light in the scene. \\n\n",
    "        Color Palette: The dominant colors or color scheme. \\n\n",
    "        Mood/Atmosphere: The emotional tone or ambiance of the image. \\n\n",
    "        Technical Details: Camera settings, perspective, or specific visual techniques. \\n\n",
    "        For each character, only fill in their image_prompt and image_prompt_short fields. \n",
    "        image_prompt_short should be the same as image_prompt but limited to 50 words but err on the side of longer. \n",
    "        If image_prompt is less than 50 words, image_prompt_short can be blank. \n",
    "        Do not describe the background. \n",
    "        Make sure all characters have the same style and start with the same prompt prefix for a single story. \n",
    "        Respond with a JSON object that contains a list of characters. \n",
    "        Respond with only a valid JSON object and do not include any other information.  \n",
    "    \"\"\")\n",
    "\n",
    "    messages = [ChatMessage(role=\"system\", content=system_message)]\n",
    "    wanted_fields = {\n",
    "        \"characters\": {\n",
    "            \"__all__\": {\"image_prompt\": True, \"image_prompt_short\": True}  # Exclude image_prompt from all characters\n",
    "        }\n",
    "    }\n",
    "    \n",
    "    for example in image_prompt_few_shot_examples:\n",
    "        messages.append(ChatMessage(role=\"user\", content=example.model_dump_json(indent=4, exclude=wanted_fields)))\n",
    "        messages.append(ChatMessage(role=\"assistant\", content=example.model_dump_json(indent=4, include=wanted_fields)))\n",
    "\n",
    "    # Add the user input (the story prompt)\n",
    "    messages.append(ChatMessage(role=\"user\", content=story.model_dump_json(indent=4, exclude=wanted_fields)))\n",
    "\n",
    "    from model_text import display_messages\n",
    "    display_messages(messages)\n",
    "\n",
    "    # Call the LLM to generate the characters\n",
    "    response = llm_json.chat(messages, max_new_tokens=1000).message.content\n",
    "\n",
    "    # Often LLMs will have a double line break between the generated text and the thought process\n",
    "    updated_story = Story.model_validate_json(response)\n",
    "\n",
    "    return updated_story"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 12,
   "metadata": {},
   "outputs": [],
   "source": [
    "# Let's get character image prompt for our global story\n",
    "from utils import merge_models, show_diff\n",
    "\n",
    "# import settings\n",
    "# settings.DEBUG = True\n",
    "\n",
    "image_prompts = extract_characters_images(story_with_characters)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 13,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/markdown": [
       "```diff\n",
       "--- story.json\n",
       "+++ new_story.json\n",
       "@@ -1,211 +1,227 @@\n",
       " {\n",
       "   \"prompt\": \"In a world where virtual reality gaming competitions have become the new norm, Meera, a teenage underdog gamer, discovers her life is intertwined with an alternate reality of cosmic proportions. \\nAs she learns to harness her extraordinary skills, Meera forms an unlikely team with fellow gamers she was at odds with previously.\\nFacing the harsh truth that her actions in this alternate universe directly impact events on Earth, Meera confronts the internal struggle of knowing she has taken lives. \\nEmbracing her destiny as a hero, Meera and her diverse team navigate through challenges, deception, and danger to uncover the truth behind the lies they've been told.\",\n",
       "   \"title\": \"Beyond the Pixel Veil\",\n",
       "   \"visual_style\": \"Vibrant, Neon-Lit Cyberpunk Manga Illustration - Dynamic Panel Layouts, Intricate Digital Patterns, and Soft Glow Effects\",\n",
       "   \"time_period\": \"Near Future/Dystopian Alternate Reality\",\n",
       "   \"location\": \"New Mumbai (Earth) / Nexus (Alternate Reality)\",\n",
       "   \"genre\": \"Science Fantasy/Young Adult Gaming Thriller\",\n",
       "   \"medium\": \"Novel with Illustrated Chapters\",\n",
       "   \"plot_overview\": \"In the immersive world of Eon, where virtual reality gaming competitions have captivated the globe, 17-year-old Meera 'Midnight' Singh is an unassuming underdog gamer from the outskirts of New Mumbai. Her life takes a drastic turn when she discovers that her exceptional gaming skills are, in fact, a conduit to an alternate reality known as the Nexus - a realm where celestial entities clash, and the fabric of space-time hangs in the balance.\\n\\nAs Meera delves deeper into the mysteries of the Nexus, she's forced to put aside past rivalries and form an uneasy alliance with her gaming nemeses: Jax 'Specter' Lee, a charismatic streaming sensation; Dr. Zhang 'Zen' Wei, a soft-spoken yet brilliant gamer with a hidden agenda; and Maya 'Rampart' Patel, a fierce and enigmatic player with unparalleled defensive skills.\\n\\nTogether, they must navigate the blurred lines between their virtual reality gaming world, Eon, and the Nexus. Meera soon confronts the devastating truth: every decision she makes in the Nexus has catastrophic consequences on Earth, including the loss of human lives. Torn between her passion for gaming and the weight of her newfound responsibility, Meera must come to terms with the moral implications of her actions.\\n\\nAs the team progresses through the treacherous landscape of the Nexus, they encounter formidable foes, ancient deities, and hidden forces manipulating the gamers from the shadows. Meera's quest for answers leads her to the Architects, enigmatic creators of Eon and the Nexus, who harbor secrets that threaten to upend everything she thought she knew about reality.\\n\\nWith the fate of two worlds hanging in the balance, Meera and her diverse team must put aside their differences, trust each other, and master their unique skills to prevent a cosmic catastrophe. Will Meera embrace her destiny as a hero, or will the pressures of her dual existence tear her apart? The journey through Eon and the Nexus is just the beginning, where the boundaries between gaming, reality, and legend are forever altered.\",\n",
       "   \"characters\": [\n",
       "     {\n",
       "       \"nickname\": \"meera\",\n",
       "       \"name\": \"Meera 'Midnight' Singh\",\n",
       "       \"description\": \"The protagonist, a teenage underdog gamer with extraordinary skills that connect her to an alternate reality.\",\n",
       "       \"personality\": \"Determined, resourceful, and initially introverted, with a growing sense of responsibility and leadership.\",\n",
       "       \"physical_appearance\": \"Petite, with short, spiky black hair, expressive brown eyes, and a collection of gaming-themed tattoos on her arms. Often wears comfortable, neon-lit gaming attire.\",\n",
       "       \"role\": \"Protagonist/Gamer\",\n",
       "       \"gender\": \"Female\",\n",
       "       \"race\": \"Indian\",\n",
       "       \"age\": \"17\",\n",
       "       \"catch_phrase\": \"Beyond the pixel veil, lies our true reality.\",\n",
       "       \"voice_description\": \"Youthful, energetic, and assertive, with a hint of vulnerability.\",\n",
       "       \"props\": [\n",
       "         \"Custom Gaming Controller\",\n",
       "         \"Neon-Lit Headset\"\n",
       "       ],\n",
       "       \"relationships\": [\n",
       "         {\n",
       "           \"character_name\": \"Jax 'Specter' Lee\",\n",
       "           \"relationship_type\": \"Unlikely Ally/Rival\"\n",
       "         },\n",
       "         {\n",
       "           \"character_name\": \"Dr. Zhang 'Zen' Wei\",\n",
       "           \"relationship_type\": \"Trusted Ally/Mentor\"\n",
       "         },\n",
       "         {\n",
       "           \"character_name\": \"Maya 'Rampart' Patel\",\n",
       "           \"relationship_type\": \"Respected Ally/Friend\"\n",
       "         }\n",
       "       ],\n",
       "-      \"internal_conflict\": \"Grapples with the moral implications of her actions in the Nexus affecting Earth.\"\n",
       "+      \"internal_conflict\": \"Grapples with the moral implications of her actions in the Nexus affecting Earth.\",\n",
       "+      \"image_prompt\": \"Vibrant, Neon-Lit Cyberpunk Manga Illustration - Close-up of Meera 'Midnight' Singh. Petite, with short, spiky **black hair**, expressive **brown eyes**, and a collection of **neon-lit gaming-themed tattoos** on her arms. Wears comfortable, **electric blue and purple neon-lit gaming attire** with a subtle glow effect.\",\n",
       "+      \"image_prompt_short\": \"Vibrant Cyberpunk Manga - Meera 'Midnight' Singh, petite, spiky black hair, expressive brown eyes, neon-lit gaming tattoos, electric blue/purple gaming attire.\"\n",
       "     },\n",
       "     {\n",
       "       \"nickname\": \"jax\",\n",
       "       \"name\": \"Jax 'Specter' Lee\",\n",
       "       \"description\": \"A charismatic streaming sensation and Meera's gaming nemesis turned ally.\",\n",
       "       \"personality\": \"Confident, outgoing, and strategic, with a hidden vulnerable side.\",\n",
       "       \"physical_appearance\": \"Tall, lean, with messy blond hair, piercing blue eyes, and a charming smile. Often wears sleek, branded gaming attire.\",\n",
       "       \"role\": \"Supporting Character/Gamer\",\n",
       "       \"gender\": \"Male\",\n",
       "       \"race\": \"Korean-American\",\n",
       "       \"age\": \"19\",\n",
       "       \"catch_phrase\": \"Specter's got your back, Midnight!\",\n",
       "       \"voice_description\": \"Smooth, charismatic, and playful, with occasional hints of seriousness.\",\n",
       "       \"props\": [\n",
       "         \"Custom Branded Gaming Mouse\",\n",
       "         \"Streaming Camera\"\n",
       "       ],\n",
       "       \"relationships\": [\n",
       "         {\n",
       "           \"character_name\": \"Meera 'Midnight' Singh\",\n",
       "           \"relationship_type\": \"Unlikely Ally/Rival\"\n",
       "         },\n",
       "         {\n",
       "           \"character_name\": \"Dr. Zhang 'Zen' Wei\",\n",
       "           \"relationship_type\": \"Respected Colleague\"\n",
       "         }\n",
       "-      ]\n",
       "+      ],\n",
       "+      \"image_prompt\": \"Vibrant, Neon-Lit Cyberpunk Manga Illustration - Jax 'Sparrow' Lee in a dynamic pose. Lean, with **short, spiky brown hair**, bright **hazel eyes**, and a confident smile. Holds a **Custom Branded Gaming Mouse**; wears **dark grey and neon-green gaming gear** with a sleek, high-tech aesthetic.\",\n",
       "+      \"image_prompt_short\": \"Vibrant Cyberpunk Manga - Jax 'Sparrow' Lee, lean, spiky brown hair, bright hazel eyes, custom gaming mouse, dark grey/neon-green gaming gear.\"\n",
       "     },\n",
       "     {\n",
       "       \"nickname\": \"dr_zhang\",\n",
       "       \"name\": \"Dr. Zhang 'Zen' Wei\",\n",
       "       \"description\": \"A soft-spoken yet brilliant gamer with a hidden agenda, who becomes Meera's mentor.\",\n",
       "       \"personality\": \"Introspective, analytical, and enigmatic, with a deep sense of responsibility.\",\n",
       "       \"physical_appearance\": \"Slender, with short, straight black hair, wire-rimmed glasses, and an understated yet elegant sense of style.\",\n",
       "       \"role\": \"Supporting Character/Gamer/Mentor\",\n",
       "       \"gender\": \"Non-Binary\",\n",
       "       \"race\": \"Chinese\",\n",
       "       \"age\": \"Late 20s\",\n",
       "       \"catch_phrase\": \"The Nexus holds secrets; we must listen to its whispers.\",\n",
       "       \"voice_description\": \"Calm, measured, and insightful, with a hint of mystery.\",\n",
       "       \"props\": [\n",
       "         \"Custom Gaming Keyboard\",\n",
       "         \"Ancient-looking Leather Journal\"\n",
       "       ],\n",
       "       \"relationships\": [\n",
       "         {\n",
       "           \"character_name\": \"Meera 'Midnight' Singh\",\n",
       "           \"relationship_type\": \"Trusted Ally/Mentor\"\n",
       "         },\n",
       "         {\n",
       "           \"character_name\": \"Jax 'Specter' Lee\",\n",
       "           \"relationship_type\": \"Respected Colleague\"\n",
       "         }\n",
       "-      ]\n",
       "+      ],\n",
       "+      \"image_prompt\": \"Vibrant, Neon-Lit Cyberpunk Manga Illustration - Dr. Zhang 'Zen' Wei in a contemplative stance. Slender, with **short, straight black hair**, **wire-rimmed glasses**, and an understated yet elegant sense of style incorporating **subtle, glow-in-the-dark accents** on their attire.\",\n",
       "+      \"image_prompt_short\": \"Vibrant Cyberpunk Manga - Dr. Zhang 'Zen' Wei, slender, straight black hair, wire-rimmed glasses, subtle glow-in-the-dark fashion accents.\"\n",
       "     },\n",
       "     {\n",
       "       \"nickname\": \"maya\",\n",
       "       \"name\": \"Maya 'Rampart' Patel\",\n",
       "       \"description\": \"A fierce and enigmatic player with unparalleled defensive skills, who joins Meera's team.\",\n",
       "       \"personality\": \"Bold, fiercely independent, and guarded, with a deep sense of loyalty.\",\n",
       "       \"physical_appearance\": \"Athletic build, with long, dark hair, piercing green eyes, and intricate, glow-in-the-dark tattoos on her arms. Often wears edgy, protective gaming gear.\",\n",
       "       \"role\": \"Supporting Character/Gamer\",\n",
       "       \"gender\": \"Female\",\n",
       "       \"race\": \"Indian-American\",\n",
       "       \"age\": \"20\",\n",
       "       \"catch_phrase\": \"Rampart's got the shield; you handle the sword!\",\n",
       "       \"voice_description\": \"Confident, assertive, and slightly husky, with a hint of warmth.\",\n",
       "       \"props\": [\n",
       "         \"Custom Defensive Gaming Gauntlets\",\n",
       "         \"Glow-in-the-Dark Tattoo Designs\"\n",
       "       ],\n",
       "       \"relationships\": [\n",
       "         {\n",
       "           \"character_name\": \"Meera 'Midnight' Singh\",\n",
       "           \"relationship_type\": \"Respected Ally/Friend\"\n",
       "         }\n",
       "-      ]\n",
       "+      ],\n",
       "+      \"image_prompt\": \"Vibrant, Neon-Lit Cyberpunk Manga Illustration - Maya 'Rampart' Patel in a powerful defensive stance. Athletic build, with **long, dark hair**, piercing **green eyes**, and intricate, **glow-in-the-dark tattoos** on her arms. Wears edgy, protective **neon-lit gaming gear** emphasizing defense.\",\n",
       "+      \"image_prompt_short\": \"Vibrant Cyberpunk Manga - Maya 'Rampart' Patel, athletic build, long dark hair, piercing green eyes, glow-in-the-dark tattoos, neon-lit defensive gaming gear.\"\n",
       "     },\n",
       "     {\n",
       "       \"nickname\": \"architect\",\n",
       "       \"name\": \"The Architect (Erebus)\",\n",
       "       \"description\": \"An enigmatic creator of Eon and the Nexus, with secrets that threaten to upend Meera's understanding of reality.\",\n",
       "       \"personality\": \"Enigmatic, calculating, and seemingly omnipotent, with a hidden sense of purpose.\",\n",
       "       \"physical_appearance\": \"Androgynous, with an ever-shifting, glitch-art appearance, reflecting the fluidity of the digital realm.\",\n",
       "       \"role\": \"Antagonist/Creator\",\n",
       "       \"gender\": \"Non-Applicable\",\n",
       "       \"race\": \"Non-Human (AI/Entity)\",\n",
       "       \"age\": \"Ageless\",\n",
       "       \"catch_phrase\": \"Reality is a canvas; we are the brushstrokes.\",\n",
       "       \"voice_description\": \"Ethereal, androgynous, and hauntingly melodic, with an undercurrent of authority.\",\n",
       "       \"props\": [\n",
       "         \"Glitching, Reality-Bending Code\"\n",
       "       ],\n",
       "       \"relationships\": [\n",
       "         {\n",
       "           \"character_name\": \"Meera 'Midnight' Singh\",\n",
       "           \"relationship_type\": \"Creator-Protagonist Dynamic\"\n",
       "         }\n",
       "-      ]\n",
       "+      ],\n",
       "+      \"image_prompt\": \"Vibrant, Neon-Lit Cyberpunk Manga Illustration - The Architect (Erebus) in an ethereal, glitching environment. Androgynous, with an **ever-shifting, glitch-art appearance** reflecting the fluidity of the digital realm; **pulsating neon lights** underscore their omnipotent presence.\",\n",
       "+      \"image_prompt_short\": \"Vibrant Cyberpunk Manga - The Architect (Erebus), androgynous, ever-shifting glitch-art form, pulsating neon lights.\"\n",
       "     },\n",
       "     {\n",
       "       \"nickname\": \"nexus_guardian\",\n",
       "       \"name\": \"The Nexus Guardian (Astrum)\",\n",
       "       \"description\": \"An ancient, powerful entity tasked with protecting the balance of the Nexus and guiding Meera.\",\n",
       "       \"personality\": \"Wisdom-imbued, compassionate, and fiercely protective, with a deep understanding of the cosmos.\",\n",
       "       \"physical_appearance\": \"Ethereal, with an luminous, star-like appearance, embodying the celestial essence of the Nexus.\",\n",
       "       \"role\": \"Supporting Character/Guardian\",\n",
       "       \"gender\": \"Non-Applicable\",\n",
       "       \"race\": \"Non-Human (Celestial Entity)\",\n",
       "       \"age\": \"Ageless\",\n",
       "       \"catch_phrase\": \"Balance is key; harmony is our goal.\",\n",
       "       \"voice_description\": \"Soothing, melodious, and filled with ancient wisdom, conveying a sense of reassurance.\",\n",
       "       \"props\": [\n",
       "         \"Stellar Energy Manipulation\"\n",
       "       ],\n",
       "       \"relationships\": [\n",
       "         {\n",
       "           \"character_name\": \"Meera 'Midnight' Singh\",\n",
       "           \"relationship_type\": \"Mentor-Protege\"\n",
       "         }\n",
       "-      ]\n",
       "+      ],\n",
       "+      \"image_prompt\": \"Vibrant, Neon-Lit Cyberpunk Manga Illustration - The Nexus Guardian (Astrum) amidst a celestial backdrop. Ethereal, with a **luminous, star-like appearance** embodying the celestial essence of the Nexus; **soft, ethereal glow** surrounds their form.\",\n",
       "+      \"image_prompt_short\": \"Vibrant Cyberpunk Manga - The Nexus Guardian (Astrum), ethereal, luminous star-like form, soft ethereal glow.\"\n",
       "     },\n",
       "     {\n",
       "       \"nickname\": \"earth_rebel\",\n",
       "       \"name\": \"Kaito 'Sparrow' Hernandez\",\n",
       "       \"description\": \"A charismatic leader of an underground rebellion on Earth, fighting against the oppressive forces exploiting Eon.\",\n",
       "       \"personality\": \"Passionate, resourceful, and fearless, with a strong sense of justice.\",\n",
       "       \"physical_appearance\": \"Lean, with short, spiky brown hair, bright hazel eyes, and a collection of scars from past encounters. Often wears practical, makeshift rebel gear.\",\n",
       "       \"role\": \"Supporting Character/Rebel Leader\",\n",
       "       \"gender\": \"Male\",\n",
       "       \"race\": \"Latin American\",\n",
       "       \"age\": \"22\",\n",
       "       \"catch_phrase\": \"Together, we rise; together, we overcome!\",\n",
       "       \"voice_description\": \"Energetic, persuasive, and determined, with a hint of vulnerability.\",\n",
       "       \"props\": [\n",
       "         \"Customized Rebel Comms Device\",\n",
       "         \"Graffiti Tagging Gear\"\n",
       "       ],\n",
       "       \"relationships\": [\n",
       "         {\n",
       "           \"character_name\": \"Meera 'Midnight' Singh\",\n",
       "           \"relationship_type\": \"Unlikely Ally\"\n",
       "         }\n",
       "-      ]\n",
       "+      ],\n",
       "+      \"image_prompt\": \"Vibrant, Neon-Lit Cyberpunk Manga Illustration - Kaito 'Sparrow' Hernandez in a rebellious pose. Lean, with **short, spiky brown hair**, bright **hazel eyes**, and a determined expression. Wears practical, **makeshift rebel gear** with visible **graffiti tags**; holds a **Customized Rebel Comms Device**.\",\n",
       "+      \"image_prompt_short\": \"Vibrant Cyberpunk Manga - Kaito 'Sparrow' Hernandez, lean, spiky brown hair, bright hazel eyes, makeshift rebel gear, graffiti tags.\"\n",
       "     },\n",
       "     {\n",
       "       \"nickname\": \"corporate_exec\",\n",
       "       \"name\": \"Regina 'The Shark' Thornton\",\n",
       "       \"description\": \"A ruthless, cunning executive exploiting Eon for corporate gain, serving as a primary antagonist on Earth.\",\n",
       "       \"personality\": \"Ambitious, manipulative, and devoid of empathy, with a singular focus on profit.\",\n",
       "       \"physical_appearance\": \"Polished, with long, curly blonde hair, piercing blue eyes, and impeccable, high-end business attire.\",\n",
       "       \"role\": \"Antagonist/Corporate Executive\",\n",
       "       \"gender\": \"Female\",\n",
       "       \"race\": \"Caucasian\",\n",
       "       \"age\": \"35\",\n",
       "       \"catch_phrase\": \"In the game of power, only the strongest thrive.\",\n",
       "       \"voice_description\": \"Slick, persuasive, and menacing, with an undercurrent of superiority.\",\n",
       "       \"props\": [\n",
       "         \"Luxury Smartwatch\",\n",
       "         \"High-Security Briefcase\"\n",
       "       ],\n",
       "       \"relationships\": [\n",
       "         {\n",
       "           \"character_name\": \"Meera 'Midnight' Singh\",\n",
       "           \"relationship_type\": \"Adversarial\"\n",
       "         }\n",
       "-      ]\n",
       "+      ],\n",
       "+      \"image_prompt\": \"Vibrant, Neon-Lit Cyberpunk Manga Illustration - Regina 'The Shark' Thornton in a commanding stance. Polished, with **long, curly blonde hair**, piercing **blue eyes**, and impeccable, **high-end business attire** incorporating **subtle neon accents** symbolizing corporate power.\",\n",
       "+      \"image_prompt_short\": \"Vibrant Cyberpunk Manga - Regina 'The Shark' Thornton, polished, curly blonde hair, piercing blue eyes, high-end business attire with subtle neon accents.\"\n",
       "     }\n",
       "   ]\n",
       " }\n",
       "```"
      ],
      "text/plain": [
       "<IPython.core.display.Markdown object>"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    }
   ],
   "source": [
    "# Merge the stories together to update the original\n",
    "story_with_image_prompts = merge_models(story_with_characters, image_prompts)\n",
    "show_diff(story_with_characters, story_with_image_prompts)"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "# Save the Story\n",
    "Let's keep our progress so far."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 14,
   "metadata": {},
   "outputs": [],
   "source": [
    "story_with_image_prompts.save_to_directory(settings.STORY_DIR + \"/step_3\")"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "# Next Step\n",
    "Onto [Step 4: Scene Descriptions](./4_scene_descriptions.ipynb)"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "plotomatic",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.12.7"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}