video-blend / app.py
Moibe
Add path result
5a2242d
raw
history blame
No virus
6.45 kB
import gradio as gr
from PIL import Image
import time
import os
import pathlib
import zipfile36 as zipfile
def save_images_as_zip(path_foto, filename, plataforma):
print("Entr茅 a la funci贸n que hace los zips...")
with zipfile.ZipFile(filename, "w") as zip_file:
for foto in os.listdir(path_foto):
print("La foto en os.listdir es: ", foto)
#Si Local
# path_foto_zippable = str(path_foto) + "\\" + foto
# #Si HuggingFace
# path_foto_zippable = str(path_foto) + "/" + foto
path_foto_zippable = str(path_foto) + (os.sep if plataforma == "local" else "/") + foto
print("La ruta textual final de esa foto en particular es: ", path_foto_zippable)
time.sleep(1)
# ruta = pathlib.Path(path_foto_zippable)
# zip_file.write(ruta)
ruta = os.path.basename(path_foto_zippable)
zip_file.write(path_foto_zippable, ruta)
def greet(input1, input2):
modo = "video"
#local o huggingface
plataforma = "huggingface"
#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)
print("Input2:")
print(input2)
path_video = input2
if plataforma == "local":
#Para local.
path_parts = path_video.split("\\")
else:
#Para HuggingFace
print("La plataforma en la que basaremos la divisi贸n es HuggingFace.")
path_parts = path_video.split("/")
print("Esto es parth_paths, de aqu铆 lo puedes obtener?: ", path_parts)
time.sleep(10)
print("Imprimiendo path_parts: ", path_parts)
#Aqu铆 obtendremos nom_video
filename = path_parts[-1]
nom_video = filename[:-4]
print("Esto es filename alias nom_video: ", nom_video)
time.sleep(1)
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)
time.sleep(1)
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 video:", ext_video)
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(2)
print(output)
print("Termin贸 la impresi贸n del output...")
print("脡ste es el momento en el que se creo result, revisar...")
time.sleep(10)
print("Ahora estamos imprimiendo las rutas para ver si apareci贸 una nueva:")
files = os.listdir(path_general)
print("Estos son los files que hay:")
print(files)
#Creaci贸n de la galer铆a:
images = []
#nom_video = "whitebeauty"
path_foto = pathlib.Path(path_particular + "/temp/" + nom_video + "/")
print("脡ste es el path foto: ", path_foto)
path_result = path_foto + "temp.mp4"
print("Y 茅ste es el path del resultado: ", path_result)
#脡ste es el segmento que crea la galer铆a de imagenes, que por el momento no usaremos por rendimiento.
#Se reintegrar谩 si agregamos interacci贸n de poder borrar cada imagen desde la interfaz web.
# for filename in os.listdir(path_foto):
# if filename.endswith(".png"):
# path = os.path.join(path_foto, filename)
# image = Image.open(path)
# images.append(image)
# print("Esto es la lista de imagenes: ", images)
nombre_zip = nom_video + ".zip"
print("El nombre del zip ser谩: ", nombre_zip)
try:
save_images_as_zip(path_foto, nombre_zip, plataforma)
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)
path_zip = pathlib.Path(nombre_zip)
path_zip_abs = os.path.abspath(path_zip)
print("脡ste es el path para video:", path)
print("Y su ruta absoluta es: ", path_abs)
print("Y 茅ste es el path para el zip: ", path_zip)
print("Y su ruta absoluta es: ", path_zip_abs)
return path, nombre_zip
else:
#Para imagen
path = pathlib.Path("result.png")
print("脡ste es el path para imagen:", path)
return path, images, images
print("Listo! Gracias!")
#As铆 para imagenes
# demo = gr.Interface(
# fn=greet, inputs=[gr.Image(), gr.Image()], outputs=[gr.Image(), gr.Image()]
# )
#As铆 para video y 3 outputs: Video, Galer铆a y Archivo Zip.
# demo = gr.Interface(
# fn=greet, inputs=[gr.Image(), gr.Video()], outputs=[gr.Video(), gr.Gallery(), gr.File()]
# )
demo = gr.Interface(
fn=greet, inputs=[gr.Image(), gr.Video()], outputs=[gr.Video(), gr.File()]
)
demo.launch()