Create app.py

app.py

@@ -0,0 +1,145 @@
+import os
+import subprocess
+import tempfile
+import uuid
+import json
+import time
+import requests
+from fastapi import FastAPI, File, UploadFile, HTTPException
+from fastapi.responses import Response, JSONResponse
+import uvicorn
+import logging
+
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
+app = FastAPI()
+
+# Запускаем ComfyUI в фоновом режиме
+comfy_process = None
+
+@app.on_event("startup")
+async def startup_event():
+    global comfy_process
+    logger.info("Starting ComfyUI...")
+    comfy_process = subprocess.Popen(
+        ["python", "/comfyui/main.py", "--listen", "0.0.0.0", "--port", "8188"],
+        stdout=subprocess.PIPE,
+        stderr=subprocess.PIPE
+    )
+    # Даём время на запуск
+    time.sleep(10)
+    logger.info("ComfyUI started")
+
+@app.on_event("shutdown")
+async def shutdown_event():
+    if comfy_process:
+        comfy_process.terminate()
+
+def load_workflow():
+    """Загружает workflow из файла или создаёт базовый"""
+    # Здесь можно загрузить ваш workflow.json
+    # Пока используем простой workflow
+    return {
+        "3": {
+            "class_type": "LoadImage",
+            "inputs": {
+                "image": "source.jpg"
+            }
+        },
+        "4": {
+            "class_type": "LoadImage",
+            "inputs": {
+                "image": "target.jpg"
+            }
+        },
+        "5": {
+            "class_type": "ReActorFaceSwap",
+            "inputs": {
+                "source_image": ["3", 0],
+                "target_image": ["4", 0],
+                "face_restorer": "gfpgan",
+                "face_restorer_weight": 0.8,
+                "swap_model": "inswapper_128.onnx",
+                "detect_model": "yolov8n-face.pt",
+                "save_original": False,
+                "output_image": ["6", 0]
+            }
+        },
+        "6": {
+            "class_type": "SaveImage",
+            "inputs": {
+                "filename_prefix": "output",
+                "images": ["5", 0]
+            }
+        }
+    }
+
+@app.post("/swap")
+async def swap_face(
+    source: UploadFile = File(...),
+    target: UploadFile = File(...)
+):
+    temp_dir = tempfile.mkdtemp()
+    try:
+        # Сохраняем входные файлы
+        source_path = os.path.join(temp_dir, "source.jpg")
+        target_path = os.path.join(temp_dir, "target.jpg")
+        
+        with open(source_path, "wb") as f:
+            f.write(await source.read())
+        with open(target_path, "wb") as f:
+            f.write(await target.read())
+
+        # Копируем файлы в папку ComfyUI
+        comfy_input_dir = "/comfyui/input"
+        os.makedirs(comfy_input_dir, exist_ok=True)
+        
+        shutil.copy(source_path, os.path.join(comfy_input_dir, "source.jpg"))
+        shutil.copy(target_path, os.path.join(comfy_input_dir, "target.jpg"))
+
+        # Загружаем workflow
+        workflow = load_workflow()
+        
+        # Отправляем задачу в ComfyUI
+        response = requests.post(
+            "http://127.0.0.1:8188/prompt",
+            json={"prompt": workflow}
+        )
+        
+        if response.status_code != 200:
+            raise HTTPException(500, f"ComfyUI error: {response.text}")
+        
+        prompt_id = response.json()["prompt_id"]
+        
+        # Ждём завершения
+        while True:
+            status = requests.get(f"http://127.0.0.1:8188/history/{prompt_id}")
+            if status.status_code == 200 and status.json():
+                history = status.json()
+                if prompt_id in history:
+                    output_images = history[prompt_id]["outputs"]
+                    # Находим выходное изображение
+                    for node_id, node_output in output_images.items():
+                        if "images" in node_output:
+                            image_info = node_output["images"][0]
+                            image_path = os.path.join("/comfyui/output", image_info["filename"])
+                            if os.path.exists(image_path):
+                                with open(image_path, "rb") as f:
+                                    image_data = f.read()
+                                return Response(content=image_data, media_type="image/jpeg")
+            time.sleep(1)
+            
+    except Exception as e:
+        logger.exception("Error")
+        raise HTTPException(500, str(e))
+    finally:
+        import shutil
+        shutil.rmtree(temp_dir, ignore_errors=True)
+
+@app.get("/health")
+async def health():
+    return {"status": "ok"}
+
+if __name__ == "__main__":
+    uvicorn.run(app, host="0.0.0.0", port=7860)