import gradio as gr

import sulkuPypi
from funciones import mass

def perform(input1, input2):

    #Primero que nada, checa cuantos tokens: 
    tokens = sulkuPypi.getTokens("gAAAAABmEZA4SLBC2YczouOrjIEi9WNCNGOIvyUcqBUnzxNsftXTdy54KaX9x8mAjFkABSI6FJrdZDQKk_5lpJOgJoMChxlniw==")
    print(f"Tienes: {tokens} tokens. ")

    #Después autoriza.
    #Si está autorizada puede ejecutar la tarea, ésta lógica si está a cargo aquí, por parte de la app y su desarrollador, no de Sulku.
    autorizacion = sulkuPypi.authorize(tokens, 'picswap')
    print("La autorización es: ", autorizacion)

    #Después ejecuta la masa.
    if autorizacion is True: 
    
        path = mass(input1,input2)

    else:
        print("No tienes suficientes tokens...")
        return "No tienes suficientes tokens" 


    print("Después de mass imprimo ésto.")

    #Y finalmente debita los tokens.
    tokens = sulkuPypi.debitTokens("gAAAAABmEZA4SLBC2YczouOrjIEi9WNCNGOIvyUcqBUnzxNsftXTdy54KaX9x8mAjFkABSI6FJrdZDQKk_5lpJOgJoMChxlniw==", "picswap")
    print(f"Y ahora tienes: {tokens} tokens.")

    return path

#Así para imagenes
demo = gr.Interface(
fn=perform, inputs=[gr.Image(), gr.Image()], outputs=[gr.Image()]
)
demo.launch()