import gradio as gr
import sulkuPypi
from funciones import mass
from data import usuarios
import encrypter
import time

#Funciones adicionales
def authenticate(username, password):
    
    for u, p in usuarios:
        if username == u and password == p:
            gr.State.usuario = username
            return True
    return False

#Función principal
def perform(input1, input2):

    print("Primero que nada, el usuarrio es: ", gr.State.usuario)
    capsule = encrypter.encripta(gr.State.usuario)
    print("Y éste es el usuario encriptado: ", capsule)
    time.sleep(5)

    #Primero que nada, checa cuantos tokens: 
    tokens = sulkuPypi.getTokens(capsule)
    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("gAAAAABmiGY0LH_WCawTg89yIlAb86moOyx2JzvvizVxYAogO9crR8Bu7NnxyJ_jNeBcuNfnSNaDmMbyiPTXNMDoa7i6gAz37A==", "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()
demo.launch(auth=authenticate)