Create app.py

app.py

@@ -0,0 +1,104 @@
+import io
+import json
+from pydub import AudioSegment
+from pydub.effects import normalize, compress_dynamic_range, low_pass_filter, high_pass_filter
+import numpy as np
+import gradio as gr
+
+# Default presets for common genres
+PRESETS = {
+    "rock": {"gain": 6, "compress_threshold": -20, "compress_ratio": 4, "low_pass": 12000, "high_pass": 80},
+    "pop": {"gain": 4, "compress_threshold": -18, "compress_ratio": 3, "low_pass": 15000, "high_pass": 100},
+    "jazz": {"gain": 3, "compress_threshold": -22, "compress_ratio": 2, "low_pass": 14000, "high_pass": 60},
+    "electronic": {"gain": 8, "compress_threshold": -16, "compress_ratio": 5, "low_pass": 20000, "high_pass": 120},
+}
+
+# JSON export extension
+EXPORT_EXT = ".master"
+
+# Custom CSS for Gradio
+CUSTOM_CSS = """
+body { background-color: #1e1e1e; color: #f0f0f0; }
+.gradio-container { border-radius: 12px; box-shadow: 0 4px 12px rgba(0,0,0,0.5); }
+#title { font-size: 2rem; margin-bottom: 1rem; }
+"""
+
+def apply_mastering(audio: AudioSegment, params: dict) -> AudioSegment:
+    # Gain
+    processed = audio + params["gain"]
+    # Compression
+    processed = compress_dynamic_range(processed,
+        threshold=params["compress_threshold"],
+        ratio=params["compress_ratio"])
+    # EQ filters
+    processed = high_pass_filter(processed, params["high_pass"])
+    processed = low_pass_filter(processed, params["low_pass"])
+    # Normalization
+    processed = normalize(processed)
+    return processed
+
+
+def process_and_export(audio_file, preset_name, gain, threshold, ratio, lp, hp):
+    # Load
+    audio = AudioSegment.from_file(audio_file.name)
+    # Determine params
+    if preset_name != "custom":
+        params = PRESETS[preset_name]
+    else:
+        params = {"gain": gain, "compress_threshold": threshold, "compress_ratio": ratio,
+                  "low_pass": lp, "high_pass": hp}
+    # Process
+    out = apply_mastering(audio, params)
+    # Export audio
+    buf = io.BytesIO()
+    out.export(buf, format="wav")
+    buf.seek(0)
+    # Create preset JSON
+    preset = {"preset": preset_name, "params": params}
+    preset_buf = io.BytesIO(json.dumps(preset, indent=2).encode())
+    preset_buf.name = "preset" + EXPORT_EXT
+    return (buf, "processed.wav"), (preset_buf, preset_buf.name)
+
+
+def load_preset_json(file_obj):
+    data = json.load(file_obj)
+    name = data.get("preset", "custom")
+    params = data.get("params", {})
+    return name, params.get("gain", 0), params.get("compress_threshold", 0), \
+           params.get("compress_ratio", 1), params.get("low_pass", 20000), params.get("high_pass", 20)
+
+with gr.Blocks(css=CUSTOM_CSS) as demo:
+    gr.Markdown("<div id='title'>Advanced Browser-based Audio Mastering Suite</div>")
+    with gr.Row():
+        with gr.Column():
+            audio_input = gr.Audio(label="Upload Audio", type="file")
+            preset_dropdown = gr.Dropdown(choices=list(PRESETS.keys()) + ["custom"], value="rock", label="Preset")
+            gain = gr.Slider(-10, 20, value=PRESETS["rock"]["gain"], label="Gain (dB)")
+            threshold = gr.Slider(-60, 0, value=PRESETS["rock"]["compress_threshold"], label="Compress Threshold (dB)")
+            ratio = gr.Slider(1, 10, value=PRESETS["rock"]["compress_ratio"], label="Compress Ratio")
+            lp = gr.Slider(1000, 20000, value=PRESETS["rock"]["low_pass"], label="Low-pass Frequency (Hz)")
+            hp = gr.Slider(20, 500, value=PRESETS["rock"]["high_pass"], label="High-pass Frequency (Hz)")
+            load_preset = gr.File(label="Load .master Preset", file_types=[".master"])
+            export_button = gr.Button("Process & Export")
+        with gr.Column():
+            output_audio = gr.Audio(label="Processed Audio")
+            export_preset_file = gr.File(label="Download Preset (.master)")
+    
+    # Interactivity
+    def sync_sliders(preset):
+        if preset != "custom":
+            p = PRESETS[preset]
+            return gr.update(value=p["gain"]), gr.update(value=p["compress_threshold"]), \
+                   gr.update(value=p["compress_ratio"]), gr.update(value=p["low_pass"]), \
+                   gr.update(value=p["high_pass"])
+        return None
+
+    preset_dropdown.change(sync_sliders, inputs=[preset_dropdown], outputs=[gain, threshold, ratio, lp, hp])
+    load_preset.upload(load_preset_json, inputs=[load_preset],
+                        outputs=[preset_dropdown, gain, threshold, ratio, lp, hp])
+    export_button.click(process_and_export,
+                         inputs=[audio_input, preset_dropdown, gain, threshold, ratio, lp, hp],
+                         outputs=[output_audio, export_preset_file])
+
+if __name__ == "__main__":
+    demo.launch()