Update app.py

app.py

@@ -232,19 +232,20 @@ def launch():
     )
 
     with gr.Blocks(title="Conversation Learning Lab (CPU)") as demo:
+        # Header
         gr.Markdown("# 🗣️ Conversation Learning Lab (CPU-friendly)")
         gr.Markdown(
             "Focus on daily dialogue. Reinforce validated language blocks. "
             "Transparent tokens and latency."
         )
 
+        # Model selector + input
         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",
@@ -252,29 +253,32 @@ def launch():
                 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
-            )
+        # Example prompts
+        gr.Markdown("### 🧪 Try an example prompt:")
+        gr.Examples(
+            examples=EXAMPLES,
+            inputs=user_in
+        )
 
+        # Generate button comes right after examples
         with gr.Row():
             generate_btn = gr.Button("Generate (CPU)")
-            reflect_btn = gr.Button("Reflect & Save Rule")
 
+        # Output + metrics
         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
+        # JSON blocks editor + Reflect button at the bottom
+        gr.Markdown("### 📋 Today's Blocks")
+        blocks_editor = gr.Textbox(
+            label="Editable rules (JSON array or 'type: rule' lines)",
+            value=default_blocks_text,
+            lines=10
         )
+        with gr.Row():
+            reflect_btn = gr.Button("Reflect & Save Rule")
 
         # Wire up events
         generate_btn.click(
@@ -282,7 +286,6 @@ def launch():
             inputs=[user_in, model_dd, blocks_editor],
             outputs=[output, metrics]
         )
-
         reflect_btn.click(
             fn=reflect_and_save,
             inputs=[user_in, output, blocks_editor],
@@ -291,6 +294,5 @@ def launch():
 
     demo.launch()
 
-
 if __name__ == "__main__":
-    launch()
+    launch()