↑
Your storybook appears here
Add a drawing, pick a theme, and tap Make my book!
""",
)
coloring_display = gr.HTML(
visible=False,
elem_classes=["book-stage"],
)
# ---- BEHIND THE MAGIC ----
with gr.Accordion("Behind the magic ✨", open=False, elem_classes=["behind-magic"]):
with gr.Tabs():
with gr.Tab("Story"):
story_info = gr.JSON(label="Generated story structure")
with gr.Tab("Images"):
image_info = gr.Textbox(label="Illustration details", interactive=False, lines=5)
with gr.Tab("Models"):
gr.Markdown(
"""
| Model | Role | Size |
|---|---|---|
| **MiniCPM5-1B** | Story writer | 1B |
| **VoxCPM2** | Voice narrator | 2B |
| **FLUX.2-klein** | Illustrator | 4B |
The *brain* of DoodleBook — the story + the voice — is a **3B small-model stack**.
FLUX is the printer. **Tiny Titan.**
"""
)
with gr.Tab("Trace"):
trace_info = gr.Textbox(
label="Generation trace (Open Trace)",
interactive=False, lines=8, show_copy_button=True,
)
gr.HTML(
"""
"""
)
# ---- WIRING ----
if create_book_fn:
make_btn.click(
fn=create_book_fn,
inputs=[doodle, char_name, theme, hero_name, voice, make_coloring],
outputs=[book_display, status, audio_narration, pdf_download,
story_info, image_info, trace_info,
coloring_display, coloring_pdf_download],
)
if load_sample_fn:
demo.load(fn=load_sample_fn, outputs=[book_display])
return demo