Update app.py

app.py

@@ -120,19 +120,14 @@ with gr.Blocks(title="EvoTransformer v2.1 – Compare Options and Learn") as dem
         outputs=[feedback_output]
     )
 
-    # === Retrain section with auto-refresh + spinner
+    # === Define UI components to allow later refresh ===
+    retrain_status = gr.Textbox(label="Retrain Status")
+    arch_box = gr.Textbox(label="Model Configuration", value=get_architecture_summary(model), lines=5, interactive=False)
+    evo_plot = gr.Plot(label="Accuracy per Generation", value=evolution_accuracy_plot())
+    dashboard_img = gr.Image(value=update_dashboard_plot(), label="Solution Votes")
+
     with gr.Row():
         retrain_button = gr.Button("♻️ Retrain Evo", variant="primary")
-        retrain_status = gr.Textbox(label="Retrain Status")
-
-    with gr.Accordion("🧬 EvoTransformer Architecture", open=False):
-        arch_box = gr.Textbox(label="Model Configuration", value=get_architecture_summary(model), lines=5, interactive=False)
-
-    with gr.Accordion("📈 Evolution Accuracy Plot", open=False):
-        evo_plot = gr.Plot(label="Accuracy per Generation", value=evolution_accuracy_plot())
-
-    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,
@@ -141,5 +136,18 @@ with gr.Blocks(title="EvoTransformer v2.1 – Compare Options and Learn") as dem
         show_progress=True
     )
 
+    # === Render inside Accordions ===
+    with gr.Accordion("🧬 EvoTransformer Architecture", open=False):
+        arch_box.render()
+
+    with gr.Accordion("📈 Evolution Accuracy Plot", open=False):
+        evo_plot.render()
+
+    with gr.Accordion("📊 User Feedback Summary", open=False):
+        dashboard_img.render()
+
+    with gr.Accordion("🔁 Retrain Status", open=False):
+        retrain_status.render()
+
 if __name__ == "__main__":
     demo.launch(share=True)