app.py

import os
import secrets

import gradio as gr
from AudioFusion import Fusion

def process_audio(input_file,
    effect_8d, pan_boundary, jump_percentage, time_l_to_r, volume_multiplier,
    effect_slowed, speed_multiplier,
    effect_reverb, room_size, damping, width, wet_level, dry_level
):
    # Load the sound file
    sound = Fusion.loadSound(input_file)
    os.remove(os.path.abspath(input_file))
    effects_str = []
    
    # Apply effects based on user choices
    if effect_8d:
        sound = Fusion.effect8D(sound, pan_boundary, jump_percentage, time_l_to_r*1000, volume_multiplier)
        effects_str.append("8d")
    if effect_slowed:
        sound = Fusion.effectSlowed(sound, speed_multiplier)
        effects_str.append("Slowed")
    if effect_reverb:
        sound = Fusion.effectReverb(sound, room_size, damping, width, wet_level, dry_level, str(secrets.token_hex(5)))
        effects_str.append("Reverb")

    output_file = f"{input_file} {' + '.join(effects_str)} - {'By AudioFusion'}"
    
    # Save the processed sound and return the output file
    output = Fusion.saveSound(sound, output_file)
    return output


with gr.Blocks(title="Audio Fusion") as iface:
    gr.Markdown("<p align='center'><h1>Audio Fusion</h1></p>")
    input_audio = gr.Audio(label="Upload your music file", type="filepath")
    
    # 8d Effect and its arguments
    with gr.Tab("8d Effect"):
        dimension_check = gr.Checkbox(label="Apply 8D effect")
        with gr.Column(visible=False) as di_col:
            pan = gr.Slider(label="8D - Pan Boundary", minimum=0, maximum=100, value=100)
            jump = gr.Slider(label="8D - Jump Percentage", minimum=1, maximum=100, value=5)
            time = gr.Slider(label="8D - Time L to R (s)", minimum=1, maximum=30, value=10)
            volx = gr.Slider(label="8D - Volume Multiplier", minimum=1, maximum=20, value=6)
    
    # SLowed Effect and its arguments
    with gr.Tab("Slowed Effect"):
        speed_check = gr.Checkbox(label="Apply slowed effect")
        with gr.Column(visible=False) as se_col:
            speed = gr.Slider(label="Slowed - Speed Multiplier", minimum=0.1, maximum=4, step=0.01, value=0.92)
    
    # Reverb Effect and its arguments
    with gr.Tab("Reverb Effect"):
        reverb_check = gr.Checkbox(label="Apply reverb effect")
        with gr.Column(visible=False) as re_col:
            room = gr.Slider(label="Reverb - Room Size", minimum=0, maximum=2, step=0.01, value=0.8)
            damp = gr.Slider(label="Reverb - Damping", minimum=0, maximum=2, value=1)
            width = gr.Slider(label="Reverb - Width", minimum=0, maximum=2, step=0.1, value=0.5)
            wet = gr.Slider(label="Reverb - Wet Level", minimum=0, maximum=2, step=0.1, value=0.3)
            dry = gr.Slider(label="Reverb - Dry Level", minimum=0, maximum=2, step=0.1, value=0.8)

    # =====================================================
    def di_v(check):
        if check:
            return {di_col: gr.Column(visible=True)}
        else:
            return {di_col: gr.Column(visible=False)}
            
    def se_v(check):
        if check:
            return {se_col: gr.Column(visible=True)}
        else:
            return {se_col: gr.Column(visible=False)}
            
    def re_v(check):
        if check:
            return {re_col: gr.Column(visible=True)}
        else:
            return {re_col: gr.Column(visible=False)}
        

    dimension_check.change(di_v, inputs=[dimension_check], outputs=[di_col])
    speed_check.change(se_v, inputs=[speed_check], outputs=[se_col])
    reverb_check.change(re_v, inputs=[reverb_check], outputs=[re_col])
    # =====================================================
    
    inputs = [input_audio, dimension_check, pan, jump, time, volx, speed_check, speed, reverb_check, room, damp, width, wet, dry]
    output = [gr.Audio(label="Download processed music", type="filepath")]
    
    btn = gr.Button("Run", size="sm", variant="primary")
    btn.click(fn=process_audio, inputs=inputs, outputs=output, api_name="AudioFusion")

iface.launch(share=False)