Update app.py

app.py

@@ -1,10 +1,11 @@
 #!/usr/bin/env python3
 """
-🎨 FaceFusion API для Hugging Face Spaces — FINAL FIXED VERSION
-✅ Disable NSFW via ENV variable (not CLI arg)
+🎨 FaceFusion API для Hugging Face Spaces — FINAL VERSION
+✅ Output всегда JPEG независимо от входного формата
 ✅ Read UploadFile once and reuse bytes
-✅ Allow model re-download if corrupted
+✅ NSFW disabled via ENV
 ✅ ghost_3_256 maximum quality
+✅ Fallback на inswapper_128 если ghost не работает
 """
 
 import os, sys, tempfile, subprocess, logging, shutil, time
@@ -13,31 +14,46 @@ from contextlib import asynccontextmanager
 from fastapi import FastAPI, File, UploadFile, HTTPException, Query
 from fastapi.responses import Response
 
-logging.basicConfig(level=logging.INFO, format='%(asctime)s [%(levelname)s] %(message)s', handlers=[logging.StreamHandler(sys.stdout)])
+# 🔥 Настройка логирования
+logging.basicConfig(
+    level=logging.INFO,
+    format='%(asctime)s [%(levelname)s] %(message)s',
+    handlers=[logging.StreamHandler(sys.stdout)]
+)
 logger = logging.getLogger(__name__)
 
 @asynccontextmanager
 async def lifespan(app: FastAPI):
+    """Startup/Shutdown события"""
     logger.info("🚀 FaceFusion CPU API (ghost_3_256) starting...")
     logger.info("📁 Cache: /tmp/facefusion_cache")
     logger.info("⚙️ Default: ghost_3_256 + gfpgan_1.4")
     logger.info("🔕 NSFW filter: disabled via ENV")
     logger.info("🔄 Model re-download: enabled")
+    logger.info("📐 Output format: always JPEG")
     logger.info("⏱️ Timeout: 900s")
     logger.info("✅ API ready!")
     yield
     logger.info("👋 API shutting down...")
 
-app = FastAPI(title="FaceFusion CPU API — High Quality", version="2.2", lifespan=lifespan)
+app = FastAPI(
+    title="FaceFusion CPU API — High Quality",
+    version="2.4",
+    lifespan=lifespan
+)
 
+# 🔥 Пути
 FACEFUSION_DIR = Path("/facefusion")
 FACEFUSION_PY = FACEFUSION_DIR / "facefusion.py"
 CACHE_DIR = Path("/tmp/facefusion_cache")
 CACHE_DIR.mkdir(parents=True, exist_ok=True)
 
+# 🔥 Проверка установки
 if not FACEFUSION_PY.exists():
-    raise RuntimeError(f"FaceFusion not found at {FACEFUSION_PY}")
+    logger.error(f"❌ FaceFusion not found at {FACEFUSION_PY}")
+    raise RuntimeError("FaceFusion not installed — check Dockerfile")
 
+# 🔥 КОНФИГУРАЦИЯ
 CONFIG = {
     "face_swapper": "ghost_3_256",
     "face_swapper_fallback": "inswapper_128",
@@ -46,42 +62,83 @@ CONFIG = {
     "face_mask_types": ["box", "occlusion"],
     "face_mask_blur": 0.6,
     "face_mask_padding": [10, 10, 10, 10],
-    "output_image_resolution": "768",
     "output_image_quality": "90",
     "timeout_high_quality": 900,
     "timeout_standard": 600,
-    "skip_download": False,  # 🔥 Разрешить перескачку если модель повреждена
+    "skip_download": False,
     "log_level": "error",
 }
 
 # 🔥 Отключаем NSFW фильтр через ENV переменную
 os.environ["FACEFUSION_CONTENT_ANALYSER"] = "none"
 
+# 🔥 Доступные модели
 AVAILABLE_SWAPPERS = ["ghost_3_256", "ghost_2_256", "ghost_1_256", "inswapper_128", "simswap_256"]
 
+# ============================================
+# 🔥 ЭНДПОИНТЫ
+# ============================================
+
 @app.get("/")
 async def root():
+    """Информация о сервисе"""
     return {
         "service": "FaceFusion CPU API — High Quality",
-        "version": "2.2",
+        "version": "2.4",
         "default_swapper": CONFIG["face_swapper"],
+        "default_enhancer": CONFIG["face_enhancer"],
+        "output_format": "jpeg (always)",
         "nsfw_disabled": os.environ.get("FACEFUSION_CONTENT_ANALYSER") == "none",
+        "available_swappers": AVAILABLE_SWAPPERS,
         "status": "running"
     }
 
 @app.get("/health")
 async def health():
+    """Health check для мониторинга"""
     return {"status": "ok"}
 
 @app.post("/swap")
 async def swap_face(
-    source: UploadFile = File(...),
-    target: UploadFile = File(...),
-    swapper_model: str = Query(default="ghost_3_256", enum=AVAILABLE_SWAPPERS),
-    enhancer_model: str = Query(default="gfpgan_1.4"),
-    enhance_blend: int = Query(default=75, ge=0, le=100),
-    use_fallback: bool = Query(default=True)
+    source: UploadFile = File(..., description="Фото лица для замены (анфас, хорошее освещение)"),
+    target: UploadFile = File(..., description="Целевое изображение"),
+    swapper_model: str = Query(
+        default="ghost_3_256",
+        description="Модель замены лица",
+        enum=AVAILABLE_SWAPPERS
+    ),
+    enhancer_model: str = Query(
+        default="gfpgan_1.4",
+        description="Модель улучшения лица (или 'none')"
+    ),
+    enhance_blend: int = Query(
+        default=75,
+        ge=0, le=100,
+        description="Сила blending улучшения (0-100)"
+    ),
+    use_fallback: bool = Query(
+        default=True,
+        description="Использовать fallback модель если основная не работает"
+    )
 ):
+    """
+    🎨 Замена лица с высоким качеством
+    
+    **Модели:**
+    - `ghost_3_256` — лучшее качество (256px, рекомендуется)
+    - `inswapper_128` — стандартное (128px, быстрее)
+    
+    **Улучшение лица:**
+    - `gfpgan_1.4` — хороший баланс качества/скорости
+    - `none` — без улучшения
+    
+    **Время обработки:**
+    - ghost_3_256: 60-120 сек
+    - inswapper_128: 30-60 сек
+    - +enhancer: +20-40 сек
+    
+    **Выход:** Всегда JPEG независимо от входного формата
+    """
     # 🔥 ЧИТАЕМ ФАЙЛЫ ОДИН РАЗ И СОХРАНЯЕМ В ПАМЯТЬ
     try:
         source_bytes = await source.read()
@@ -94,9 +151,11 @@ async def swap_face(
     
     logger.info(f"📥 Request: source={len(source_bytes)}b, target={len(target_bytes)}b")
     
+    # Нормализуем enhancer_model
     if enhancer_model and enhancer_model.lower() == "none":
         enhancer_model = None
     
+    # Формируем список моделей для попытки
     models_to_try = [swapper_model]
     if use_fallback and swapper_model != CONFIG["face_swapper_fallback"]:
         models_to_try.append(CONFIG["face_swapper_fallback"])
@@ -110,8 +169,9 @@ async def swap_face(
             result = await _process_face_swap(
                 source_bytes=source_bytes,
                 target_bytes=target_bytes,
+                target_filename=target.filename,
                 swapper_model=model,
-                enhancer_model=enhancer_model if attempt == 1 else None,
+                enhancer_model=enhancer_model if attempt == 1 else None,  # Без enhancer для fallback
                 enhance_blend=enhance_blend,
                 timeout=CONFIG["timeout_high_quality"] if model.startswith("ghost") else CONFIG["timeout_standard"]
             )
@@ -126,59 +186,81 @@ async def swap_face(
     
     raise HTTPException(500, "Unexpected error")
 
+# ============================================
+# 🔥 ВНУТРЕННЯЯ ФУНКЦИЯ ОБРАБОТКИ
+# ============================================
+
 async def _process_face_swap(
     source_bytes: bytes,
     target_bytes: bytes,
+    target_filename: str,
     swapper_model: str,
     enhancer_model: str,
     enhance_blend: int,
     timeout: int
 ):
+    """Обработка замены лица"""
     temp_dir = None
     
     try:
         temp_dir = tempfile.mkdtemp(prefix="ff_")
+        
+        # 🔥 Входные файлы (любой формат)
         source_path = Path(temp_dir) / "source.jpg"
         target_path = Path(temp_dir) / "target.jpg"
-        output_path = Path(temp_dir) / "output.png"
         
-        logger.info(f"🔧 Processing: swapper={swapper_model}, enhancer={enhancer_model}")
+        # 🔥 ВЫХОД ВСЕГДА .jpg
+        output_path = Path(temp_dir) / "output.jpg"
+        
+        logger.info(f"🔧 Processing: swapper={swapper_model}, enhancer={enhancer_model}, output=jpg")
         
+        # Записываем байты
         source_path.write_bytes(source_bytes)
         target_path.write_bytes(target_bytes)
         
+        # Формируем список процессоров
         processors = ["face_swapper"]
         if enhancer_model:
             processors.append("face_enhancer")
         
-        # 🔥 CLI команда БЕЗ --skip-nsfw-filter (не поддерживается)
+        # 🔥 CLI команда
         cmd = [
             "python", str(FACEFUSION_PY),
             "headless-run",
+            
+            # Файлы
             "-s", str(source_path),
             "-t", str(target_path),
-            "-o", str(output_path),
+            "-o", str(output_path),  # 🔥 Всегда .jpg!
+            
+            # Процессоры и модель
             "--processors", *processors,
             "--face-swapper-model", swapper_model,
+            
+            # Маски для качественного blending
             "--face-mask-types", *CONFIG["face_mask_types"],
             "--face-mask-blur", str(CONFIG["face_mask_blur"]),
             "--face-mask-padding", *map(str, CONFIG["face_mask_padding"]),
-            "--output-image-resolution", CONFIG["output_image_resolution"],
+            
+            # Качество
             "--output-image-quality", CONFIG["output_image_quality"],
-            # ❌ УБРАЛ: --skip-nsfw-filter (не поддерживается в этой версии)
+            
+            # Логирование
             "--log-level", CONFIG["log_level"],
         ]
         
-        # 🔥 skip-download=False по умолчанию — разрешаем перескачку моделей
+        # Skip download если нужно
         if CONFIG["skip_download"]:
             cmd.append("--skip-download")
         
+        # Enhancer если указан
         if enhancer_model:
             cmd.extend([
                 "--face-enhancer-model", enhancer_model,
                 "--face-enhancer-blend", str(enhance_blend),
             ])
         
+        # Убираем None значения
         cmd = [arg for arg in cmd if arg is not None]
         
         logger.info(f"🚀 Running: {' '.join(cmd)}")
@@ -188,8 +270,12 @@ async def _process_face_swap(
         env = {**os.environ, "OMP_NUM_THREADS": "4", "FACEFUSION_CONTENT_ANALYSER": "none"}
         
         process = subprocess.Popen(
-            cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE,
-            text=True, cwd=str(FACEFUSION_DIR), env=env
+            cmd,
+            stdout=subprocess.PIPE,
+            stderr=subprocess.PIPE,
+            text=True,
+            cwd=str(FACEFUSION_DIR),
+            env=env
         )
         
         try:
@@ -197,48 +283,51 @@ async def _process_face_swap(
         except subprocess.TimeoutExpired:
             process.kill()
             process.communicate()
-            raise HTTPException(504, f"⏱️ Timeout: >{timeout}s")
+            logger.error(f"⏱️ Timeout after {timeout}s")
+            raise HTTPException(504, f"⏱️ Timeout: обработка заняла больше {timeout} секунд")
         
         elapsed = time.time() - process_start
         logger.info(f"⏱️ Done in {elapsed:.1f}s, code={process.returncode}")
         
+        # 🔥 Фильтруем harmless warnings из stderr
         if stderr:
-            # Фильтруем harmless warnings
-            stderr_clean = '\n'.join(l for l in stderr.split('\n') if 'iCCP' not in l and 'libpng' not in l)
-            if stderr_clean.strip():
+            stderr_clean = '\n'.join(
+                l for l in stderr.split('\n') 
+                if 'iCCP' not in l and 'libpng' not in l.lower()
+            ).strip()
+            if stderr_clean:
                 logger.info(f"📋 STDERR: {stderr_clean[-400:]}")
         
+        # 🔥 Проверяем результат
         if process.returncode != 0:
             error_msg = stderr.strip()[-400:] if stderr else "Unknown error"
             logger.error(f"❌ FaceFusion failed: {error_msg}")
             raise RuntimeError(f"FaceFusion error: {error_msg}")
         
-        # Ищем результат
-        result_file = None
-        for ext in [".png", ".jpg", ".jpeg", ".webp"]:
-            candidate = output_path.with_suffix(ext)
-            if candidate.exists():
-                result_file = candidate
-                break
-        
+        # 🔥 Ищем выходной файл
+        result_file = output_path if output_path.exists() else None
         if not result_file:
-            for f in Path(temp_dir).glob("output*"):
-                if f.suffix.lower() in [".png", ".jpg", ".jpeg", ".webp"]:
-                    result_file = f
-                    break
+            # Fallback: ищем любой output*.jpg
+            for f in Path(temp_dir).glob("output*.jpg"):
+                result_file = f
+                break
         
         if not result_file or not result_file.exists():
             raise RuntimeError("Output image not created")
         
+        # 🔥 Читаем и возвращаем результат
         image_data = result_file.read_bytes()
+        
         logger.info(f"✅ Success: {len(image_data)/1024:.1f}KB in {elapsed:.1f}s")
         
         return Response(
             content=image_data,
-            media_type="image/png",
+            media_type="image/jpeg",  # 🔥 ВСЕГДА JPEG!
             headers={
                 "X-Processing-Time": f"{elapsed:.1f}",
                 "X-Swapper-Model": swapper_model,
+                "X-Enhancer-Model": enhancer_model or "none",
+                "X-Output-Format": "jpeg",
                 "Cache-Control": "no-store"
             }
         )
@@ -249,17 +338,66 @@ async def _process_face_swap(
         logger.exception(f"💥 Error: {e}")
         raise
     finally:
+        # 🔥 Очистка временных файлов
         if temp_dir and Path(temp_dir).exists():
-            shutil.rmtree(temp_dir, ignore_errors=True)
+            try:
+                shutil.rmtree(temp_dir, ignore_errors=True)
+                logger.debug(f"🧹 Cleaned: {temp_dir}")
+            except Exception as e:
+                logger.warning(f"⚠️ Could not clean: {e}")
+
+# ============================================
+# 🔥 УПРОЩЁННЫЕ ЭНДПОИНТЫ
+# ============================================
 
 @app.post("/swap/quick")
-async def swap_quick(source: UploadFile = File(...), target: UploadFile = File(...)):
-    return await swap_face(source=source, target=target, swapper_model="inswapper_128", enhancer_model="none", use_fallback=False)
+async def swap_quick(
+    source: UploadFile = File(..., description="Фото лица"),
+    target: UploadFile = File(..., description="Целевое изображение")
+):
+    """
+    ⚡ Быстрая замена лица (inswapper_128 без enhancer)
+    Время: ~30-60 секунд
+    Выход: JPEG
+    """
+    return await swap_face(
+        source=source,
+        target=target,
+        swapper_model="inswapper_128",
+        enhancer_model="none",
+        use_fallback=False
+    )
 
 @app.post("/swap/hq")
-async def swap_hq(source: UploadFile = File(...), target: UploadFile = File(...), with_enhancer: bool = True):
-    return await swap_face(source=source, target=target, swapper_model="ghost_3_256", enhancer_model="gfpgan_1.4" if with_enhancer else "none", enhance_blend=80, use_fallback=True)
+async def swap_hq(
+    source: UploadFile = File(..., description="Фото лица"),
+    target: UploadFile = File(..., description="Целевое изображение"),
+    with_enhancer: bool = Query(default=True, description="Включить улучшение лица GFPGAN")
+):
+    """
+    ⭐ Максимальное качество (ghost_3_256 + gfpgan_1.4)
+    Время: ~90-180 секунд
+    Выход: JPEG
+    """
+    return await swap_face(
+        source=source,
+        target=target,
+        swapper_model="ghost_3_256",
+        enhancer_model="gfpgan_1.4" if with_enhancer else "none",
+        enhance_blend=80,
+        use_fallback=True
+    )
+
+# ============================================
+# 🔥 ЗАПУСК
+# ============================================
 
 if __name__ == "__main__":
     import uvicorn
-    uvicorn.run(app, host="0.0.0.0", port=7860, log_level="info", timeout_keep_alive=900)
+    uvicorn.run(
+        app,
+        host="0.0.0.0",
+        port=7860,
+        log_level="info",
+        timeout_keep_alive=900  # 15 минут keep-alive
+    )