up logic

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 import tempfile
 import os
+import re
 from typing import List, Optional, Tuple
 from pydub import AudioSegment
 from pydub.silence import detect_silence
@@ -26,6 +27,15 @@ def ms_to_hhmmss(ms: int) -> str:
     s = seconds % 60
     return f"{h:02d}:{m:02d}:{s:02d}" if h > 0 else f"{m:02d}:{s:02d}"
 
+def clean_title_from_filename(name: str) -> str:
+    # remove extension
+    base = os.path.splitext(os.path.basename(name))[0]
+    # remove leading track numbers and separators like "01. " / "1 - " / "01_" etc
+    base = re.sub(r"^\s*\d{1,3}[\.\-\)_\s]+", "", base)
+    # squeeze spaces
+    base = re.sub(r"\s{2,}", " ", base).strip()
+    return base
+
 def read_titles_from_text(titles_text: Optional[str], n_tracks: int) -> List[str]:
     if not titles_text:
         return [f"Track {i+1:02d}" for i in range(n_tracks)]
@@ -43,7 +53,7 @@ def detect_starts_by_silence(audio: AudioSegment, silence_thresh_db: float, min_
     )
     starts = [0]
     silent_regions.sort(key=lambda x: x[0])
-    for start_ms, end_ms in silent_regions:
+    for _, end_ms in silent_regions:
         if end_ms - starts[-1] > 1000:
             starts.append(end_ms)
     starts = sorted(set(starts))
@@ -59,7 +69,7 @@ def clamp_starts(starts: List[int], audio_len: int) -> List[int]:
     return clean
 
 # ---------- Core ----------
-def generate_timecodes(
+def generate_timecodes_single(
     audio_file,
     mode,
     manual_starts_text,
@@ -71,7 +81,6 @@ def generate_timecodes(
     if audio_file is None:
         return "Please upload an audio file.", None
 
-    # Load audio with pydub/ffmpeg
     audio = AudioSegment.from_file(audio_file.name)
     audio_len = len(audio)
 
@@ -91,13 +100,11 @@ def generate_timecodes(
     starts = clamp_starts(starts, audio_len)
     titles = read_titles_from_text(titles_text, len(starts))
 
-    # Compose output lines
     lines = []
     for i, start_ms in enumerate(starts):
         ts = ms_to_hhmmss(start_ms)
         lines.append(f"{ts} – {titles[i]}")
 
-    # Save to temp file
     tmpdir = tempfile.mkdtemp()
     out_path = os.path.join(tmpdir, "timecodes.txt")
     with open(out_path, "w", encoding="utf-8") as f:
@@ -105,44 +112,83 @@ def generate_timecodes(
 
     return "\n".join(lines), out_path
 
-# ---------- UI ----------
-with gr.Blocks(title="YouTube Timecode Generator | City Pop Edition") as demo:
-    gr.Markdown(
-        "## 🎧 YouTube Timecode Generator\n"
-        "Generate **YouTube-style timecodes** from a long audio file.\n"
-        "- **Auto mode**: detect track starts from silence (adjust thresholds).\n"
-        "- **Manual mode**: enter start times like `0, 215, 07:10, 12:35`.\n"
-        "- Paste a **title list** (one per line) to label tracks."
-    )
+def generate_timecodes_multi(files, titles_text, natural_sort=True) -> Tuple[str, str]:
+    if not files:
+        return "Please upload multiple audio files.", None
 
-    with gr.Row():
-        audio_file = gr.File(label="Upload audio (mp3/wav/flac)", file_types=[".mp3", ".wav", ".flac", ".m4a"])
-        mode = gr.Radio(choices=["Auto (detect silence)", "Manual (comma-separated starts)"], value="Auto (detect silence)", label="Mode")
+    # Sort files by filename for consistent album order
+    file_objs = sorted(files, key=lambda f: f.name if natural_sort else files.index(f))
 
-    with gr.Row():
-        manual_starts_text = gr.Textbox(label="Manual starts (seconds or mm:ss or hh:mm:ss, comma-separated)", placeholder="0, 215, 07:10, 12:35", lines=2)
-        titles_text = gr.Textbox(label="Titles (one per line, optional)", placeholder="Overture\nShadow Love\nOffice Groove\n...")
+    # Calculate cumulative starts
+    starts_ms = []
+    titles = []
+    cursor = 0
+    for f in file_objs:
+        starts_ms.append(cursor)
+        titles.append(clean_title_from_filename(f.name))
+        seg = AudioSegment.from_file(f.name)
+        cursor += len(seg)
 
-    with gr.Accordion("Auto-detect parameters", open=False):
-        silence_thresh = gr.Slider(-60, 0, value=-30, step=1, label="Silence threshold (dBFS)")
-        min_silence_ms = gr.Slider(200, 4000, value=1500, step=50, label="Minimum silence length (ms)")
-        seek_step_ms = gr.Slider(1, 30, value=5, step=1, label="Seek step (ms)")
+    # If user provided custom titles, override
+    custom_titles = read_titles_from_text(titles_text, len(titles))
+    if titles_text:
+        titles = custom_titles
 
-    run_btn = gr.Button("Generate Timecodes", variant="primary")
-    output_text = gr.Textbox(label="Timecodes", lines=18)
-    download = gr.File(label="Download timecodes.txt")
+    lines = []
+    for ts_ms, title in zip(starts_ms, titles):
+        lines.append(f"{ms_to_hhmmss(ts_ms)} – {title}")
 
-    run_btn.click(
-        fn=generate_timecodes,
-        inputs=[audio_file, mode, manual_starts_text, titles_text, silence_thresh, min_silence_ms, seek_step_ms],
-        outputs=[output_text, download]
+    tmpdir = tempfile.mkdtemp()
+    out_path = os.path.join(tmpdir, "timecodes.txt")
+    with open(out_path, "w", encoding="utf-8") as f:
+        f.write("\n".join(lines))
+
+    return "\n".join(lines), out_path
+
+# ---------- UI ----------
+with gr.Blocks(title="YouTube Timecode Generator | Single or Multiple Audio") as demo:
+    gr.Markdown(
+        "## 🎧 YouTube Timecode Generator\n"
+        "Make YouTube-style **timecodes** from either **one long audio** or **multiple track files**.\n"
+        "- **Multiple Files**: upload each song file → timecodes use cumulative Durations.\n"
+        "- **Single File**: auto-detect with silence or enter manual start times.\n"
     )
 
+    with gr.Tabs():
+        with gr.Tab("Multiple Files (recommended)"):
+            multi_files = gr.Files(label="Upload multiple audio files (mp3/wav/flac/m4a)", file_types=[".mp3", ".wav", ".flac", ".m4a"])
+            multi_titles = gr.Textbox(label="Optional: Titles (one per line; leave blank to use file names)", lines=6, placeholder="Morning Light\nDaydream Train\n...")
+            gen_multi = gr.Button("Generate Timecodes", variant="primary")
+            out_multi = gr.Textbox(label="Timecodes", lines=18)
+            dl_multi = gr.File(label="Download timecodes.txt")
+            gen_multi.click(fn=generate_timecodes_multi, inputs=[multi_files, multi_titles], outputs=[out_multi, dl_multi])
+
+        with gr.Tab("Single File (auto/manual)"):
+            audio_file = gr.File(label="Upload audio (mp3/wav/flac/m4a)", file_types=[".mp3", ".wav", ".flac", ".m4a"])
+            mode = gr.Radio(choices=["Auto (detect silence)", "Manual (comma-separated starts)"], value="Auto (detect silence)", label="Mode")
+            manual_starts_text = gr.Textbox(label="Manual starts (seconds or mm:ss or hh:mm:ss, comma-separated)", placeholder="0, 215, 07:10, 12:35", lines=2)
+            titles_text = gr.Textbox(label="Titles (one per line, optional)", placeholder="Overture\nShadow Love\nOffice Groove\n...", lines=6)
+
+            with gr.Accordion("Auto-detect parameters", open=False):
+                silence_thresh = gr.Slider(-60, 0, value=-30, step=1, label="Silence threshold (dBFS)")
+                min_silence_ms = gr.Slider(200, 4000, value=1500, step=50, label="Minimum silence length (ms)")
+                seek_step_ms = gr.Slider(1, 30, value=5, step=1, label="Seek step (ms)")
+
+            gen_single = gr.Button("Generate Timecodes", variant="primary")
+            out_single = gr.Textbox(label="Timecodes", lines=18)
+            dl_single = gr.File(label="Download timecodes.txt")
+
+            gen_single.click(
+                fn=generate_timecodes_single,
+                inputs=[audio_file, mode, manual_starts_text, titles_text, silence_thresh, min_silence_ms, seek_step_ms],
+                outputs=[out_single, dl_single]
+            )
+
     gr.Markdown(
-        "Tips:\n"
-        "- Live/concert audio → naikkan `Minimum silence` & turunkan `Silence threshold` agar tidak sensitif.\n"
-        "- Studio mix yang keras → coba `silence-threshold -26` s.d. `-22`.\n"
-        "- Manual mode paling akurat untuk album tanpa jeda hening."
+        "### Tips\n"
+        "- **Multiple files** cocok untuk folder lagu terpisah (paling mudah & akurat).\n"
+        "- **Auto mode**: untuk satu file album; naikkan `Minimum silence` & turunkan `Silence threshold` bila live/crowded.\n"
+        "- Output sudah siap **copy–paste** ke deskripsi YouTube."
     )
 
 if __name__ == "__main__":