import gradio as gr
from infer_rvc_python import infer
import soundfile as sf

# Model files
MODEL_PATH = "model.pth"
INDEX_PATH = "model.index"

# Function to run inference
def convert_voice(input_audio):
    # Load uploaded audio
    audio, sr = sf.read(input_audio)
    # Run RVC inference
    output = infer(
        audio=audio,
        sr=sr,
        model_path=MODEL_PATH,
        index_path=INDEX_PATH,
        f0_method="crepe",
        index_rate=0.75,
    )
    # Save output
    sf.write("output.wav", output, sr)
    return "output.wav"

# Gradio UI
iface = gr.Interface(
    fn=convert_voice,
    inputs=gr.Audio(source="upload", type="filepath"),
    outputs=gr.Audio(type="filepath"),
    title="RVC Voice Conversion",
    description="Upload audio to convert with RVC.",
    allow_flagging=False
)

iface.launch()