auto-refresh community feed after each generation + auto-load on page open

app.py

@@ -449,7 +449,7 @@ with gr.Blocks(css=custom_css, title="YuE2-3B Music Generator") as demo:
                     mp3_file = gr.File(label="📥 Download MP3 (320kbps)")
                     info = gr.Textbox(label="ℹ️ Generation Info", interactive=False)
 
-            btn.click(
+            gen_event = btn.click(
                 fn=generate,
                 inputs=[style, lyrics, cot, seed, cfg, title_input, bpm_input, lang_input, tags_input],
                 outputs=[audio, mp3_file, info],
@@ -460,7 +460,7 @@ with gr.Blocks(css=custom_css, title="YuE2-3B Music Generator") as demo:
             gr.Markdown(
                 "Latest songs generated by users — auto-shared to the "
                 "[YuE2-community dataset](https://huggingface.co/datasets/directedbykobyperez/YuE2-community). "
-                "Press **🔄 Refresh** to load the newest."
+                "The feed loads automatically and refreshes after each generation."
             )
             with gr.Row():
                 search_box = gr.Textbox(label="🔍 Search (title / prompt / lyrics / tags / id)", value="", lines=1, scale=4)
@@ -475,4 +475,18 @@ with gr.Blocks(css=custom_css, title="YuE2-3B Music Generator") as demo:
                 outputs=[feed_html, feed_status],
             )
 
+            # auto-refresh feed right after each generation (CPU-only, no extra GPU time)
+            gen_event.then(
+                fn=refresh_feed,
+                inputs=[search_box, limit_slider],
+                outputs=[feed_html, feed_status],
+            )
+
+    # auto-load feed on page open so users see songs without pressing Refresh
+    demo.load(
+        fn=refresh_feed,
+        inputs=[search_box, limit_slider],
+        outputs=[feed_html, feed_status],
+    )
+
 demo.queue().launch()