import requests
from dotenv import load_dotenv
import os

load_dotenv()

_CHUTES_API_KEY    = os.getenv("CHUTES_API_KEY")

def generate_music(prompt, lyrics=None, audio_b64=None):
	api_token = _CHUTES_API_KEY

	headers = {
		"Authorization": "Bearer " + api_token,
        "Content-Type": "application/json"
	}
	
	body =     {
    "style_prompt": prompt,
    "lyrics": lyrics,
    "audio_b64": audio_b64
    }

	response = requests.post(
		"https://chutes-diffrhythm.chutes.ai/generate",
		headers=headers,
		json=body
	)

	# Print status code and response
	print(f"Status code: {response.status_code}")
	print(response.json())

generate_music("a happy children song")