Initial Kids Rhyme Studio setup

app.py

@@ -0,0 +1,112 @@
+"""Kids Rhyme Studio: simple, four-language Gradio app."""
+
+from audio_engine import make_sung_song
+from dataclasses import asdict
+import gradio as gr
+from rhyme_engine import parse_request, validate_lyric_for_audio, write_rhyme
+from themes import THEME_CHOICES
+
+
+CSS = """
+.gradio-container {max-width: 1100px !important; margin: auto !important;}
+#hero {background: linear-gradient(120deg,#fff0f7,#fff6d5 72%,#e7f5ff);
+       border: 1px solid #ffdfe9; border-radius: 24px; padding: 24px 30px;
+       margin: 12px 0 20px;}
+#hero h1 {font-size: clamp(2rem,5vw,3.5rem); margin: 0 0 8px; color:#42204d;}
+#hero p {font-size: 1.09rem; margin: 0; color:#584a61;}
+#instruction {margin: 2px 0 8px; color:#5d5061;}
+#make-rhyme, #make-audio {border-radius: 12px !important; font-weight: 700;}
+#lyrics textarea {line-height: 1.9; font-size: 1.06rem;}
+"""
+
+
+def create_rhyme(name, words, goal, language, mood, theme, use_ai):
+    try:
+        req = parse_request(name, words, goal, language, mood, theme)
+    except ValueError as exc:
+        raise gr.Error(str(exc)) from exc
+    rhyme, writer = write_rhyme(req, use_ai)
+    return rhyme, asdict(req), f"✨ {language} rhyme ready · {theme} ({writer}). You can edit it, then sing it.", None
+
+
+def create_audio(lyrics, settings, name, words, goal, language, mood, theme):
+    try:
+        if not settings:
+            raise ValueError("Write a rhyme first, then make its song.")
+        req = parse_request(name, words, goal, language, mood, theme)
+        if asdict(req) != settings:
+            raise ValueError("Your options changed. Click Make my rhyme again so the lyrics match your language and theme.")
+        safe_lyrics = validate_lyric_for_audio(lyrics)
+        result, status = make_sung_song(safe_lyrics, req.language, req.mood, req.theme, req.goal)
+        return result, f"🎤 {language} · {theme} — {status}"
+    except (ValueError, RuntimeError) as exc:
+        raise gr.Error(str(exc)) from exc
+
+
+with gr.Blocks(title="Kids Rhyme Studio") as demo:
+    gr.HTML("""
+      <section id="hero">
+        <h1>🎵 Kids Rhyme Studio</h1>
+        <p>Choose a theme, add favorite words, and make a little learning song.</p>
+      </section>
+    """)
+    gr.Markdown("**1. Add words → 2. Make a rhyme → 3. Sing and download**", elem_id="instruction")
+
+    settings = gr.State(value=None)
+    with gr.Row():
+        with gr.Column(scale=1):
+            name = gr.Textbox(label="Child's first name (optional)", placeholder="Maya", max_length=32)
+            words = gr.Textbox(label="Words to use (1–10)", placeholder="moon, bunny, stars, rainbow…",
+                               info="Separate up to 10 words with commas. In Telugu, try Telugu words.",
+                               max_length=300)
+            with gr.Row():
+                language = gr.Dropdown(["English", "Español", "తెలుగు", "Deutsch"],
+                                       value="English", label="Language")
+                mood = gr.Dropdown(["Bouncy", "Calm"], value="Bouncy", label="Music mood")
+            theme = gr.Dropdown(THEME_CHOICES, value="Good Habits", label="Song theme")
+            goal = gr.Textbox(label="Theme prompt (optional; required for Custom theme)",
+                              placeholder="Teach washing hands before meals and putting toys away",
+                              info="Use the chosen language for your words and prompt. Leave blank to use the theme's suggested lesson.",
+                              lines=2, max_length=160)
+            with gr.Accordion("Optional: AI writer", open=False):
+                use_ai = gr.Checkbox(
+                    label="Use AI for more varied lyrics",
+                    info="If enabled, the name, words, theme, and prompt are sent to a hosted model. A Space token must be configured.",
+                    value=False,
+                )
+            rhyme_button = gr.Button("✨ Make my rhyme", variant="primary", elem_id="make-rhyme")
+            gr.Examples(
+                examples=[
+                    ["Maya", "soap, toothbrush, toys", "", "English", "Bouncy", "Good Habits"],
+                    ["Maya", "moon, bunny, stars, rainbow, puddle, pencil, apple, rocket, kitten, garden",
+                     "", "English", "Bouncy", "Animals & Nature"],
+                    ["Sofía", "luna, conejo, estrellas", "", "Español", "Calm", "Bedtime"],
+                    ["అను", "చందమామ, బొమ్మ, నక్షత్రం", "", "తెలుగు", "Calm", "Bedtime"],
+                    ["Emil", "Apfel, Hase, Sterne", "", "Deutsch", "Bouncy", "Counting"],
+                    ["Maya", "bus, school, friends", "Feeling brave on the first day of school", "English", "Bouncy", "Custom theme"],
+                ],
+                inputs=[name, words, goal, language, mood, theme],
+                label="Try an example",
+            )
+        with gr.Column(scale=1):
+            lyrics = gr.Textbox(label="Your rhyme • you can edit it", lines=13, max_lines=15,
+                                buttons=["copy"], elem_id="lyrics")
+            audio_button = gr.Button("🎤 Sing my rhyme", elem_id="make-audio")
+            audio = gr.Audio(label="Sung song • listen or download WAV", interactive=False,
+                             buttons=["download"])
+            status = gr.Markdown("Your rhyme will appear here. Song creation uses a GPU and may take a minute.")
+
+    gr.Markdown(
+        "<small>Made for playful practice. A parent should review each rhyme. "
+        "The song model sings the rhyme; pronunciation and exact words can vary. "
+        "Use a nickname; this app keeps no account or rhyme database.</small>"
+    )
+
+    rhyme_button.click(create_rhyme, [name, words, goal, language, mood, theme, use_ai],
+                       [lyrics, settings, status, audio], concurrency_limit=2)
+    audio_button.click(create_audio, [lyrics, settings, name, words, goal, language, mood, theme],
+                       [audio, status], concurrency_limit=1)
+
+
+if __name__ == "__main__":
+    demo.queue().launch(theme=gr.themes.Soft(primary_hue="pink", secondary_hue="yellow"), css=CSS)