sulkuGateway.py

import os
import time
import nodes
import requests
import gradio_client

def getTokens(sulkukey):

    print("Entre a SulkuGateway...")
    time.sleep(3)

    #var_secreta = os.environ.get("huggingface_key")
    var_secreta = os.environ.get("hf_write")
    #print("Esto es var_secreta: ", var_secreta)
    #FUTURE: Recuerdo que no sirvió usar la clave secreta como token para otro repositorio, intenta pasarla como str? 

    print("A punto de hacer llamado a cliente de API via Sulku...")
    
    try:
        #client = gradio_client.Client("Moibe/sulku", hf_token=var_secreta, verbose=True)
        client = gradio_client.Client("Moibe/sulku", verbose=True)
        print("ESTE ES EL CLIENTE:", client)
        time.sleep(7)
        print("Eso fue el cliente...")
        resultado = client.predict(sulkukey, api_name="/getTokens")
        
        return resultado
    
    # except gradio_client.exceptions.APIError as e:
    #     print(f"Error 404 calling the API: {e}")
    #     return None
    except requests.exceptions.RequestException as e:
        print(f"Error 405: Network error, {e}")
        return None


def debitTokens(sulkukey, work):
     
    #client = gradio_client.Client(nodes.validator, nodes.hf_token, verbose=False)
    client = gradio_client.Client("Moibe/sulku", verbose=True)
    
        
    tokens = client.predict(
            sulkukey,
            work,
            api_name="/debitTokens"
    ) 

    print(f"Available tokens now: {tokens}.")   
    time.sleep(1) 

    return tokens