import os
import requests
from dotenv import load_dotenv


load_dotenv()

# global
TOKEN =os.environ.get('HF_ACCESS_TOKEN')
URL = "https://ojcld767sjuhadix.us-east-1.aws.endpoints.huggingface.cloud"
headers = {"Authorization": f"Bearer {TOKEN}"}

# prepare sample payload
prompt = "Can you describe this picture focusing on specifics visual artifacts and ambiance (objects, colors, person, athmosphere..). Please stay concise only output keywords and concepts detected."
image = "https://media.rolex.com/image/upload/q_auto/f_auto/c_limit,w_2440/v1708384234/rolexcom/about-rolex/hub/about-rolex-hub-cover-aca202310cw-0002-portrait"

def call(payload):
    response = requests.post(
        URL,
        headers=headers,
        json=payload
    )
    return response.json()

def test_local(payload):
    from handler import EndpointHandler

    # init handler
    my_handler = EndpointHandler(path=".")
    my_handler(payload)


payload = {
    "inputs": {
        "prompt": prompt,
        "image": image
    }
}

# out_local = test_local(payload)
# print(out_local)

out = call(payload)
print(out)