import gradio as gr
import time

def generate_music(prompt, duration):
    print(f"🎵 Génération demandée : {prompt} pour {duration} secondes")
    # Simule un temps de calcul
    time.sleep(2) 
    return "Dummy audio generated !"

# Interface pour Open WebUI
demo = gr.Interface(
    fn=generate_music,
    inputs=["text", "number"],
    outputs="text",
    title="ACE-Step Audio API"
)

if __name__ == "__main__":
    demo.launch(server_name="0.0.0.0", server_port=7860)