import gradio as gr
from AudioFusion import Fusion

def process_audio(input_file, effect_8d, effect_slowed, effect_reverb,
                  pan_boundary, jump_percentage, time_l_to_r, volume_multiplier,
                  speed_multiplier,
                  room_size, damping, width, wet_level, dry_level):
    try:
        # Load the sound file
        sound = Fusion.loadSound(input_file)

        # Apply effects based on user choices
        if effect_8d:
            sound = Fusion.effect8D(sound, pan_boundary, jump_percentage, time_l_to_r, volume_multiplier)
        if effect_slowed:
            sound = Fusion.effectSlowed(sound, speed_multiplier)
        if effect_reverb:
            sound = Fusion.effectReverb(sound, room_size, damping, width, wet_level, dry_level)

        # Save the processed sound and return the output file
        output_file = "output"
        Fusion.saveSound(sound, output_file, effect_reverb)
        return output_file

    except Fusion.InvalidMusicFileError as e:
        return str(e)
    except Exception as e:
        return f"An error occurred: {str(e)}"

iface = gr.Interface(
    fn=process_audio,
    inputs=[
        gr.File(label="Upload your music file"),
        gr.Checkbox(label="Apply 8D effect"),
        gr.Checkbox(label="Apply slowed effect"),
        gr.Checkbox(label="Apply reverb effect"),
        gr.Number(label="8D - Pan Boundary", default=100, min=0, max=1000),
        gr.Number(label="8D - Jump Percentage", default=5, min=1, max=100),
        gr.Number(label="8D - Time L to R (ms)", default=10000, min=1),
        gr.Number(label="8D - Volume Multiplier", default=6, min=1),
        gr.Number(label="Slowed - Speed Multiplier)", default=0.92, min=0.1, max=2, step=0.01),
        gr.Number(label="Reverb - Room Size", default=0.8, min=0, max=1, step=0.1),
        gr.Number(label="Reverb - Damping", default=1, min=0, max=10),
        gr.Number(label="Reverb - Width", default=0.5, min=0, max=1, step=0.1),
        gr.Number(label="Reverb - Wet Level", default=0.3, min=0, max=1, step=0.1),
        gr.Number(label="Reverb - Dry Level", default=0.8, min=0, max=1, step=0.1),
    ],
    outputs=gr.File(label="Download processed music"),
    live=True,
    capture_session=True
)

iface.launch()