Update app.py

app.py

@@ -120,13 +120,11 @@ with gr.Blocks(title="EvoTransformer v2.1 – Compare Options and Learn") as dem
         outputs=[feedback_output]
     )
 
+    # === Retrain section with auto-refresh + spinner
     with gr.Row():
-        retrain_button = gr.Button("♻️ Retrain Evo")
+        retrain_button = gr.Button("♻️ Retrain Evo", variant="primary")
         retrain_status = gr.Textbox(label="Retrain Status")
 
-    retrain_button.click(fn=retrain_model, inputs=[], outputs=[retrain_status])
-
-    # === Accordion Visual Panels ===
     with gr.Accordion("🧬 EvoTransformer Architecture", open=False):
         arch_box = gr.Textbox(label="Model Configuration", value=get_architecture_summary(model), lines=5, interactive=False)
 
@@ -136,5 +134,12 @@ with gr.Blocks(title="EvoTransformer v2.1 – Compare Options and Learn") as dem
     with gr.Accordion("📊 User Feedback Summary", open=False):
         dashboard_img = gr.Image(value=update_dashboard_plot(), label="Solution Votes")
 
+    retrain_button.click(
+        fn=retrain_model,
+        inputs=[],
+        outputs=[arch_box, evo_plot, retrain_status],
+        show_progress=True
+    )
+
 if __name__ == "__main__":
     demo.launch(share=True)