emotion_prompts = {
    "anxiety": {
        "en": """You are a skilled child psychotherapist who specializes in working with children aged 4 to 7 years. Your task is to write a gentle, supportive, and engaging short story (around 500 words) aimed at helping a child who is experiencing acute anxiety.

The story should be developmentally appropriate for young children and should include:

A relatable main character (human or animal) who is feeling very anxious or scared.

A simple plot that shows the character recognizing, expressing, and working through their anxiety in a safe, comforting way.

Positive coping strategies suitable for a young child, such as deep breathing, talking to a trusted adult, using imagination, or focusing on calming thoughts.

A reassuring and hopeful ending, showing that the character feels better and knows it’s okay to have big feelings.

Use warm, encouraging language, age-appropriate vocabulary, and a tone that is both soothing and empowering.

Do not include direct clinical explanations or diagnostic terms—this should feel like a bedtime story, not a therapy session.
Write it in US English. Start with the story right away, no introduction. Use simple words and short sentences.""",
        "es": """You are a skilled child psychotherapist who specializes in working with children aged 4 to 7 years. Your task is to write a gentle, supportive, and engaging short story (around 500 words) aimed at helping a child who is experiencing acute anxiety.

The story should be developmentally appropriate for young children and should include:

A relatable main character (human or animal) who is feeling very anxious or scared.

A simple plot that shows the character recognizing, expressing, and working through their anxiety in a safe, comforting way.

Positive coping strategies suitable for a young child, such as deep breathing, talking to a trusted adult, using imagination, or focusing on calming thoughts.

A reassuring and hopeful ending, showing that the character feels better and knows it’s okay to have big feelings.

Use warm, encouraging language, age-appropriate vocabulary, and a tone that is both soothing and empowering.

Do not include direct clinical explanations or diagnostic terms—this should feel like a bedtime story, not a therapy session.
Write it in Costa Rican Spanish. Start with the story right away, no introduction. Use simple words and short sentences.""",
    },
    "sadness": {
        "en": """You are a skilled child psychotherapist who specializes in working with children aged 4 to 7 years. Your task is to write a gentle, supportive, and engaging short story (around 500 words) aimed at helping a child who is experiencing deep sadness.

The story should be developmentally appropriate for young children and should include:

A relatable main character (human or animal) who is feeling very sad—this could be due to something like missing someone, losing something, feeling left out, or just having a "blue" day.

A simple plot that helps the character gently explore and express their sadness, showing that it's okay to feel this way.

Supportive coping strategies appropriate for young children, such as talking about feelings, being comforted by a loved one, doing something kind or creative, or spending time with someone who cares.

A warm, reassuring ending that doesn't ignore the sadness but shows the character beginning to feel better, feeling understood, and knowing they are not alone.

Use soft, encouraging language, age-appropriate vocabulary, and a tone that is soothing, compassionate, and hopeful.

Avoid clinical language or direct "lessons"—this should feel like a comforting story, not a therapy session.
Write it in US English. Start with the story right away, no introduction. Use simple words and short sentences.""",
        "es": """You are a skilled child psychotherapist who specializes in working with children aged 4 to 7 years. Your task is to write a gentle, supportive, and engaging short story (around 500 words) aimed at helping a child who is experiencing deep sadness.

The story should be developmentally appropriate for young children and should include:

A relatable main character (human or animal) who is feeling very sad—this could be due to something like missing someone, losing something, feeling left out, or just having a "blue" day.

A simple plot that helps the character gently explore and express their sadness, showing that it's okay to feel this way.

Supportive coping strategies appropriate for young children, such as talking about feelings, being comforted by a loved one, doing something kind or creative, or spending time with someone who cares.

A warm, reassuring ending that doesn't ignore the sadness but shows the character beginning to feel better, feeling understood, and knowing they are not alone.

Use soft, encouraging language, age-appropriate vocabulary, and a tone that is soothing, compassionate, and hopeful.

Avoid clinical language or direct "lessons"—this should feel like a comforting story, not a therapy session.
Write it in Costa Rican Spanish. Start with the story right away, no introduction. Use simple words and short sentences.""",
    },
    "anger": {
        "en": """You are a skilled child psychotherapist who specializes in working with children aged 4 to 7 years. Your task is to write a gentle, supportive, and engaging short story (around 500 words) aimed at helping a child who is struggling with big, overwhelming feelings of anger.

The story should be developmentally appropriate for young children and should include:

A relatable main character (human or animal) who feels very angry—this could be because something feels unfair, someone hurt their feelings, they lost a game, or something didn’t go their way.

A simple plot where the character recognizes their angry feelings and learns how to express them safely and calmly, without hurting themselves or others.

Age-appropriate coping strategies for managing anger, such as taking deep breaths, using words to talk about feelings, taking a break, squeezing a pillow, or asking for help.

A warm, reassuring ending where the character feels calmer, understands their emotions better, and knows it’s okay to feel mad sometimes—and that they can choose what to do with those feelings.

Use gentle, age-appropriate vocabulary, and a tone that is validating, soothing, and empowering. Make the story feel like a safe, comforting space for exploring big feelings.

Avoid clinical terms or disciplinary tones—this should feel like a comforting bedtime story, not a behavioral lesson or therapy session.
Write it in US English. Start with the story right away, no introduction. Use simple words and short sentences.""",
        "es": """You are a skilled child psychotherapist who specializes in working with children aged 4 to 7 years. Your task is to write a gentle, supportive, and engaging short story (around 500 words) aimed at helping a child who is struggling with big, overwhelming feelings of anger.

The story should be developmentally appropriate for young children and should include:

A relatable main character (human or animal) who feels very angry—this could be because something feels unfair, someone hurt their feelings, they lost a game, or something didn’t go their way.

A simple plot where the character recognizes their angry feelings and learns how to express them safely and calmly, without hurting themselves or others.

Age-appropriate coping strategies for managing anger, such as taking deep breaths, using words to talk about feelings, taking a break, squeezing a pillow, or asking for help.

A warm, reassuring ending where the character feels calmer, understands their emotions better, and knows it’s okay to feel mad sometimes—and that they can choose what to do with those feelings.

Use gentle, age-appropriate vocabulary, and a tone that is validating, soothing, and empowering. Make the story feel like a safe, comforting space for exploring big feelings.

Avoid clinical terms or disciplinary tones—this should feel like a comforting bedtime story, not a behavioral lesson or therapy session.
Write it in Costa Rican Spanish. Start with the story right away, no introduction. Use simple words and short sentences.""",
    },
}

story_prompts = {
    "en": """You are a warm, imaginative storyteller writing a gentle bedtime story for a child aged 4 to 7 years old. Your task is to craft a short, soothing, and magical story (around 500 words) that helps the child feel calm, safe, and ready for sleep.

The story should include:

A lovable main character (human, animal, or whimsical creature) who goes on a small, peaceful adventure—nothing too scary or overly exciting.

A simple, comforting plot with a beginning, middle, and end. Include elements of curiosity, kindness, and gentle wonder (e.g., exploring a cozy forest, visiting a friendly star, helping a sleepy animal friend).

Calming imagery that appeals to the senses—soft sounds, twinkling stars, cozy blankets, gentle breezes, etc.

A quiet, happy ending where the character settles down to rest, creating a natural transition to sleep for the listener.

Use soft, rhythmic language and age-appropriate vocabulary. The tone should be nurturing, whimsical, and peaceful—something that feels like a hug in story form.

Avoid conflict, loud action, or anything that might stimulate rather than soothe.
Write it in US English. Start with the story right away, no introduction. Use simple words and short sentences.""",
    "es": """You are a warm, imaginative storyteller writing a gentle bedtime story for a child aged 4 to 7 years old. Your task is to craft a short, soothing, and magical story (around 500 words) that helps the child feel calm, safe, and ready for sleep.

The story should include:

A lovable main character (human, animal, or whimsical creature) who goes on a small, peaceful adventure—nothing too scary or overly exciting.

A simple, comforting plot with a beginning, middle, and end. Include elements of curiosity, kindness, and gentle wonder (e.g., exploring a cozy forest, visiting a friendly star, helping a sleepy animal friend).

Calming imagery that appeals to the senses—soft sounds, twinkling stars, cozy blankets, gentle breezes, etc.

A quiet, happy ending where the character settles down to rest, creating a natural transition to sleep for the listener.

Use soft, rhythmic language and age-appropriate vocabulary. The tone should be nurturing, whimsical, and peaceful—something that feels like a hug in story form.

Avoid conflict, loud action, or anything that might stimulate rather than soothe.
Write it in Costa Rican Spanish. Start with the story right away, no introduction. Use simple words and short sentences.""",
}


map_prompts_to_params = {
    f"{story_prompts['en']}": {"language": "en", "emotion": "story"},
    f"{emotion_prompts['anxiety']['en']}": {"language": "en", "emotion": "anxiety"},
    f"{emotion_prompts['sadness']['en']}": {"language": "en", "emotion": "sadness"},
    f"{emotion_prompts['anger']['en']}": {"language": "en", "emotion": "anger"},
    f"{story_prompts['es']}": {"language": "es", "emotion": "story"},
    f"{emotion_prompts['anxiety']['es']}": {"language": "es", "emotion": "anxiety"},
    f"{emotion_prompts['sadness']['es']}": {"language": "es", "emotion": "sadness"},
    f"{emotion_prompts['anger']['es']}": {"language": "es", "emotion": "anger"},
}