Upload api_only.py

iopaint/api_only.py

@@ -7,7 +7,7 @@ import uvicorn
 from loguru import logger
 
 from iopaint.api import Api
-from iopaint.schema import ApiConfig, Device
+from iopaint.schema import ApiConfig, Device, InteractiveSegModel, RealESRGANModel, RemoveBGModel
 from iopaint.runtime import setup_model_dir, check_device, dump_environment_info
 from iopaint.const import DEFAULT_MODEL_DIR
 
@@ -15,14 +15,26 @@ from iopaint.const import DEFAULT_MODEL_DIR
 host = os.environ.get("IOPAINT_HOST", "0.0.0.0")
 port = int(os.environ.get("IOPAINT_PORT", "7860"))
 model = os.environ.get("IOPAINT_MODEL", "lama")
-model_dir_str = os.environ.get("IOPAINT_MODEL_DIR", str(DEFAULT_MODEL_DIR))
+
+# 修改模型目录路径，使用/app或/tmp目录
+model_dir_str = os.environ.get("IOPAINT_MODEL_DIR", "/app/models")
+
 device_str = os.environ.get("IOPAINT_DEVICE", "cpu")
 api_key = os.environ.get("IOPAINT_API_KEY", None)
 allowed_origins = os.environ.get("ALLOWED_ORIGINS", "*").split(",")
 
 # 初始化目录和环境
 model_dir = Path(model_dir_str)
-model_dir.mkdir(parents=True, exist_ok=True)
+try:
+    model_dir.mkdir(parents=True, exist_ok=True)
+    logger.info(f"Successfully created model directory: {model_dir}")
+except Exception as e:
+    logger.error(f"Failed to create model directory: {e}")
+    # 如果失败，尝试使用/tmp目录
+    model_dir = Path("/tmp/iopaint/models")
+    model_dir.mkdir(parents=True, exist_ok=True)
+    logger.info(f"Using alternative model directory: {model_dir}")
+
 device = check_device(Device(device_str))
 dump_environment_info()
 
@@ -33,7 +45,7 @@ logger.info(f"Allowed origins: {allowed_origins}")
 # 初始化FastAPI
 app = FastAPI(title="IOPaint API")
 
-# 配置API
+# 配置API，添加所有缺失的必填字段
 config = ApiConfig(
     host=host,
     port=port,
@@ -46,6 +58,21 @@ config = ApiConfig(
     no_half=os.environ.get("IOPAINT_NO_HALF", "false").lower() == "true",
     cpu_offload=os.environ.get("IOPAINT_CPU_OFFLOAD", "false").lower() == "true",
     disable_nsfw=os.environ.get("IOPAINT_DISABLE_NSFW", "false").lower() == "true",
+    # 添加缺失的必填字段
+    enable_interactive_seg=False,
+    interactive_seg_model=InteractiveSegModel.sam2_1_tiny,
+    interactive_seg_device=Device.cpu,
+    enable_remove_bg=False,
+    remove_bg_device=Device.cpu,
+    remove_bg_model=RemoveBGModel.briaai_rmbg_1_4,
+    enable_anime_seg=False,
+    enable_realesrgan=False,
+    realesrgan_device=Device.cpu,
+    realesrgan_model=RealESRGANModel.realesr_general_x4v3,
+    enable_gfpgan=False,
+    gfpgan_device=Device.cpu,
+    enable_restoreformer=False,
+    restoreformer_device=Device.cpu,
 )
 
 # 配置CORS