import requests

# Substitua pelo seu URL do ngrok ou Space
url = "https://8fcf-34-147-121-171.ngrok-free.app/"  # Certifique-se de usar o endpoint correto

def infer_simple(prompt):
    # Envia uma requisição POST para o endpoint do servidor
    response = requests.post(url, json={"prompt": prompt})
    
    # Verifica se a requisição foi bem-sucedida
    if response.status_code == 200:
        return response.json()["result"]
    else:
        print(f"Erro: {response.status_code} - {response.text}")
        return None

# Testando a função
if __name__ == "__main__":
    prompt = "A beautiful brunette woman wearing a blue leather pants"
    result = infer_simple(prompt)
    print(f"Resultado: {result}")