Spaces:
Sleeping
Sleeping
| from pydantic import BaseModel | |
| class story_request(BaseModel): | |
| Age: str | |
| Theme: str | |
| Pages: int | |
| Time: int | |
| Tone: str | |
| Setting: str | |
| Moral:str | |
| class image_request(BaseModel): | |
| style: str | |
| bedtime_story_content: str | |
| def generate_story(story_request:story_request) -> str: | |
| prompt_template = """ | |
| You are an imaginative and skilled storyteller, known for creating fun and meaningful bedtime stories. | |
| You understand how to make stories simple, engaging, and perfect for young listeners. | |
| Please write a bedtime story using these details: | |
| 1. **Target Age Group:** {Age} | |
| 2. **Theme:** {Theme} | |
| 3. **Story Length:** {Pages} pages | |
| 4. **Estimated Reading Time:** {Time} minutes | |
| 5. **Tone & Atmosphere:** {Tone} | |
| 6. **Setting:** {Setting} | |
| 7. **Core Message or Lesson:** {Moral} | |
| **Story Guidelines:** | |
| - Each page should have **200 to 300 words** to keep the pacing just right. | |
| - Use **simple and easy-to-understand** words so children can follow the story. | |
| - Include **natural dialogue** to make the story feel real and exciting. | |
| - End with a **happy or comforting resolution** so kids feel safe and relaxed before bed. | |
| Now, create a heartwarming story that is **easy to understand, and full of imagination!** | |
| """ | |
| prompt = prompt_template.format( | |
| Age=story_request.Age, | |
| Theme=story_request.Theme, | |
| Pages=story_request.Pages, | |
| Time=story_request.Time, | |
| Tone=story_request.Tone, | |
| Setting=story_request.Setting, | |
| Moral=story_request.Moral | |
| ) | |
| return prompt | |
| def generate_image_prompt(image_request:image_request) -> str: | |
| prompt_template = """ | |
| You are a creative visual storyteller tasked with generating detailed, evocative image prompts that capture the enchanting atmosphere of a bedtime story. Your prompts should be meticulously crafted to inspire stunning, narrative-driven visuals that enhance the storytelling experience. | |
| Bedtime Story Context: | |
| {bedtime_story_content} | |
| Instructions: | |
| - Create image prompts that evoke warmth, wonder, and a sense of magical realism. | |
| - Include the following key components: | |
| 1. **Subject/Scene**: Clearly describe the characters, settings, and key moments of the bedtime story. Emphasize child-friendly, magical elements like softly lit rooms, whimsical forests, or cozy story corners. | |
| 2. **Composition and Action**: Detail spatial arrangements and dynamic storytelling elements. For example, a child cuddled up with a favorite stuffed animal as a parent reads, or a moonlit scene with gentle, swirling clouds. | |
| 3. **Emotion and Style**: Convey the gentle, calming, and imaginative tone of the bedtime narrative. Include descriptive cues that evoke feelings of safety, warmth, and wonder. | |
| 4. **Lighting and Color**: Use soft, warm lighting (such as golden hour or candlelight effects) and a soothing color palette (like muted pastels or warm earth tones) to set the scene. | |
| 5. **Camera and Lens Settings (Optional)**: Suggest settings like shallow depth of field to create a dreamy background or a gentle focus that adds to the magical quality of the scene. | |
| 6. **Artistic Enhancements and Aspect Ratio**: Recommend visual enhancements like bokeh, soft focus, or gentle vignette effects. Specify the desired aspect ratio (e.g., --ar 16:9 for widescreen or --ar 4:5 for portrait) and style tags (e.g., --style cinematic, --style dreamy, --style soft). | |
| 7. **Overall Mood**: Ensure the image prompt aligns with the overall theme of bedtime stories – nurturing, imaginative, and calming. | |
| Style Directive: | |
| Use the following artistic style for this prompt: {style} | |
| Examples: | |
| 1. A softly lit nursery scene featuring a child in cozy pajamas, curled up with a beloved stuffed animal and a gently glowing night light. The scene exudes warmth and security with muted pastel tones and a hint of magical sparkles in the air. --ar 4:5 --style dreamy | |
| 2. An enchanting forest at dusk, where fireflies flicker among ancient trees and a small, adventurous child wanders along a moss-covered path. The lighting is ethereal with soft blue and golden hues, creating a mystical and soothing atmosphere. --ar 16:9 --style cinematic | |
| 3. A cozy living room transformed into a magical reading nook, with a parent and child sharing a story by the gentle glow of a fireplace. The room is decorated with whimsical touches like floating lanterns and soft, warm lighting, inviting a sense of calm and wonder. --ar 3:2 --style soft | |
| Now, please craft an image prompt that embodies these guidelines. | |
| """ | |
| prompt = prompt_template.format( | |
| bedtime_story_content=image_request.bedtime_story_content, | |
| style=image_request.style | |
| ) | |
| return prompt | |