accept files until total audio cap reached, skip rest with warning

app.py

@@ -487,21 +487,26 @@ def gradio_main():
         _log(f"[INFO] Preparing {len(audio_files)} audio files...")
         yield _log_text(), gr.update(visible=False), gr.update(visible=True), gr.update()
 
+        import librosa as _lr
         total_dur = 0.0
+        accepted = 0
+        skipped = 0
         for f in audio_files:
             src = f.name if hasattr(f, "name") else str(f)
-            shutil.copy2(src, os.path.join(audio_dir, os.path.basename(src)))
             try:
-                import librosa
-                total_dur += librosa.get_duration(path=src)
+                dur = _lr.get_duration(path=src)
             except Exception:
-                pass
+                dur = 0.0
+            if total_dur + dur > MAX_TOTAL_AUDIO and accepted > 0:
+                skipped += 1
+                continue
+            shutil.copy2(src, os.path.join(audio_dir, os.path.basename(src)))
+            total_dur += dur
+            accepted += 1
 
-        if total_dur > MAX_TOTAL_AUDIO:
-            _log(f"[FAIL] Total audio {total_dur:.0f}s exceeds {MAX_TOTAL_AUDIO}s limit ({MAX_TOTAL_AUDIO/60:.0f} min)")
-            yield _log_text(), gr.update(visible=True), gr.update(visible=False), gr.update()
-            return
-        _log(f"[INFO] Total audio: {total_dur:.0f}s ({total_dur/60:.1f} min)")
+        if skipped > 0:
+            _log(f"[WARN] Skipped {skipped} files to stay under {MAX_TOTAL_AUDIO/60:.0f} min limit")
+        _log(f"[INFO] Total audio: {total_dur:.0f}s ({total_dur/60:.1f} min), {accepted} files")
 
         _log(f"[INFO] LoRA: '{lora_name}' | Files: {len(audio_files)} | "
              f"Epochs: {epochs} | LR: {lr} | Rank: {rank}")