Mass Path Ok

app.py

@@ -1,8 +1,4 @@
 import gradio as gr
-import time
-import os
-import pathlib
-from PIL import Image
 
 import sulkuPypi
 from funciones import mass
@@ -14,127 +10,16 @@ def perform(input1, input2):
     print(f"Tienes: {tokens} tokens. ")
 
     #Después autoriza.
-
-    #video o cualquier otro sería para imagenes.
-    modo = "pic"
-    #local o huggingface
-    plataforma = "huggingface"
-    #face_swapper o face_enhancer o la combinación de ellos.
-    procesador = "face_swapper"
-
-    print(f"Inicio: Estamos en modo {modo}, plataforma: {plataforma} y procesador: {procesador}.")
-    
-    path_video = input2
-    print("Path_video es:", path_video)
-    time.sleep(4)
-
-    if modo == "video":
-    
-        if plataforma == "local":
-            #Para local.
-            path_parts = path_video.split("\\")
-        else:
-            #Para HuggingFace
-            #Creo que no va en imagen.
-            print("La plataforma en la que basaremos la división es HuggingFace.")
-            path_parts = path_video.split("/")
-
-            #Aquí obtendremos nom_video
-            #Creo no va en imagen
-            filename = path_parts[-1]
-            nom_video = filename[:-4]
-            print("Esto es filename alias nom_video: ", nom_video)
-            path_particular = "/".join(path_parts[0:len(path_parts) - 1])
-            path_general = "/".join(path_parts[0:len(path_parts) - 2])
-            path_general = path_general.replace("\\", "/")
-            path_particular = path_particular.replace("\\", "/")
-            print("Path general: ", path_general)
-            print("Path general: ", path_particular)
-            path = pathlib.Path("result.mp4")
-            files = os.listdir(path_general)
-        
-            print("Estos son los files que hay:")
-            print(files)
-
-    ext_imagen = "png"
-    ext_video = "mp4"
-
-    #Selector de modo.
-    if modo == "video": 
-        print("Se asigno la extensión de video:", ext_video)
-        extension = ext_video
-    else:
-        print("Se asigno la extensión de imagen:", ext_imagen)
-        extension = ext_imagen
-
-    #El source siempre es una imagen.
-    source_path = "source.png"
-    target_path = "target." + extension
-    result_path = "result." + extension
-
-    #La primera siempre será una imagen, por eso no entra en el modo selector.
-    source_image = Image.fromarray(input1)
-    print("Esto es source_image: ", source_image)
-    source_image.save(source_path)
-        
-    #Aquí trabajaremos solo el target.
-    if modo == "video":
-        #Para Video
-        target_path = input2
-    else:
-        #Es decir si es modo imagen
-        #Para Imagenes
-        target_image = Image.fromarray(input2)
-        print("Esto es target_image: ", target_image)
-        target_image.save(target_path)
-
-    print("Después de los selectores de modo los paths quedaron así:")
-    print("source_path: ", source_path)
-    print("target_path: ", target_path)
-
-    command = f"python run.py -s {source_path}  -t {target_path} -o {result_path} --frame-processor {procesador} --execution-provider cuda"
-    print(command)
-    time.sleep(1)
-    proc = os.popen(command)
-    output = proc.read()
-
-    print("Output (resultado de la ejecución del código):")
-    time.sleep(1)
-    print(output)
-    print("Terminó la impresión del output...")
-
-    print("Éste es el momento en el que se creo result, revisar...")
-    time.sleep(1)
-
-    try:
-        print("Ésta vez no crearemos archivo zip.")
-
-    except Exception as e:
-        # código que se ejecutará si se produce la excepción
-        print(e)
-    
-    if modo == "video":
-        #Para video
-        path = pathlib.Path("result.mp4")
-        path_abs = os.path.abspath(path)
-        print("Éste es el path para video:", path)
-        print("Y su ruta absoluta es: ", path_abs)
-        return path
-    else:
-        #Para imagen
-        path = pathlib.Path("result.png")
-        print("Éste es el path para imagen:", path)
-        return path
-    print("Listo! Gracias!")
     
     #Después ejecuta la masa.
-    #path = mass(input1,input2)
+    path = mass(input1,input2)
 
     print("Después de mass imprimo ésto.")
-    time.sleep(10)
 
     #Y finalmente debita los tokens.
 
+    return path
+
 #Así para imagenes
 demo = gr.Interface(
 fn=perform, inputs=[gr.Image(), gr.Image()], outputs=[gr.Image()]