import secrets
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 = secrets.token_hex(5)
        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", min=0, max=100),
        gr.Number(label="8D - Jump Percentage", min=1, max=100),
        gr.Number(label="8D - Time L to R (ms)", min=1),
        gr.Number(label="8D - Volume Multiplier", min=1),
        gr.Number(label="Slowed - Speed Multiplier)", min=0.1, max=2, step=0.01),
        gr.Number(label="Reverb - Room Size", min=0, max=1, step=0.1),
        gr.Number(label="Reverb - Damping", min=0, max=10),
        gr.Number(label="Reverb - Width", min=0, max=1, step=0.1),
        gr.Number(label="Reverb - Wet Level", min=0, max=1, step=0.1),
        gr.Number(label="Reverb - Dry Level", min=0, max=1, step=0.1),
    ],
    outputs=gr.File(label="Download processed music"),
    live=True,
    capture_session=True
)

iface.launch()