Update app.py

app.py

@@ -631,6 +631,15 @@ def analyze_and_separate(file_input, url_input, stem_mode, manual_bpm):
 
         stems = map_stems(track_folder, stem_mode)
 
+        # --- UPDATE UI CHECKBOXES DYNAMICALLY ---
+        available_stems = list(stems.keys())
+        # Default checked = all stems for export
+        new_export_stems = gr.CheckboxGroup(choices=available_stems, value=available_stems)
+        # Default checked = all except vocals for loops
+        loop_defaults = [s for s in available_stems if s != "Vocals"]
+        new_loop_stems = gr.CheckboxGroup(choices=available_stems, value=loop_defaults)
+        # ----------------------------------------
+
         p_drums = str(stems["Drums"]) if "Drums" in stems and stems["Drums"].exists() else None
         p_bass = str(stems["Bass"]) if "Bass" in stems and stems["Bass"].exists() else None
         p_guitar = str(stems["Guitar"]) if "Guitar" in stems and stems["Guitar"].exists() else None
@@ -645,7 +654,11 @@ def analyze_and_separate(file_input, url_input, stem_mode, manual_bpm):
 
         p_vocals = str(stems["Vocals"]) if "Vocals" in stems and stems["Vocals"].exists() else None
 
-        return p_drums, p_bass, p_guitar, p_piano, p_other, p_vocals, bpm, str(track_folder), stem_mode
+        return (
+            p_drums, p_bass, p_guitar, p_piano, p_other, p_vocals, 
+            bpm, str(track_folder), stem_mode,
+            new_export_stems, new_loop_stems # Return the dynamic updates
+        )
 
     except Exception as e:
         raise gr.Error(f"Process Failed: {str(e)}")
@@ -1010,7 +1023,11 @@ with gr.Blocks(title="Night Pulse | Studio Pro") as app:
     btn_analyze.click(
         fn=analyze_and_separate,
         inputs=[input_file, input_url, stem_mode, manual_bpm],
-        outputs=[p_drums, p_bass, p_guitar, p_piano, p_other, p_vocals, stored_bpm, stored_folder, stored_mode],
+        outputs=[
+            p_drums, p_bass, p_guitar, p_piano, p_other, p_vocals, 
+            stored_bpm, stored_folder, stored_mode,
+            export_stems, loop_stems  # Targeted updates for checkboxes
+        ],
     )
 
     btn_package.click(