import requests
from dotenv import load_dotenv
from gradio_client import Client, handle_file
import os


if __name__ == "__main__":
    load_dotenv()
    HF_TOKEN = os.environ.get("HF_TOKEN")
    HF_USERNAME = os.environ.get("HF_USERNAME")
    HF_SPACENAME = os.environ.get("HF_SPACENAME")

    SPACE_URL = f"https://{HF_USERNAME}-{HF_SPACENAME}.hf.space"
    API_URL = f"{SPACE_URL}/run/predict"

    client = Client(f"{HF_USERNAME}/{HF_SPACENAME}", hf_token=HF_TOKEN)
    # client = Client("http://127.0.0.1:7860")

    client.view_api()

    result = client.predict(
        handle_file("truck.m4a"),
        api_name="/predict"
    )

    print(result)