Update app.py

app.py

@@ -1,5 +1,5 @@
 #!/usr/bin/env python3
-"""FaceFusion API для Hugging Face Spaces — финальная версия"""
+"""FaceFusion API — новая версия"""
 
 import os
 import tempfile
@@ -19,32 +19,20 @@ app = FastAPI(title="FaceFusion API")
 FACEFUSION_DIR = "/facefusion"
 FACEFUSION_SCRIPT = os.path.join(FACEFUSION_DIR, "facefusion.py")
 
-if not os.path.exists(FACEFUSION_SCRIPT):
-    logger.error(f"FaceFusion script not found at {FACEFUSION_SCRIPT}")
-else:
-    logger.info(f"FaceFusion script found at {FACEFUSION_SCRIPT}")
-
-# Конфигурация для CPU (правильное значение)
+# Конфигурация для CPU
 EXECUTION_PROVIDERS = ["cpu"]
 
-# Включаем процессоры замены лица и улучшения
-PROCESSORS = ["face_swapper", "face_enhancer"]
-
-# Параметры масок
+# Максимальное качество — теперь параметры работают!
+FACE_SWAPPER_MODEL = "ghost_3_256"        # лучшая замена лица
+FACE_ENHANCER_MODEL = "codeformer"         # лучшее улучшение
+FACE_ENHANCER_BLEND = 80                    # вес улучшения
 FACE_MASK_TYPES = ["box", "occlusion", "region"]
 FACE_MASK_BLUR = 0.7
-
-# Качество выходного изображения (1-100)
 OUTPUT_IMAGE_QUALITY = 95
 
-# Кэширование моделей
-os.environ["FACEFUSION_CACHE"] = "/tmp/facefusion_cache"
-os.makedirs("/tmp/facefusion_cache", exist_ok=True)
-logger.info(f"Cache directory: /tmp/facefusion_cache")
-
 @app.get("/health")
 async def health():
-    return {"status": "ok", "cache": os.listdir("/tmp/facefusion_cache")}
+    return {"status": "ok"}
 
 @app.post("/swap")
 async def swap_face(source: UploadFile = File(...), target: UploadFile = File(...)):
@@ -54,20 +42,21 @@ async def swap_face(source: UploadFile = File(...), target: UploadFile = File(..
     output_path = os.path.join(temp_dir, f"output_{uuid.uuid4().hex}.jpg")
 
     try:
-        logger.info(f"Received source: {source.filename}, target: {target.filename}")
         with open(source_path, "wb") as f:
             f.write(await source.read())
         with open(target_path, "wb") as f:
             f.write(await target.read())
 
-        # Формируем команду с правильными параметрами для этой версии
+        # Теперь все параметры поддерживаются!
         cmd = [
             "python", FACEFUSION_SCRIPT,
             "headless-run",
             "--source", source_path,
             "--target", target_path,
             "--output-path", output_path,
-            "--processors", *PROCESSORS,                # Включаем замену и улучшение
+            "--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-quality", str(OUTPUT_IMAGE_QUALITY),
@@ -84,33 +73,17 @@ async def swap_face(source: UploadFile = File(...), target: UploadFile = File(..
             text=True
         )
 
-        try:
-            stdout, stderr = process.communicate(timeout=180)
-        except subprocess.TimeoutExpired:
-            process.kill()
-            stdout, stderr = process.communicate()
-            logger.error("Process timed out")
-            raise HTTPException(status_code=504, detail="Processing timeout")
-
+        stdout, stderr = process.communicate(timeout=180)
         elapsed = time.time() - start_time
         logger.info(f"Processing took {elapsed:.2f} seconds")
-        if stdout:
-            logger.info(f"STDOUT: {stdout}")
-        if stderr:
-            logger.error(f"STDERR: {stderr}")
 
         if process.returncode != 0:
-            logger.error(f"FaceFusion failed with code {process.returncode}")
+            logger.error(f"FaceFusion error: {stderr}")
             raise HTTPException(status_code=500, detail=f"FaceFusion error: {stderr}")
 
-        if not os.path.exists(output_path):
-            logger.error("Output file not created")
-            raise HTTPException(status_code=500, detail="Output file missing")
-
         with open(output_path, "rb") as f:
             image_data = f.read()
 
-        logger.info("Success, returning image")
         return Response(content=image_data, media_type="image/jpeg")
 
     except Exception as e: