import requests
import gradio as gr
import os

API_URL = "https://api-inference.huggingface.co/models/facebook/musicgen-medium"
API_KEY = os.getenv("HUGGINGFACE_API_KEY") 

def query(payload, api_key):
    headers = {"Authorization": "Bearer " + api_key}
    response = requests.post(API_URL, headers=headers, json=payload)
    return response.content

def generate_music(prompt):
    audio_bytes = query({"inputs": prompt}, API_KEY)
    return audio_bytes, None  # Return audio bytes and no error

title = "MusicGen"
description = "Generate music based on the provided prompt using Facebook's music generation model."

iface = gr.Interface(
    fn=generate_music,
    inputs=gr.Textbox(lines=3, label="Enter music prompt"),
    outputs=gr.Audio(type="numpy", label="Generated Music"),
    title=title,
    description=description
)

iface.launch()