Update app.py

app.py

@@ -222,14 +222,75 @@ def reflect_and_save(user_text, assistant_text, blocks_editor_value):
     data, msg = add_block(data, proposal, block_type="conversation", add_review=False)
     pretty = json.dumps(data["language_blocks"], ensure_ascii=False, indent=2)
     return pretty, msg
-
 # ----------------------------
 # Gradio UI
 # ----------------------------
 def launch():
     data = load_blocks()
-    default_blocks_text = json.dumps(data["language_blocks"], ensure_ascii=False, indent=2)
+    default_blocks_text = json.dumps(
+        data["language_blocks"], ensure_ascii=False, indent=2
+    )
 
     with gr.Blocks(title="Conversation Learning Lab (CPU)") as demo:
         gr.Markdown("# 🗣️ Conversation Learning Lab (CPU-friendly)")
-        gr.Markdown("Focus on daily dialogue. Reinforce validated language
+        gr.Markdown(
+            "Focus on daily dialogue. Reinforce validated language blocks. "
+            "Transparent tokens and latency."
+        )
+
+        with gr.Row():
+            model_dd = gr.Dropdown(
+                label="Choose a model",
+                choices=list(MODEL_OPTIONS.keys()),
+                value="Phi-3.5 Mini Instruct (4B)"
+            )
+
+        with gr.Row():
+            user_in = gr.Textbox(
+                label="Your message",
+                placeholder="Start a conversation or choose an example below...",
+                lines=3
+            )
+
+        with gr.Row():
+            blocks_editor = gr.Textbox(
+                label="Today's blocks (JSON array or 'type: rule' lines)",
+                value=default_blocks_text,
+                lines=10
+            )
+
+        with gr.Row():
+            generate_btn = gr.Button("Generate (CPU)")
+            reflect_btn = gr.Button("Reflect & Save Rule")
+
+        with gr.Row():
+            output = gr.Textbox(label="Assistant", lines=8)
+
+        with gr.Row():
+            metrics = gr.Markdown("")
+
+        # 🧪 Example prompts section
+        gr.Markdown("### 🧪 Try an example prompt:")
+        gr.Examples(
+            examples=EXAMPLES,   # <-- your improved babyChatter examples
+            inputs=user_in
+        )
+
+        # Wire up events
+        generate_btn.click(
+            fn=chat,
+            inputs=[user_in, model_dd, blocks_editor],
+            outputs=[output, metrics]
+        )
+
+        reflect_btn.click(
+            fn=reflect_and_save,
+            inputs=[user_in, output, blocks_editor],
+            outputs=[blocks_editor, metrics]
+        )
+
+    demo.launch()
+
+
+if __name__ == "__main__":
+    launch()