Upload folder using huggingface_hub

README.md

@@ -1,13 +1,42 @@
 ---
 title: Bedtime Story Machine
-emoji: 📚
-colorFrom: red
-colorTo: red
+emoji: 🌙
+colorFrom: indigo
+colorTo: purple
 sdk: gradio
-sdk_version: 6.17.3
-python_version: '3.13'
+sdk_version: 5.29.0
 app_file: app.py
 pinned: false
+license: mit
+short_description: Personalized illustrated bedtime stories for kids
+tags:
+  - hackathon
+  - thousand-token-wood
+  - storytelling
+  - children
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+# 🌙 Bedtime Story Machine
+
+A personalized illustrated bedtime story generator, crafted just for your little one.
+
+## How it works
+
+1. Enter your child's name, age, and pick a theme
+2. AI writes a gentle 4-scene bedtime story with your child as the protagonist
+3. Each scene gets a beautiful watercolor-style illustration
+4. Read it together at bedtime! 🛏️
+
+## Models Used (≤32B total)
+
+- **Story Generation**: [Qwen/Qwen2.5-7B-Instruct](https://huggingface.co/Qwen/Qwen2.5-7B-Instruct) (7B params)
+- **Illustrations**: [black-forest-labs/FLUX.1-schnell](https://huggingface.co/black-forest-labs/FLUX.1-schnell) (12B params)
+- **Total**: ~19B parameters ✅
+
+## Track
+
+🍄 **Thousand Token Wood** — Build something delightful that wouldn't exist without AI.
+
+## Built for
+
+[Build Small Hackathon](https://huggingface.co/build-small-hackathon) by Gradio & Hugging Face

app.py

@@ -0,0 +1,164 @@
+import gradio as gr
+from huggingface_hub import InferenceClient
+import json
+import re
+import os
+from pathlib import Path
+from dotenv import load_dotenv
+
+# Load .env from parent dir or current dir
+load_dotenv(Path(__file__).resolve().parent.parent / ".env")
+load_dotenv()
+
+token = os.environ.get("HF_TOKEN") or os.environ.get("HF_API_KEY")
+
+text_client = InferenceClient(provider="together", token=token)
+image_client = InferenceClient(provider="hf-inference", token=token)
+
+STORY_MODEL = "Qwen/Qwen2.5-7B-Instruct"
+IMAGE_MODEL = "black-forest-labs/FLUX.1-schnell"
+
+
+def generate_story(name, age, theme, extra_details):
+    """Generate a 4-scene bedtime story with image prompts."""
+    age = int(age)
+    prompt = f"""Write a short bedtime story for a {age}-year-old child named {name}.
+Theme: {theme}
+{f"Extra details: {extra_details}" if extra_details else ""}
+
+Requirements:
+- Exactly 4 short scenes (3-4 sentences each)
+- Gentle, calming tone appropriate for bedtime
+- {name} is the main character
+- Happy, peaceful ending where {name} falls asleep or feels safe
+- Age-appropriate vocabulary for a {age}-year-old
+
+Return ONLY valid JSON in this exact format, no other text:
+{{"title": "Story Title", "scenes": [{{"text": "Scene text here", "image_prompt": "A children's book illustration of: brief visual description, soft watercolor style, warm colors, whimsical"}}]}}"""
+
+    response = text_client.chat_completion(
+        model=STORY_MODEL,
+        messages=[{"role": "user", "content": prompt}],
+        max_tokens=1024,
+        temperature=0.8,
+    )
+
+    content = response.choices[0].message.content.strip()
+    # Extract JSON from response
+    match = re.search(r'\{.*\}', content, re.DOTALL)
+    if match:
+        content = match.group()
+    return json.loads(content)
+
+
+def generate_illustration(prompt):
+    """Generate a single illustration."""
+    image = image_client.text_to_image(
+        prompt=prompt,
+        model=IMAGE_MODEL,
+        width=768,
+        height=512,
+    )
+    return image
+
+
+def create_story(name, age, theme, extra_details, progress=gr.Progress()):
+    """Main function: generate story then illustrations."""
+    if not name.strip():
+        raise gr.Error("Please enter the child's name!")
+    if not theme.strip():
+        raise gr.Error("Please choose a theme!")
+
+    progress(0.1, desc="✨ Writing your story...")
+    story = generate_story(name, age, theme, extra_details)
+
+    title = story["title"]
+    scenes = story["scenes"]
+    results = []
+
+    for i, scene in enumerate(scenes):
+        progress((0.2 + i * 0.2), desc=f"🎨 Painting scene {i+1} of {len(scenes)}...")
+        image = generate_illustration(scene["image_prompt"])
+        results.append((image, scene["text"]))
+
+    progress(1.0, desc="📖 Story complete!")
+    return title, results
+
+
+def build_output(name, age, theme, extra_details):
+    """Build the story and return components for display."""
+    title, scenes = create_story(name, age, theme, extra_details)
+
+    gallery_items = []
+    story_text = f"# 🌙 {title}\n\n"
+    images = []
+
+    for i, (image, text) in enumerate(scenes):
+        story_text += f"**Scene {i+1}:**\n{text}\n\n---\n\n"
+        gallery_items.append((image, f"Scene {i+1}"))
+        images.append(image)
+
+    return story_text, images[0] if images else None, images[1] if len(images) > 1 else None, images[2] if len(images) > 2 else None, images[3] if len(images) > 3 else None
+
+
+# --- UI ---
+theme = gr.themes.Soft(
+    primary_hue="purple",
+    secondary_hue="blue",
+)
+
+with gr.Blocks(theme=theme, title="🌙 Bedtime Story Machine") as demo:
+    gr.Markdown("""
+    # 🌙 Bedtime Story Machine
+    *A personalized illustrated bedtime story, crafted just for your little one*
+
+    Enter your child's details and watch as AI weaves a gentle tale with beautiful illustrations.
+    """)
+
+    with gr.Row():
+        with gr.Column(scale=1):
+            name_input = gr.Textbox(label="Child's Name", placeholder="e.g. Luna")
+            age_input = gr.Slider(label="Age", minimum=2, maximum=10, value=5, step=1)
+            theme_input = gr.Dropdown(
+                label="Story Theme",
+                choices=[
+                    "A magical forest adventure",
+                    "Friendly dragons and castles",
+                    "Under the sea with talking fish",
+                    "A trip to the moon and stars",
+                    "Friendly animals in a cozy barn",
+                    "A tiny fairy's garden party",
+                    "Flying on a cloud to dreamland",
+                ],
+                value="A magical forest adventure",
+                allow_custom_value=True,
+            )
+            extra_input = gr.Textbox(
+                label="Extra Details (optional)",
+                placeholder="e.g. loves dinosaurs, has a cat named Whiskers",
+            )
+            generate_btn = gr.Button("✨ Generate Bedtime Story", variant="primary", size="lg")
+
+        with gr.Column(scale=2):
+            story_output = gr.Markdown(label="Story")
+            with gr.Row():
+                img1 = gr.Image(label="Scene 1", show_label=True)
+                img2 = gr.Image(label="Scene 2", show_label=True)
+            with gr.Row():
+                img3 = gr.Image(label="Scene 3", show_label=True)
+                img4 = gr.Image(label="Scene 4", show_label=True)
+
+    generate_btn.click(
+        fn=build_output,
+        inputs=[name_input, age_input, theme_input, extra_input],
+        outputs=[story_output, img1, img2, img3, img4],
+    )
+
+    gr.Markdown("""
+    ---
+    *Built with 🤗 Hugging Face Inference API | Story: Qwen2.5-7B-Instruct | Art: FLUX.1-schnell*
+    *For the Build Small Hackathon — Thousand Token Wood track 🍄*
+    """)
+
+if __name__ == "__main__":
+    demo.launch()

requirements.txt

@@ -0,0 +1,3 @@
+gradio==5.29.0
+huggingface_hub==0.30.2
+python-dotenv==1.1.0