Update app.py

app.py

@@ -1,23 +1,34 @@
 #!/usr/bin/env python3
-"""FaceFusion API для Hugging Face Spaces — использует FaceFusion для замены лиц"""
+"""FaceFusion API для Hugging Face Spaces — МАКСИМАЛЬНАЯ КОНФИГУРАЦИЯ"""
 
 import os
 import tempfile
 import subprocess
 import logging
-from fastapi import FastAPI, File, UploadFile, HTTPException
-from fastapi.responses import Response
 import shutil
 import uuid
+from fastapi import FastAPI, File, UploadFile, HTTPException
+from fastapi.responses import Response
 
 logging.basicConfig(level=logging.INFO, format='%(levelname)s:%(message)s')
 logger = logging.getLogger(__name__)
 
-app = FastAPI(title="FaceFusion API")
+app = FastAPI(title="FaceFusion API - Maximum Quality")
 
-# Путь, куда будет установлен FaceFusion (внутри контейнера)
+# Путь к FaceFusion
 FACEFUSION_DIR = "/facefusion"
-FACEFUSION_SCRIPT = os.path.join(FACEFUSION_DIR, "facefusion.py")  # может отличаться
+FACEFUSION_SCRIPT = os.path.join(FACEFUSION_DIR, "facefusion.py")
+
+# Определяем доступные провайдеры выполнения (GPU, если есть)
+EXECUTION_PROVIDERS = ["CUDAExecutionProvider", "CPUExecutionProvider"]  # ZeroGPU даст CUDA
+
+# Максимальная конфигурация
+FACE_SWAPPER_MODEL = "ghost_3_256"        # лучшая замена лица
+FACE_ENHANCER_MODEL = "codeformer"         # улучшение лица с сохранением черт
+FACE_ENHANCER_BLEND = 80                    # вес улучшения (0-100)
+FACE_MASK_TYPES = ["box", "occlusion", "region"]  # все доступные маски
+FACE_MASK_BLUR = 0.7                         # размытие границ маски
+OUTPUT_IMAGE_RESOLUTION = 1024                # выходное разрешение
 
 @app.get("/health")
 async def health():
@@ -27,9 +38,8 @@ async def health():
 async def swap_face(source: UploadFile = File(...), target: UploadFile = File(...)):
     """
     Принимает два изображения: source (фото с лицом) и target (целевое изображение).
-    Возвращает обработанное изображение с заменённым лицом.
+    Возвращает обработанное изображение с максимальным качеством.
     """
-    # Создаём временные файлы
     temp_dir = tempfile.mkdtemp()
     source_path = os.path.join(temp_dir, f"source_{uuid.uuid4().hex}.jpg")
     target_path = os.path.join(temp_dir, f"target_{uuid.uuid4().hex}.jpg")
@@ -42,27 +52,32 @@ async def swap_face(source: UploadFile = File(...), target: UploadFile = File(..
         with open(target_path, "wb") as f:
             f.write(await target.read())
 
-        # Запускаем FaceFusion
-        # Предполагается, что facefusion.py принимает аргументы:
-        # --source SOURCE --target TARGET --output OUTPUT [другие параметры]
+        # Формируем команду с максимальными параметрами
         cmd = [
             "python", FACEFUSION_SCRIPT,
             "--source", source_path,
             "--target", target_path,
             "--output", output_path,
-            "--headless"  # если есть такой флаг
+            "--face-swapper-model", FACE_SWAPPER_MODEL,
+            "--face-enhancer-model", FACE_ENHANCER_MODEL,
+            "--face-enhancer-blend", str(FACE_ENHANCER_BLEND),
+            "--face-mask-types", *FACE_MASK_TYPES,
+            "--face-mask-blur", str(FACE_MASK_BLUR),
+            "--output-image-resolution", str(OUTPUT_IMAGE_RESOLUTION),
+            "--execution-providers", *EXECUTION_PROVIDERS,
+            "--headless"
         ]
+
         logger.info(f"Запуск команды: {' '.join(cmd)}")
         result = subprocess.run(cmd, capture_output=True, text=True, timeout=120)
+
         if result.returncode != 0:
             logger.error(f"FaceFusion ошибка: {result.stderr}")
             raise HTTPException(status_code=500, detail="FaceFusion processing failed")
 
-        # Проверяем, создался ли выходной файл
         if not os.path.exists(output_path):
             raise HTTPException(status_code=500, detail="Output file not created")
 
-        # Читаем и возвращаем результат
         with open(output_path, "rb") as f:
             image_data = f.read()
 
@@ -75,5 +90,4 @@ async def swap_face(source: UploadFile = File(...), target: UploadFile = File(..
         logger.exception("Ошибка при обработке")
         raise HTTPException(status_code=500, detail=str(e))
     finally:
-        # Удаляем временные файлы
         shutil.rmtree(temp_dir, ignore_errors=True)