import time
import requests
import os

base_url = os.environ["VERCEL_API"]

def custom_generate_audio(payload):
    url = f"{base_url}api/custom_generate"
    response = requests.post(url, json=payload, headers={'Content-Type': 'application/json'})
    return response.json()


def extend_audio(payload):
    url = f"{base_url}api/extend_audio"
    response = requests.post(url, json=payload, headers={'Content-Type': 'application/json'})
    return response.json()

def generate_audio_by_prompt(payload):
    url = f"{base_url}api/generate"
    response = requests.post(url, json=payload, headers={'Content-Type': 'application/json'})
    return response.json()


def get_audio_information(audio_ids):
    url = f"{base_url}api/get?ids={audio_ids}"
    response = requests.get(url)
    return response.json()


def generate_lyrics(payload):
    url = f"{base_url}api/generate_lyrics"
    response = requests.post(url, json=payload, headers={'Content-Type': 'application/json'})
    return response.json()
    
def get_quota_information():
    url = f"{base_url}api/get_limit"
    response = requests.get(url)
    return response.json()

def get_clip(clip_id):
    url = f"{base_url}api/clip?id={clip_id}"
    response = requests.get(url)
    return response.json()

def generate_whole_song(clip_id):
    payloyd = {"clip_id": clip_id}
    url = f"{base_url}api/concat"
    response = requests.post(url, json=payload)
    return response.json()
    
def generate_lyrics(prompt):
    payload = {"prompt": prompt}
    url = f"{base_url}api/generate_lyrics"
    print(url)
    response = requests.post(url, json=payload)
    print(response)
    return response.json()