Display image as result too

app.py

@@ -7,7 +7,13 @@ import pathlib
 def greet(input1, input2):
 
     modo = "video"
+    #local o huggingface
+    plataforma = "local"
+    #face_swapper o face_enhancer o la combinación de ellos.
+    procesador = "face_swapper"
     print("Inicio: Estamos en modo ", modo)
+    print("Estamos en la plataforma:", plataforma)
+    print("El procesador es: ", procesador)
 
     print("Input1:")
     print(input1)
@@ -15,22 +21,35 @@ def greet(input1, input2):
     print(input2)
 
     path_video = input2
-    #Para local.
-    #path_parts = path_video.split("\\")
-    #Para HuggingFace
-    path_parts = path_video.split("/")
+
+    if plataforma == "local":
+        #Para local.
+        path_parts = path_video.split("\\")
+    else:
+        #Para HuggingFace
+        path_parts = path_video.split("/")
+    
+    
     print("Imprimiendo path_parts: ", path_parts)
-    path_bueno = "/".join(path_parts[0:len(path_parts) - 2])
-    #path_bueno = path_bueno.replace("\\", "/")
+    path_particular = "/".join(path_parts[0:len(path_parts) - 1])
+    path_general = "/".join(path_parts[0:len(path_parts) - 2])
 
-    print("Path bueno:")
-    print(path_bueno)
+    
+    path_general = path_general.replace("\\", "/")
+    path_particular = path_particular.replace("\\", "/")
 
-    files = os.listdir(path_bueno)
+    print("Path general: ", path_general)
+    print("Path general: ", path_particular)
+
+    path = pathlib.Path("result.mp4")
+    path_foto = pathlib.Path(path_particular + "/temp/whitebeauty/0015.png")
+    print("Éste es el path foto: ", path_foto)
+    
+    files = os.listdir(path_general)
     print("Estos son los files que hay:")
     print(files)
 
-    time.sleep(10)
+    time.sleep(5)
 
     ext_imagen = "png"
     ext_video = "mp4"
@@ -68,7 +87,7 @@ def greet(input1, input2):
     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 face_swapper"
+    command = f"python run.py -s {source_path}  -t {target_path} -o {result_path} --frame-processor {procesador}"
     print(command)
     time.sleep(1)
     proc = os.popen(command)
@@ -80,15 +99,13 @@ def greet(input1, input2):
     print("Terminó la impresión del output...")
 
     print("Ahora estamos imprimiendo las rutas para ver si apareció una nueva:")
-    files = os.listdir(path_bueno)
+    files = os.listdir(path_general)
     print("Estos son los files que hay:")
     print(files)
 
     if modo == "video":
         #Para video
-        path = pathlib.Path("result.mp4")
-        print("Éste es el path para video:", path)
-        return path, path
+        return path, path_foto
     else:
         #Para imagen
         path = pathlib.Path("result.png")
@@ -102,7 +119,7 @@ def greet(input1, input2):
 
 #Así para video
 demo = gr.Interface(
-fn=greet, inputs=[gr.Image(), gr.Video()], outputs=[gr.Video(), gr.Video()]
+fn=greet, inputs=[gr.Image(), gr.Video()], outputs=[gr.Video(), gr.Image()]
 )
 
 demo.launch()

hasher.py

@@ -0,0 +1,16 @@
+import base64
+import io
+import os
+
+def convert_video_to_base64(filename):
+    with open(filename, "rb") as f:
+        video_bytes = f.read()
+
+    base64_string = base64.b64encode(video_bytes)
+    return base64_string.decode("utf-8")
+
+if __name__ == "__main__":
+    filename = "result.mp4"
+    base64_string = convert_video_to_base64(filename)
+
+    print(base64_string)