Update app.py

app.py

@@ -607,235 +607,235 @@ harmony_btn.click(
 )
 
 # --- Batch Processing ---
-    with gr.Tab("🔊 Batch Processing"):
-         batch_files = gr.File(label="Upload Multiple Files", file_count="multiple")
-         batch_effects = gr.CheckboxGroup(choices=preset_choices["Default"], label="Apply Effects in Order")
-         batch_isolate = gr.Checkbox(label="Isolate Vocals After Effects")
-         batch_preset_dropdown = gr.Dropdown(choices=preset_names, label="Select Preset", value=preset_names[0])
-         batch_export_format = gr.Dropdown(choices=["MP3", "WAV"], label="Export Format", value="MP3")
-         batch_process_btn = gr.Button("Process All Files")
-         batch_download = gr.File(label="Download ZIP of All Processed Files")
-         batch_status = gr.Textbox(label="Status", value="Ready", interactive=False)
-
-         batch_process_btn.click(
-         fn=batch_process_audio,
-         inputs=[batch_files, batch_effects, batch_isolate, batch_preset_dropdown, batch_export_format],
-         outputs=[batch_download, batch_status]
-        )   
-
-    # --- AI Auto-Tune ---
-    with gr.Tab("🎤 AI Auto-Tune"):
-        auto_tune_file = gr.File(label="Source Voice Clip")
-        auto_tune_key = gr.Textbox(label="Target Key", value="C", lines=1)
-        auto_tune_output = gr.Audio(label="Pitch-Corrected Output", type="filepath")
-        auto_tune_btn = gr.Button("Auto-Tune")
-
-        auto_tune_btn.click(
-            fn=auto_tune_vocal,
-            inputs=[auto_tune_file, auto_tune_key],
-            outputs=auto_tune_output
-        )
+    # --- Batch Processing ---
+with gr.Tab("🔊 Batch Processing"):
+    batch_files = gr.File(label="Upload Multiple Files", file_count="multiple")
+    batch_effects = gr.CheckboxGroup(choices=preset_choices["Default"], label="Apply Effects in Order")
+    batch_isolate = gr.Checkbox(label="Isolate Vocals After Effects")
+    batch_preset_dropdown = gr.Dropdown(choices=preset_names, label="Select Preset", value=preset_names[0])
+    batch_export_format = gr.Dropdown(choices=["MP3", "WAV"], label="Export Format", value="MP3")
+    batch_process_btn = gr.Button("Process All Files")
+    batch_download = gr.File(label="Download ZIP of All Processed Files")
+    batch_status = gr.Textbox(label="Status", value="Ready", interactive=False)
+
+    batch_process_btn.click(
+        fn=batch_process_audio,
+        inputs=[batch_files, batch_effects, batch_isolate, batch_preset_dropdown, batch_export_format],
+        outputs=[batch_download, batch_status]
+    )
 
-    # --- Frequency Spectrum ---
-    with gr.Tab("📊 Frequency Spectrum"):
-        spectrum_input = gr.Audio(label="Upload Track", type="filepath")
-        spectrum_output = gr.Image(label="Spectrum Analysis")
-        spectrum_btn = gr.Button("Visualize Spectrum")
+# --- AI Auto-Tune ---
+with gr.Tab("🎤 AI Auto-Tune"):
+    auto_tune_file = gr.File(label="Source Voice Clip")
+    auto_tune_key = gr.Textbox(label="Target Key", value="C", lines=1)
+    auto_tune_output = gr.Audio(label="Pitch-Corrected Output", type="filepath")
+    auto_tune_btn = gr.Button("Auto-Tune")
+
+    auto_tune_btn.click(
+        fn=auto_tune_vocal,
+        inputs=[auto_tune_file, auto_tune_key],
+        outputs=auto_tune_output
+    )
 
-        spectrum_btn.click(
-            fn=visualize_spectrum,
-            inputs=spectrum_input,
-            outputs=spectrum_output
-        )
+# --- Frequency Spectrum ---
+with gr.Tab("📊 Frequency Spectrum"):
+    spectrum_input = gr.Audio(label="Upload Track", type="filepath")
+    spectrum_output = gr.Image(label="Spectrum Analysis")
+    spectrum_btn = gr.Button("Visualize Spectrum")
 
-    # --- Loudness Graph ---
-    with gr.Tab("📈 Loudness Graph"):
-        loudness_input = gr.Audio(label="Upload Track", type="filepath")
-        loudness_target = gr.Slider(minimum=-24, maximum=-6, value=-14, step=0.5, label="Target LUFS")
-        loudness_output = gr.Audio(label="Normalized Output", type="filepath")
-        loudness_btn = gr.Button("Match Loudness")
-
-        loudness_btn.click(
-            fn=match_loudness,
-            inputs=[loudness_input, loudness_target],
-            outputs=loudness_output
-        )
+    spectrum_btn.click(
+        fn=visualize_spectrum,
+        inputs=spectrum_input,
+        outputs=spectrum_output
+    )
 
-    # --- Save/Load Project ---
-    with gr.Tab("📁 Save/Load Project"):
-        with gr.Row():
-            with gr.Column(min_width=300):
-                project_audio_file = gr.File(label="Original Audio")
-                project_preset = gr.Dropdown(choices=preset_names, label="Used Preset", value=preset_names[0])
-                project_effects = gr.CheckboxGroup(choices=preset_choices["Default"], label="Applied Effects")
-                save_proj_btn = gr.Button("Save Project")
-                project_file_out = gr.File(label="Project File (.aiproj)")
-            with gr.Column(min_width=300):
-                load_proj_file = gr.File(label="Upload .aiproj File")
-                loaded_preset_out = gr.Dropdown(choices=preset_names, label="Loaded Preset")
-                loaded_effects_out = gr.CheckboxGroup(choices=preset_choices["Default"], label="Loaded Effects")
-                load_proj_btn = gr.Button("Load Project")
-
-        save_proj_btn.click(
-            fn=save_project,
-            inputs=[project_audio_file, project_preset, project_effects],
-            outputs=project_file_out
-        )
+# --- Loudness Graph ---
+with gr.Tab("📈 Loudness Graph"):
+    loudness_input = gr.Audio(label="Upload Track", type="filepath")
+    loudness_target = gr.Slider(minimum=-24, maximum=-6, value=-14, step=0.5, label="Target LUFS")
+    loudness_output = gr.Audio(label="Normalized Output", type="filepath")
+    loudness_btn = gr.Button("Match Loudness")
+
+    loudness_btn.click(
+        fn=match_loudness,
+        inputs=[loudness_input, loudness_target],
+        outputs=loudness_output
+    )
 
-        load_proj_btn.click(
-            fn=load_project,
-            inputs=load_proj_file,
-            outputs=[loaded_preset_out, loaded_effects_out]
-        )
+# --- Save/Load Project ---
+with gr.Tab("📁 Save/Load Project"):
+    with gr.Row():
+        with gr.Column(min_width=300):
+            project_audio_file = gr.File(label="Original Audio")
+            project_preset = gr.Dropdown(choices=preset_names, label="Used Preset", value=preset_names[0])
+            project_effects = gr.CheckboxGroup(choices=preset_choices["Default"], label="Applied Effects")
+            save_proj_btn = gr.Button("Save Project")
+            project_file_out = gr.File(label="Project File (.aiproj)")
+        with gr.Column(min_width=300):
+            load_proj_file = gr.File(label="Upload .aiproj File")
+            loaded_preset_out = gr.Dropdown(choices=preset_names, label="Loaded Preset")
+            loaded_effects_out = gr.CheckboxGroup(choices=preset_choices["Default"], label="Loaded Effects")
+            load_proj_btn = gr.Button("Load Project")
+
+    save_proj_btn.click(
+        fn=save_project,
+        inputs=[project_audio_file, project_preset, project_effects],
+        outputs=project_file_out
+    )
 
-    # --- Prompt-Based Editing ---
-    with gr.Tab("🧠 Prompt-Based Editing"):
-        prompt_audio = gr.File(label="Upload Audio", type="filepath")
-        prompt_text = gr.Textbox(label="Describe What You Want", lines=5)
-        prompt_output = gr.Audio(label="Edited Output", type="filepath")
-        prompt_btn = gr.Button("Process Prompt")
-
-        prompt_btn.click(
-            fn=process_prompt,
-            inputs=[prompt_audio, prompt_text],
-            outputs=prompt_output
-        )
+    load_proj_btn.click(
+        fn=load_project,
+        inputs=load_proj_file,
+        outputs=[loaded_preset_out, loaded_effects_out]
+    )
 
-    # --- Custom EQ Editor ---
-    with gr.Tab("🎛 Custom EQ Editor"):
-        eq_audio = gr.Audio(label="Upload Track", type="filepath")
-        genre_list = list(preset_choices.keys())
-        eq_genre = gr.Dropdown(choices=genre_list, value="Pop", label="Genre")
-        eq_output = gr.Audio(label="EQ-Enhanced Output", type="filepath")
-        eq_btn = gr.Button("Apply EQ")
-
-        eq_btn.click(
-            fn=lambda audio_path, genre: auto_eq(AudioSegment.from_file(audio_path), genre),
-            inputs=[eq_audio, eq_genre],
-            outputs=eq_output
-        )
+# --- Prompt-Based Editing ---
+with gr.Tab("🧠 Prompt-Based Editing"):
+    prompt_audio = gr.File(label="Upload Audio", type="filepath")
+    prompt_text = gr.Textbox(label="Describe What You Want", lines=5)
+    prompt_output = gr.Audio(label="Edited Output", type="filepath")
+    prompt_btn = gr.Button("Process Prompt")
+
+    prompt_btn.click(
+        fn=process_prompt,
+        inputs=[prompt_audio, prompt_text],
+        outputs=prompt_output
+    )
 
-    # --- A/B Compare ---
-    with gr.Tab("🎯 A/B Compare"):
-        ab_track1 = gr.Audio(label="Version A", type="filepath")
-        ab_track2 = gr.Audio(label="Version B", type="filepath")
-        ab_out1 = gr.Audio(label="Version A", type="filepath")
-        ab_out2 = gr.Audio(label="Version B", type="filepath")
-        ab_btn = gr.Button("Compare")
-
-        ab_btn.click(
-            fn=compare_ab,
-            inputs=[ab_track1, ab_track2],
-            outputs=[ab_out1, ab_out2]
-        )
+# --- Custom EQ Editor ---
+with gr.Tab("🎛 Custom EQ Editor"):
+    eq_audio = gr.Audio(label="Upload Track", type="filepath")
+    genre_list = list(preset_choices.keys())
+    eq_genre = gr.Dropdown(choices=genre_list, value="Pop", label="Genre")
+    eq_output = gr.Audio(label="EQ-Enhanced Output", type="filepath")
+    eq_btn = gr.Button("Apply EQ")
+
+    eq_btn.click(
+        fn=lambda audio_path, genre: auto_eq(AudioSegment.from_file(audio_path), genre),
+        inputs=[eq_audio, eq_genre],
+        outputs=eq_output
+    )
 
-    # --- Loop Playback ---
-    with gr.Tab("🔁 Loop Playback"):
-        loop_audio = gr.Audio(label="Upload Track", type="filepath")
-        loop_start = gr.Slider(minimum=0, maximum=30000, step=100, value=5000, label="Start MS")
-        loop_end = gr.Slider(minimum=100, maximum=30000, step=100, value=10000, label="End MS")
-        loop_repeats = gr.Slider(minimum=1, maximum=10, value=2, label="Repeat Loops")
-        loop_output = gr.Audio(label="Looped Output", type="filepath")
-        loop_btn = gr.Button("Loop Section")
-
-        loop_btn.click(
-            fn=loop_section,
-            inputs=[loop_audio, loop_start, loop_end, loop_repeats],
-            outputs=loop_output
-        )
+# --- A/B Compare ---
+with gr.Tab("🎯 A/B Compare"):
+    ab_track1 = gr.Audio(label="Version A", type="filepath")
+    ab_track2 = gr.Audio(label="Version B", type="filepath")
+    ab_out1 = gr.Audio(label="Version A", type="filepath")
+    ab_out2 = gr.Audio(label="Version B", type="filepath")
+    ab_btn = gr.Button("Compare")
+
+    ab_btn.click(
+        fn=compare_ab,
+        inputs=[ab_track1, ab_track2],
+        outputs=[ab_out1, ab_out2]
+    )
 
-    # --- Share Effect Chain ---
-    with gr.Tab("🔗 Share Effect Chain"):
-        share_effects = gr.CheckboxGroup(choices=preset_choices["Default"], label="Select Effects")
-        share_code = gr.Textbox(label="Share Code", lines=2)
-        share_btn = gr.Button("Generate Share Code")
+# --- Loop Playback ---
+with gr.Tab("🔁 Loop Playback"):
+    loop_audio = gr.Audio(label="Upload Track", type="filepath")
+    loop_start = gr.Slider(minimum=0, maximum=30000, step=100, value=5000, label="Start MS")
+    loop_end = gr.Slider(minimum=100, maximum=30000, step=100, value=10000, label="End MS")
+    loop_repeats = gr.Slider(minimum=1, maximum=10, value=2, label="Repeat Loops")
+    loop_output = gr.Audio(label="Looped Output", type="filepath")
+    loop_btn = gr.Button("Loop Section")
+
+    loop_btn.click(
+        fn=loop_section,
+        inputs=[loop_audio, loop_start, loop_end, loop_repeats],
+        outputs=loop_output
+    )
 
-        share_btn.click(
-            fn=lambda x: json.dumps(x),
-            inputs=share_effects,
-            outputs=share_code
-        )
+# --- Share Effect Chain ---
+with gr.Tab("🔗 Share Effect Chain"):
+    share_effects = gr.CheckboxGroup(choices=preset_choices["Default"], label="Select Effects")
+    share_code = gr.Textbox(label="Share Code", lines=2)
+    share_btn = gr.Button("Generate Share Code")
 
-    with gr.Tab("📥 Load Shared Chain"):
-        load_code = gr.Textbox(label="Paste Shared Code", lines=2)
-        loaded_effects = gr.CheckboxGroup(choices=preset_choices["Default"], label="Loaded Effects")
-        load_code_btn = gr.Button("Load Effects")
-
-        def load_shared_code(code_str):
-            try:
-                return json.loads(code_str)
-            except Exception:
-                return []
-
-        load_code_btn.click(
-            fn=load_shared_code,
-            inputs=load_code,
-            outputs=loaded_effects
-        )
+    share_btn.click(
+        fn=lambda x: json.dumps(x),
+        inputs=share_effects,
+        outputs=share_code
+    )
 
-    # --- Keyboard Shortcuts ---
-    with gr.Tab("⌨ Keyboard Shortcuts"):
-        gr.Markdown("""
-        ### Keyboard Controls  
-        - `Ctrl + Z`: Undo last effect  
-        - `Ctrl + Y`: Redo  
-        - `Spacebar`: Play/Stop playback  
-        - `Ctrl + S`: Save current session  
-        - `Ctrl + O`: Open session  
-        - `Ctrl + C`: Copy effect chain  
-        - `Ctrl + V`: Paste effect chain  
-        """)
-
-    # --- Vocal Formant Correction ---
-    with gr.Tab("🧑‍🎤 Vocal Formant Correction"):
-        formant_audio = gr.Audio(label="Upload Vocal Track", type="filepath")
-        formant_shift = gr.Slider(minimum=-2, maximum=2, value=1.0, step=0.1, label="Formant Shift")
-        formant_output = gr.Audio(label="Natural-Sounding Vocal", type="filepath")
-        formant_btn = gr.Button("Apply Correction")
-
-        formant_btn.click(
-            fn=formant_correct,
-            inputs=[formant_audio, formant_shift],
-            outputs=formant_output
-        )
+with gr.Tab("📥 Load Shared Chain"):
+    load_code = gr.Textbox(label="Paste Shared Code", lines=2)
+    loaded_effects = gr.CheckboxGroup(choices=preset_choices["Default"], label="Loaded Effects")
+    load_code_btn = gr.Button("Load Effects")
+
+    def load_shared_code(code_str):
+        try:
+            return json.loads(code_str)
+        except Exception:
+            return []
+
+    load_code_btn.click(
+        fn=load_shared_code,
+        inputs=load_code,
+        outputs=loaded_effects
+    )
 
-    # --- Voice Swap / Cloning ---
-    with gr.Tab("🔁 Voice Swap / Cloning"):
-        source_voice = gr.File(label="Source Voice Clip")
-        reference_voice = gr.File(label="Reference Voice")
-        clone_output = gr.Audio(label="Converted Output", type="filepath")
-        clone_btn = gr.Button("Clone Voice")
-
-        clone_btn.click(
-            fn=clone_voice,
-            inputs=[source_voice, reference_voice],
-            outputs=clone_output
-        )
+# --- Keyboard Shortcuts ---
+with gr.Tab("⌨ Keyboard Shortcuts"):
+    gr.Markdown("""
+    ### Keyboard Controls  
+    - `Ctrl + Z`: Undo last effect  
+    - `Ctrl + Y`: Redo  
+    - `Spacebar`: Play/Stop playback  
+    - `Ctrl + S`: Save current session  
+    - `Ctrl + O`: Open session  
+    - `Ctrl + C`: Copy effect chain  
+    - `Ctrl + V`: Paste effect chain  
+    """)
+
+# --- Vocal Formant Correction ---
+with gr.Tab("🧑‍🎤 Vocal Formant Correction"):
+    formant_audio = gr.Audio(label="Upload Vocal Track", type="filepath")
+    formant_shift = gr.Slider(minimum=-2, maximum=2, value=1.0, step=0.1, label="Formant Shift")
+    formant_output = gr.Audio(label="Natural-Sounding Vocal", type="filepath")
+    formant_btn = gr.Button("Apply Correction")
+
+    formant_btn.click(
+        fn=formant_correct,
+        inputs=[formant_audio, formant_shift],
+        outputs=formant_output
+    )
 
-    # --- DAW Template Export ---
-    with gr.Tab("🎛 DAW Template Export"):
-        daw_stems = gr.File(label="Upload Stems", file_count="multiple")
-        daw_output = gr.File(label="DAW Template (.json/.als/.flp)")
-        daw_btn = gr.Button("Generate Template")
+# --- Voice Swap / Cloning ---
+with gr.Tab("🔁 Voice Swap / Cloning"):
+    source_voice = gr.File(label="Source Voice Clip")
+    reference_voice = gr.File(label="Reference Voice")
+    clone_output = gr.Audio(label="Converted Output", type="filepath")
+    clone_btn = gr.Button("Clone Voice")
+
+    clone_btn.click(
+        fn=clone_voice,
+        inputs=[source_voice, reference_voice],
+        outputs=clone_output
+    )
 
-        daw_btn.click(
-            fn=generate_ableton_template,
-            inputs=daw_stems,
-            outputs=daw_output
-        )
+# --- DAW Template Export ---
+with gr.Tab("🎛 DAW Template Export"):
+    daw_stems = gr.File(label="Upload Stems", file_count="multiple")
+    daw_output = gr.File(label="DAW Template (.json/.als/.flp)")
+    daw_btn = gr.Button("Generate Template")
 
-    # --- Export Full Mix ZIP ---
-    with gr.Tab("📁 Export Full Mix ZIP"):
-        stems_files = gr.File(label="Stems", file_count="multiple")
-        final_mix_file = gr.File(label="Final Mix")
-        full_zip_output = gr.File(label="Full Mix Archive (.zip)")
-        export_zip_btn = gr.Button("Export ZIP")
-
-        export_zip_btn.click(
-            fn=export_full_mix,
-            inputs=[stems_files, final_mix_file],
-            outputs=full_zip_output
-        )
-    
+    daw_btn.click(
+        fn=generate_ableton_template,
+        inputs=daw_stems,
+        outputs=daw_output
+    )
+
+# --- Export Full Mix ZIP ---
+with gr.Tab("📁 Export Full Mix ZIP"):
+    stems_files = gr.File(label="Stems", file_count="multiple")
+    final_mix_file = gr.File(label="Final Mix")
+    full_zip_output = gr.File(label="Full Mix Archive (.zip)")
+    export_zip_btn = gr.Button("Export ZIP")
+
+    export_zip_btn.click(
+        fn=export_full_mix,
+        inputs=[stems_files, final_mix_file],
+        outputs=full_zip_output
+    )
 
 demo.launch()