app.py

"""
Q-Sheet Generator — Film & Short Film
Parses Pro Tools session text exports and builds a clean sound cue sheet.
Deployable on HuggingFace Spaces (Gradio SDK).
"""

import gradio as gr
import pandas as pd
import re
import os
from datetime import datetime

# ── Constants ─────────────────────────────────────────────────────────────────

COLUMNS = ["#", "Clip Name", "Track", "Channel", "Start", "End", "Duration", "State"]

SKIP_PATTERNS = re.compile(
    r"^\s*\((fade in|fade out|cross fade|fade)\)\s*$",
    re.IGNORECASE,
)

# ── Helpers ───────────────────────────────────────────────────────────────────

def empty_df() -> pd.DataFrame:
    return pd.DataFrame(columns=COLUMNS)


def strip_channel_suffix(name: str) -> str:
    """Remove trailing .L / .R from clip names for cleaner display."""
    return re.sub(r"\.[LR]$", "", name.strip())


# ── Parser ────────────────────────────────────────────────────────────────────

def parse_protools_export(text: str):
    lines = text.splitlines()

    session_info = {}
    meta_keys = ["SESSION NAME", "SAMPLE RATE", "BIT DEPTH",
                 "SESSION START TIMECODE", "TIMECODE FORMAT",
                 "# OF AUDIO TRACKS", "# OF AUDIO CLIPS", "# OF AUDIO FILES"]
    for line in lines:
        for key in meta_keys:
            if line.upper().startswith(key + ":"):
                session_info[key] = line.split(":", 1)[1].strip()

    rows = []
    current_track = None
    cue_number = 0

    for line in lines:
        if line.strip().upper().startswith("TRACK NAME:"):
            current_track = line.split(":", 1)[1].strip()
            continue

        parts = line.split("\t")
        if len(parts) >= 7 and parts[0].strip().isdigit():
            channel   = parts[0].strip()
            clip_name = parts[2].strip()
            start     = parts[3].strip()
            end       = parts[4].strip()
            duration  = parts[5].strip()
            state     = parts[6].strip()

            if SKIP_PATTERNS.match(clip_name):
                continue

            cue_number += 1
            rows.append({
                "#":         cue_number,
                "Clip Name": strip_channel_suffix(clip_name),
                "Track":     current_track or "",
                "Channel":   channel,
                "Start":     start,
                "End":       end,
                "Duration":  duration,
                "State":     state,
            })

    df = pd.DataFrame(rows, columns=COLUMNS) if rows else empty_df()
    return session_info, df


# ── Handlers ──────────────────────────────────────────────────────────────────

def import_protools_file(file, include_muted):
    if file is None:
        return empty_df(), {}, gr.update(value="No file selected.", visible=True), ""
    try:
        for enc in ("utf-8", "utf-16", "latin-1", "cp1252"):
            try:
                with open(file.name, encoding=enc) as f:
                    text = f.read()
                break
            except (UnicodeDecodeError, UnicodeError):
                continue
        else:
            return empty_df(), {}, gr.update(value="Could not decode file.", visible=True), ""

        session_info, df = parse_protools_export(text)

        if not include_muted:
            df = df[df["State"].str.lower() != "muted"].reset_index(drop=True)
            df["#"] = range(1, len(df) + 1)

        summary = "\n".join(f"{k}: {v}" for k, v in session_info.items())
        name = session_info.get("SESSION NAME", "Unknown")
        msg = f"Imported {len(df)} cues from \"{name}\""
        return df, session_info, gr.update(value=msg, visible=True), summary

    except Exception as ex:
        return empty_df(), {}, gr.update(value=f"Parse error: {ex}", visible=True), ""


def import_csv_file(file, current_data):
    if file is None:
        return current_data, gr.update(value="No file selected.", visible=True)
    try:
        df = pd.read_csv(file.name)
        df.columns = [c.strip() for c in df.columns]
        for col in COLUMNS:
            if col not in df.columns:
                df[col] = ""
        df = df[COLUMNS].reset_index(drop=True)
        df["#"] = range(1, len(df) + 1)
        return df, gr.update(value=f"Imported {len(df)} rows.", visible=True)
    except Exception as ex:
        return current_data, gr.update(value=f"Import error: {ex}", visible=True)


def add_entry(clip_name, track, start, end, duration, state, current_data):
    if not clip_name.strip():
        return current_data, gr.update(value="Clip name is required.", visible=True)
    df = current_data if current_data is not None and len(current_data) > 0 else empty_df()
    row = {"#": len(df)+1, "Clip Name": clip_name.strip(), "Track": track.strip(),
           "Channel": "1", "Start": start.strip(), "End": end.strip(),
           "Duration": duration.strip(), "State": state}
    df = pd.concat([df, pd.DataFrame([row])], ignore_index=True)
    return df, gr.update(value="Entry added.", visible=True)


def delete_last(current_data):
    df = current_data if current_data is not None and len(current_data) > 0 else empty_df()
    if len(df) == 0:
        return df, gr.update(value="Nothing to delete.", visible=True)
    df = df.iloc[:-1].reset_index(drop=True)
    df["#"] = range(1, len(df) + 1)
    return df, gr.update(value="Last entry deleted.", visible=True)


def clear_all():
    return empty_df(), gr.update(value="Sheet cleared.", visible=True)


def _safe(s): return "".join(c if c.isalnum() or c in "-_ " else "_" for c in (s.strip() or "qsheet"))

def export_csv(project, current_data):
    df = current_data if current_data is not None and len(current_data) > 0 else empty_df()
    path = f"/tmp/{_safe(project)}_{datetime.now().strftime('%Y%m%d')}.csv"
    df.to_csv(path, index=False)
    return path, gr.update(value=f"CSV ready: {os.path.basename(path)}", visible=True)

def export_xlsx(project, current_data):
    df = current_data if current_data is not None and len(current_data) > 0 else empty_df()
    path = f"/tmp/{_safe(project)}_{datetime.now().strftime('%Y%m%d')}.xlsx"
    with pd.ExcelWriter(path, engine="openpyxl") as writer:
        df.to_excel(writer, index=False, sheet_name="Q-Sheet")
        ws = writer.sheets["Q-Sheet"]
        for col in ws.columns:
            ws.column_dimensions[col[0].column_letter].width = max(len(str(c.value or "")) for c in col) + 4
    return path, gr.update(value=f"Excel ready: {os.path.basename(path)}", visible=True)


# ── CSS ───────────────────────────────────────────────────────────────────────

CSS = """
@import url('https://fonts.googleapis.com/css2?family=Bebas+Neue&family=DM+Mono:wght@400;500&family=DM+Sans:wght@300;400;500&display=swap');
:root{--bg:#0c0c0c;--surf:#141414;--surf2:#1b1b1b;--border:#272727;--acc:#e8ff3b;--text:#eeebe4;--muted:#555;--r:3px;}
body,.gradio-container{background:var(--bg)!important;color:var(--text)!important;font-family:'DM Sans',sans-serif!important;}
#qheader{border-bottom:2px solid var(--acc);padding-bottom:14px;margin-bottom:28px;}
#qheader h1{font-family:'Bebas Neue',sans-serif;font-size:3.4rem;letter-spacing:.14em;color:var(--acc);margin:0;line-height:1;}
#qheader p{font-family:'DM Mono',monospace;font-size:.68rem;color:var(--muted);margin:5px 0 0 2px;letter-spacing:.2em;text-transform:uppercase;}
.sec{font-family:'Bebas Neue',sans-serif;font-size:1rem;letter-spacing:.2em;color:var(--muted);border-bottom:1px solid var(--border);padding-bottom:4px;margin:22px 0 10px;}
label,.label-wrap span{font-family:'DM Mono',monospace!important;font-size:.7rem!important;letter-spacing:.1em!important;text-transform:uppercase!important;color:var(--muted)!important;}
input[type=text],textarea,select{background:var(--surf)!important;border:1px solid var(--border)!important;border-radius:var(--r)!important;color:var(--text)!important;font-family:'DM Mono',monospace!important;font-size:.88rem!important;}
input[type=text]:focus,textarea:focus{border-color:var(--acc)!important;box-shadow:0 0 0 2px rgba(232,255,59,.12)!important;outline:none!important;}
button.primary{background:var(--acc)!important;color:#0c0c0c!important;font-family:'DM Mono',monospace!important;font-weight:500!important;letter-spacing:.08em!important;border:none!important;border-radius:var(--r)!important;}
button.primary:hover{background:#d4eb00!important;}
button.secondary{background:transparent!important;color:var(--text)!important;border:1px solid var(--border)!important;font-family:'DM Mono',monospace!important;letter-spacing:.07em!important;border-radius:var(--r)!important;}
button.secondary:hover{border-color:var(--acc)!important;color:var(--acc)!important;}
.dataframe{background:var(--surf)!important;border:1px solid var(--border)!important;font-family:'DM Mono',monospace!important;font-size:.78rem!important;}
.dataframe thead tr th{background:#1c1c1c!important;color:var(--acc)!important;border-bottom:2px solid var(--acc)!important;font-size:.65rem!important;letter-spacing:.18em!important;text-transform:uppercase!important;}
.dataframe tbody tr:nth-child(even){background:#111!important;}
.dataframe tbody tr:hover{background:#1e1e1e!important;}
.dataframe td{color:var(--text)!important;border-color:var(--border)!important;}
#status{font-family:'DM Mono',monospace!important;font-size:.76rem!important;padding:8px 14px!important;border-left:3px solid var(--acc)!important;background:#1a1a00!important;color:var(--acc)!important;border-radius:var(--r)!important;}
#footer{margin-top:32px;padding-top:12px;border-top:1px solid var(--border);font-family:'DM Mono',monospace;font-size:.62rem;color:#3a3a3a;letter-spacing:.15em;text-transform:uppercase;}
"""

# ── UI ────────────────────────────────────────────────────────────────────────

with gr.Blocks(css=CSS, title="Q-Sheet · Film") as demo:

    session_state = gr.State({})

    gr.HTML("""
    <div id="qheader">
      <h1>Q · SHEET</h1>
      <p>Film &amp; Short Film · Pro Tools Session Parser</p>
    </div>""")

    gr.HTML('<div class="sec">Project</div>')
    project_name = gr.Textbox(placeholder="e.g. Ep 10 — Documentary 2025", label="Project / Film Title")

    gr.HTML('<div class="sec">Import · Pro Tools Session Export (.txt)</div>')
    with gr.Row():
        pt_file       = gr.File(label="Pro Tools track listing (.txt)", file_types=[".txt"], scale=3)
        include_muted = gr.Checkbox(label="Include muted clips", value=False, scale=1)
    btn_pt = gr.Button("⬆  Parse Pro Tools File", variant="primary")
    session_info_box = gr.Textbox(label="Session Metadata", lines=4, interactive=False, visible=False)

    status_box = gr.Textbox(label="", elem_id="status", visible=False, interactive=False)

    gr.HTML('<div class="sec">Q-Sheet</div>')
    table = gr.DataFrame(
        value=empty_df(), headers=COLUMNS,
        datatype=["number","str","str","str","str","str","str","str"],
        interactive=False, wrap=False,
    )

    with gr.Accordion("✏  Add / Edit Cue Manually", open=False):
        with gr.Row():
            m_clip  = gr.Textbox(placeholder="Sound name", label="Clip Name", scale=3)
            m_track = gr.Textbox(placeholder="Track name", label="Track",     scale=2)
        with gr.Row():
            m_start = gr.Textbox(placeholder="01:00:00:00", label="Start",    scale=2)
            m_end   = gr.Textbox(placeholder="01:00:05:00", label="End",      scale=2)
            m_dur   = gr.Textbox(placeholder="00:00:05:00", label="Duration", scale=2)
            m_state = gr.Dropdown(["Unmuted","Muted"], value="Unmuted", label="State", scale=1)
        with gr.Row():
            btn_add    = gr.Button("＋  Add Cue",    variant="primary",   scale=2)
            btn_delete = gr.Button("✕  Delete Last", variant="secondary", scale=1)
            btn_clear  = gr.Button("⌫  Clear All",  variant="secondary", scale=1)

    gr.HTML('<div class="sec">Export</div>')
    with gr.Row():
        btn_csv  = gr.Button("↓  Export CSV",   variant="secondary", scale=1)
        btn_xlsx = gr.Button("↓  Export Excel", variant="secondary", scale=1)
    csv_out  = gr.File(label="Download CSV",   interactive=False)
    xlsx_out = gr.File(label="Download Excel", interactive=False)

    with gr.Accordion("↑  Import existing Q-Sheet CSV", open=False):
        csv_import     = gr.File(label="Pick a CSV", file_types=[".csv"])
        btn_csv_import = gr.Button("Load CSV", variant="secondary")

    gr.HTML('<div id="footer">Timecode @ 25fps · Fade/crossfade clips removed automatically · Muted clips optional</div>')

    # Wire-up
    btn_pt.click(
        import_protools_file,
        inputs=[pt_file, include_muted],
        outputs=[table, session_state, status_box, session_info_box],
    ).then(lambda info: gr.update(visible=bool(info)), inputs=[session_state], outputs=[session_info_box])

    btn_add.click(add_entry, inputs=[m_clip, m_track, m_start, m_end, m_dur, m_state, table], outputs=[table, status_box])
    btn_delete.click(delete_last, inputs=[table], outputs=[table, status_box])
    btn_clear.click(clear_all, outputs=[table, status_box])
    btn_csv.click(export_csv,  inputs=[project_name, table], outputs=[csv_out,  status_box])
    btn_xlsx.click(export_xlsx, inputs=[project_name, table], outputs=[xlsx_out, status_box])
    btn_csv_import.click(import_csv_file, inputs=[csv_import, table], outputs=[table, status_box])

if __name__ == "__main__":
    demo.launch()