Update app.py

app.py

@@ -28,7 +28,7 @@ except NameError:
     pass 
 
 def load_model():
-    print(">>> 📡 Connecting to Neural Engine...")
+    print(">>> Loading Engine...")
     try:
         checkpoint_path = hf_hub_download(
             repo_id="Tachyeon/IAM-RoFormer-Model-Weights",
@@ -98,159 +98,204 @@ def separate_audio(audio_path):
     return outputs
 
 # ==========================================
-# 2. SHARP & PUNCHY UI DESIGN (No Blur)
+# 2. PRO DAW INTERFACE STYLING
 # ==========================================
 css = """
-@import url('https://fonts.googleapis.com/css2?family=Teko:wght@300;500;700&family=Work+Sans:wght@400;600;800&display=swap');
+@import url('https://fonts.googleapis.com/css2?family=Inter:wght@400;600;800&display=swap');
 
 :root {
-    --bg-color: #050505;
-    --card-bg: #111111;
-    --border-color: #333333;
-    --highlight-cyan: #00F0FF;
-    --highlight-violet: #8F00FF;
-    --text-white: #FFFFFF;
-    --text-gray: #888888;
+    --bg-base: #121212;
+    --bg-panel: #1E1E1E;
+    --bg-input: #252525;
+    --border-dim: #333333;
+    --accent-blue: #3B82F6; /* Professional Studio Blue */
+    --accent-hover: #2563EB;
+    --text-main: #FFFFFF;
+    --text-muted: #A0A0A0;
 }
 
 body, .gradio-container {
-    background-color: var(--bg-color) !important;
-    color: var(--text-white) !important;
-    font-family: 'Work Sans', sans-serif !important;
+    background-color: var(--bg-base) !important;
+    color: var(--text-main) !important;
+    font-family: 'Inter', sans-serif !important;
     margin: 0; padding: 0;
 }
 
-/* CONTAINER RESET */
-.contain {
-    padding: 30px !important;
-    max-width: 1400px !important;
-    margin: 0 auto !important;
+/* APP CONTAINER */
+.app-wrapper {
+    max-width: 1600px;
+    margin: 0 auto;
+    padding: 20px;
+    height: 100vh;
+    display: flex;
+    flex-direction: column;
 }
 
-/* HEADER STYLE */
-.header-wrapper {
-    margin-bottom: 40px;
-    border-bottom: 2px solid var(--border-color);
-    padding-bottom: 20px;
+/* TOP NAVIGATION */
+.nav-bar {
+    display: flex;
+    justify-content: space-between;
+    align-items: center;
+    border-bottom: 1px solid var(--border-dim);
+    padding-bottom: 15px;
+    margin-bottom: 20px;
 }
-.title-main {
-    font-family: 'Teko', sans-serif;
-    font-size: 5rem; /* Massive Title */
-    line-height: 0.9;
-    font-weight: 700;
-    text-transform: uppercase;
-    color: white;
-    letter-spacing: 2px;
+.logo {
+    font-weight: 800;
+    font-size: 1.5rem;
+    letter-spacing: -0.5px;
+    color: var(--text-main);
 }
-.title-sub {
-    font-family: 'Work Sans', sans-serif;
-    color: var(--highlight-cyan);
-    font-size: 1.2rem;
+.badge {
+    font-size: 0.75rem;
+    background: #333;
+    padding: 4px 8px;
+    border-radius: 4px;
+    color: var(--text-muted);
     font-weight: 600;
-    letter-spacing: 1px;
-    text-transform: uppercase;
-    margin-top: 5px;
 }
 
-/* SHARP CARD PANELS (No Blur) */
-.sharp-card {
-    background-color: var(--card-bg);
-    border: 1px solid var(--border-color);
-    padding: 0px; /* Reset Padding */
-    margin-bottom: 0px;
+/* MAIN DASHBOARD GRID */
+.dashboard-grid {
+    display: flex;
+    gap: 20px;
+    height: 100%;
 }
 
-.card-header {
-    background-color: #1A1A1A;
-    padding: 15px 20px;
-    border-bottom: 1px solid var(--border-color);
-    font-family: 'Teko', sans-serif;
-    font-size: 1.8rem;
-    color: var(--text-gray);
+/* PANELS */
+.panel {
+    background: var(--bg-panel);
+    border: 1px solid var(--border-dim);
+    border-radius: 8px;
+    padding: 20px;
+    display: flex;
+    flex-direction: column;
+}
+
+.panel-label {
+    font-size: 0.75rem;
+    font-weight: 600;
+    color: var(--text-muted);
     text-transform: uppercase;
     letter-spacing: 1px;
-}
-.card-content {
-    padding: 25px;
+    margin-bottom: 15px;
+    border-bottom: 1px solid #2a2a2a;
+    padding-bottom: 8px;
 }
 
-/* BUTTON STYLING */
-#run-btn {
-    background: linear-gradient(90deg, var(--highlight-violet), var(--highlight-cyan)) !important;
-    color: #000 !important;
-    font-family: 'Work Sans', sans-serif !important;
-    font-weight: 800 !important;
-    font-size: 1.2rem !important;
-    text-transform: uppercase !important;
-    border: none !important;
-    border-radius: 0px !important; /* Sharp Edges */
-    padding: 20px !important;
-    transition: all 0.2s ease-in-out !important;
+/* LEFT SIDE: MASTER TRACK */
+.master-track {
+    flex: 1; /* 33% width */
 }
-#run-btn:hover {
-    filter: brightness(1.2);
-    transform: scale(1.01);
+
+/* RIGHT SIDE: MIXER */
+.mixer-rack {
+    flex: 2; /* 66% width */
 }
 
-/* AUDIO PLAYERS */
-.audio-player {
-    background: transparent !important;
-    border: 1px solid var(--border-color) !important;
-    border-radius: 0px !important;
-    margin-bottom: 15px !important;
+/* AUDIO COMPONENTS */
+.audio-comp {
+    background: var(--bg-input) !important;
+    border: none !important;
+    border-radius: 6px !important;
 }
 
-/* LABEL OVERRIDES */
-span.svelte-1gfkn6j {
-    color: var(--highlight-cyan) !important;
+/* BUTTON */
+#process-btn {
+    background: var(--accent-blue) !important;
+    color: white !important;
     font-weight: 600 !important;
-    text-transform: uppercase;
+    font-size: 0.9rem !important;
+    padding: 12px !important;
+    border-radius: 6px !important;
+    border: none !important;
+    margin-top: auto; /* Push to bottom */
+    transition: background 0.2s;
+}
+#process-btn:hover {
+    background: var(--accent-hover) !important;
+}
+
+/* GRID FOR STEMS */
+.stem-grid {
+    display: grid;
+    grid-template-columns: 1fr 1fr;
+    gap: 15px;
+    height: 100%;
+}
+.stem-slot {
+    background: var(--bg-input);
+    border-radius: 6px;
+    padding: 15px;
+    border: 1px solid var(--border-dim);
+}
+.stem-title {
     font-size: 0.8rem;
+    font-weight: 600;
+    color: var(--text-muted);
+    margin-bottom: 10px;
 }
 """
 
 with gr.Blocks(css=css, theme=gr.themes.Base()) as demo:
     
-    with gr.Column(elem_classes="contain"):
+    with gr.Column(elem_classes="app-wrapper"):
         
-        # 1. HEADER AREA
-        with gr.Group(elem_classes="header-wrapper"):
+        # 1. TOP BAR
+        with gr.Row(elem_classes="nav-bar"):
             gr.HTML("""
-            <div>
-                <div class="title-main">Swara Studio</div>
-                <div class="title-sub">Neural Audio Source Separation Engine</div>
+            <div style="width:100%; display:flex; justify-content:space-between; align-items:center;">
+                <div class="logo">SWARA STUDIO</div>
+                <div class="badge">PRO EDITION</div>
             </div>
             """)
 
-        # 2. MAIN GRID
-        with gr.Row(equal_height=True, variant="compact"):
+        # 2. MAIN CONSOLE
+        with gr.Row(elem_classes="dashboard-grid"):
             
-            # LEFT COLUMN: INPUT
-            with gr.Column(scale=1):
-                with gr.Group(elem_classes="sharp-card"):
-                    gr.HTML('<div class="card-header">Input Source</div>')
-                    with gr.Column(elem_classes="card-content"):
-                        input_audio = gr.Audio(
-                            label="MASTER AUDIO TRACK",
-                            type="filepath",
-                            elem_classes="audio-player"
-                        )
-                        run_btn = gr.Button("RUN SEPARATION PROCESS", elem_id="run-btn")
+            # LEFT: MASTER INPUT
+            with gr.Column(elem_classes="panel master-track"):
+                gr.HTML('<div class="panel-label">Master Input</div>')
+                
+                input_audio = gr.Audio(
+                    label="Source Audio",
+                    type="filepath",
+                    elem_classes="audio-comp"
+                )
+                
+                gr.Markdown(
+                    "Supports WAV, FLAC, MP3. \nProcessing time: ~10s per minute.",
+                    elem_id="info-text"
+                )
+                
+                process_btn = gr.Button("Separate Stems", elem_id="process-btn")
+
+            # RIGHT: STEM MIXER
+            with gr.Column(elem_classes="panel mixer-rack"):
+                gr.HTML('<div class="panel-label">Stem Mixer</div>')
+                
+                with gr.Row():
+                    # ROW 1
+                    with gr.Column(elem_classes="stem-slot"):
+                        gr.HTML('<div class="stem-title">VOCALS</div>')
+                        out_vocals = gr.Audio(label=None, interactive=False, show_label=False, elem_classes="audio-comp")
+                    
+                    with gr.Column(elem_classes="stem-slot"):
+                        gr.HTML('<div class="stem-title">DRUMS (MRIDANGAM)</div>')
+                        out_drums = gr.Audio(label=None, interactive=False, show_label=False, elem_classes="audio-comp")
+                
+                with gr.Row():
+                    # ROW 2
+                    with gr.Column(elem_classes="stem-slot"):
+                        gr.HTML('<div class="stem-title">BASS (TANPURA)</div>')
+                        out_bass = gr.Audio(label=None, interactive=False, show_label=False, elem_classes="audio-comp")
 
-            # RIGHT COLUMN: OUTPUT
-            with gr.Column(scale=2):
-                with gr.Group(elem_classes="sharp-card"):
-                    gr.HTML('<div class="card-header">Stem Output</div>')
-                    with gr.Column(elem_classes="card-content"):
-                        with gr.Row():
-                            out_vocals = gr.Audio(label="VOCALS", interactive=False, elem_classes="audio-player")
-                            out_drums = gr.Audio(label="DRUMS / MRIDANGAM", interactive=False, elem_classes="audio-player")
-                        with gr.Row():
-                            out_bass = gr.Audio(label="BASS / TANPURA", interactive=False, elem_classes="audio-player")
-                            out_other = gr.Audio(label="OTHER / VIOLIN", interactive=False, elem_classes="audio-player")
+                    with gr.Column(elem_classes="stem-slot"):
+                        gr.HTML('<div class="stem-title">OTHER (VIOLIN)</div>')
+                        out_other = gr.Audio(label=None, interactive=False, show_label=False, elem_classes="audio-comp")
 
     # LOGIC
-    run_btn.click(
+    process_btn.click(
         fn=separate_audio,
         inputs=[input_audio],
         outputs=[out_vocals, out_drums, out_bass, out_other]