Update app.py

app.py

@@ -23,7 +23,7 @@ def swap_face(source_file, target_file, doFaceEnhancer):
     session_dir = "temp"  # Sử dụng thư mục cố định
     os.makedirs(session_dir, exist_ok=True)
 
-    # Создаем уникальные имена файлов
+    # Tạo tên file ngẫu nhiên
     source_filename = f"source_{random.randint(1000, 9999)}.jpg"
     target_filename = f"target_{random.randint(1000, 9999)}.jpg"
     output_filename = f"output_{random.randint(1000, 9999)}.jpg"
@@ -31,21 +31,19 @@ def swap_face(source_file, target_file, doFaceEnhancer):
     source_path = os.path.join(session_dir, source_filename)
     target_path = os.path.join(session_dir, target_filename)
 
-    # Сохраняем изображения по указанным путям
-    source_image = Image.open(source_file.name)
-    source_image.save(source_path)
-    target_image = Image.open(target_file.name)
-    target_image.save(target_path)
+    # Сохранение изображений
+    Image.fromarray(source_file).save(source_path)
+    Image.fromarray(target_file).save(target_path)
 
     print("source_path: ", source_path)
     print("target_path: ", target_path)
 
-    # Проверяем, обнаружена ли лицо на изображении-источнике
+    # Проверка наличия лица на изображении источника
     source_face = get_one_face(cv2.imread(source_path))
     if source_face is None:
         raise gradio.exceptions.Error("No face in source path detected.")
 
-    # Проверяем, обнаружена ли лицо на изображении-цели
+    # Проверка наличия лица на изображении целевого изображения
     target_face = get_one_face(cv2.imread(target_path))
     if target_face is None:
         raise gradio.exceptions.Error("No face in target path detected.")
@@ -78,15 +76,18 @@ def swap_face(source_file, target_file, doFaceEnhancer):
     roop.globals.execution_threads = 8
     
     start()
-    return normalized_output_path
+
+    # Возвращаем изображение как массив байтов
+    with open(normalized_output_path, "rb") as image_file:
+        return image_file.read()
 
 app = gr.Interface(
     fn=swap_face, 
     inputs=[
-        gr.Image(type="filepath"),  # Изменено на "filepath"
-        gr.Image(type="filepath"),  # Изменено на "filepath"
+        gr.Image(type="numpy"), 
+        gr.Image(type="numpy"), 
         gr.Checkbox(label="Face Enhancer?", info="Do face enhancement?")
     ], 
     outputs="image"
 )
-app.launch()
+app.launch()