import time

import requests

X_RAPIDAPI_KEY = "<YOUR_RAPID_API_KEY"

url = "https://lanytek-audio2midi.p.rapidapi.com/audios/gen-from-youtube"

querystring = {
    "url": "https://www.youtube.com/watch?v=lRVTVB94zTg&ab_channel=Passenger"
}

headers = {
    "content-type": "application/json",
    "X-RapidAPI-Key": X_RAPIDAPI_KEY,
    "X-RapidAPI-Host": "lanytek-audio2midi.p.rapidapi.com",
}

# submit translation request
response = requests.request("POST", url, json={}, headers=headers, params=querystring)

request_id = response.json()["requestId"]
print(f"request_id: {request_id}")

# wait for translation to be ready
while True:
    time.sleep(10)

    url = "https://lanytek-audio2midi.p.rapidapi.com/audios"

    querystring = {"req_id": request_id}

    headers = {
        "X-RapidAPI-Key": X_RAPIDAPI_KEY,
        "X-RapidAPI-Host": "lanytek-audio2midi.p.rapidapi.com",
    }

    response = requests.request("GET", url, headers=headers, params=querystring)
    if response.status_code == 200:
        data = response.json()
        music_sheet_pdf_url = data["pdf"]
        midi_url = data["midi"]

        print(f"music_sheet_pdf_url: {music_sheet_pdf_url}")
        print(f"midi_url: {midi_url}")
        break

    print(response.json()["detail"])