ailabtools-sdk 0.1.0__py3-none-any.whl

ailabtools/__init__.py

@@ -0,0 +1,4 @@
+from .client import AILabClient
+from .errors import AILabApiError, ErrorDetail
+
+from .generated.endpoints import *

ailabtools/client.py

@@ -0,0 +1,17 @@
+from .types import ClientOptions
+from .http import Requester
+from .common import CommonAPI
+from .image import ImageAPI
+from .portrait import PortraitAPI
+from .cutout import CutoutAPI
+
+class AILabClient:
+    def __init__(self, api_key: str, base_url: str = "https://www.ailabapi.com", timeout: float = 30.0):
+        self._requester = Requester(api_key=api_key, base_url=base_url, timeout=timeout)
+        self.common = CommonAPI(self._requester)
+        self.image = ImageAPI(self._requester)
+        self.portrait = PortraitAPI(self._requester)
+        self.cutout = CutoutAPI(self._requester)
+
+    async def aclose(self):
+        await self._requester.aclose()

ailabtools/common.py

@@ -0,0 +1,26 @@
+from .http import Requester
+from .utils import map_params
+from .generated.endpoints import (
+    CommonQueryAsyncTaskResultParams,
+    CommonQueryAsyncTaskResultResponse,
+    CommonQueryAsyncTaskResultParamMap,
+    validate_commonQueryAsyncTaskResult_params,
+    CommonQueryCreditsParams,
+    CommonQueryCreditsResponse,
+    CommonQueryCreditsParamMap,
+    validate_commonQueryCredits_params
+)
+
+class CommonAPI:
+    def __init__(self, requester: Requester):
+        self._requester = requester
+
+    async def commonQueryAsyncTaskResult(self, params: CommonQueryAsyncTaskResultParams) -> CommonQueryAsyncTaskResultResponse:
+        validate_commonQueryAsyncTaskResult_params(params)
+        mapped = map_params(params, CommonQueryAsyncTaskResultParamMap)
+        return await self._requester.request('GET', '/api/common/query-async-task-result', query=mapped)
+
+    async def commonQueryCredits(self, params: CommonQueryCreditsParams) -> CommonQueryCreditsResponse:
+        validate_commonQueryCredits_params(params)
+        mapped = map_params(params, CommonQueryCreditsParamMap)
+        return await self._requester.request('GET', '/api/common/query-credits', query=mapped)

ailabtools/cutout.py

@@ -0,0 +1,71 @@
+from .http import Requester
+from .utils import map_params
+from .generated.endpoints import (
+    CutoutClothingBackgroundRemovalParams,
+    CutoutClothingBackgroundRemovalResponse,
+    CutoutClothingBackgroundRemovalParamMap,
+    validate_cutoutClothingBackgroundRemoval_params,
+    CutoutProductBackgroundRemovalParams,
+    CutoutProductBackgroundRemovalResponse,
+    CutoutProductBackgroundRemovalParamMap,
+    validate_cutoutProductBackgroundRemoval_params,
+    CutoutFoodBackgroundRemovalParams,
+    CutoutFoodBackgroundRemovalResponse,
+    CutoutFoodBackgroundRemovalParamMap,
+    validate_cutoutFoodBackgroundRemoval_params,
+    CutoutUniversalBackgroundRemovalParams,
+    CutoutUniversalBackgroundRemovalResponse,
+    CutoutUniversalBackgroundRemovalParamMap,
+    validate_cutoutUniversalBackgroundRemoval_params,
+    CutoutAvatarExtractionParams,
+    CutoutAvatarExtractionResponse,
+    CutoutAvatarExtractionParamMap,
+    validate_cutoutAvatarExtraction_params,
+    CutoutHairExtractionParams,
+    CutoutHairExtractionResponse,
+    CutoutHairExtractionParamMap,
+    validate_cutoutHairExtraction_params,
+    CutoutHumanBackgroundRemovalParams,
+    CutoutHumanBackgroundRemovalResponse,
+    CutoutHumanBackgroundRemovalParamMap,
+    validate_cutoutHumanBackgroundRemoval_params
+)
+
+class CutoutAPI:
+    def __init__(self, requester: Requester):
+        self._requester = requester
+
+    async def cutoutClothingBackgroundRemoval(self, params: CutoutClothingBackgroundRemovalParams) -> CutoutClothingBackgroundRemovalResponse:
+        validate_cutoutClothingBackgroundRemoval_params(params)
+        mapped = map_params(params, CutoutClothingBackgroundRemovalParamMap)
+        return await self._requester.request('POST', '/api/cutout/general/apparel-background-removal', body=mapped, multipart=True)
+
+    async def cutoutProductBackgroundRemoval(self, params: CutoutProductBackgroundRemovalParams) -> CutoutProductBackgroundRemovalResponse:
+        validate_cutoutProductBackgroundRemoval_params(params)
+        mapped = map_params(params, CutoutProductBackgroundRemovalParamMap)
+        return await self._requester.request('POST', '/api/cutout/general/commodity-background-removal', body=mapped, multipart=True)
+
+    async def cutoutFoodBackgroundRemoval(self, params: CutoutFoodBackgroundRemovalParams) -> CutoutFoodBackgroundRemovalResponse:
+        validate_cutoutFoodBackgroundRemoval_params(params)
+        mapped = map_params(params, CutoutFoodBackgroundRemovalParamMap)
+        return await self._requester.request('POST', '/api/cutout/general/food-background-removal', body=mapped, multipart=True)
+
+    async def cutoutUniversalBackgroundRemoval(self, params: CutoutUniversalBackgroundRemovalParams) -> CutoutUniversalBackgroundRemovalResponse:
+        validate_cutoutUniversalBackgroundRemoval_params(params)
+        mapped = map_params(params, CutoutUniversalBackgroundRemovalParamMap)
+        return await self._requester.request('POST', '/api/cutout/general/universal-background-removal', body=mapped, multipart=True)
+
+    async def cutoutAvatarExtraction(self, params: CutoutAvatarExtractionParams) -> CutoutAvatarExtractionResponse:
+        validate_cutoutAvatarExtraction_params(params)
+        mapped = map_params(params, CutoutAvatarExtractionParamMap)
+        return await self._requester.request('POST', '/api/cutout/portrait/avatar-extraction', body=mapped, multipart=True)
+
+    async def cutoutHairExtraction(self, params: CutoutHairExtractionParams) -> CutoutHairExtractionResponse:
+        validate_cutoutHairExtraction_params(params)
+        mapped = map_params(params, CutoutHairExtractionParamMap)
+        return await self._requester.request('POST', '/api/cutout/portrait/hairstyle-extraction', body=mapped, multipart=True)
+
+    async def cutoutHumanBackgroundRemoval(self, params: CutoutHumanBackgroundRemovalParams) -> CutoutHumanBackgroundRemovalResponse:
+        validate_cutoutHumanBackgroundRemoval_params(params)
+        mapped = map_params(params, CutoutHumanBackgroundRemovalParamMap)
+        return await self._requester.request('POST', '/api/cutout/portrait/portrait-background-removal', body=mapped, multipart=True)

ailabtools/endpoints.py

@@ -0,0 +1,110 @@
+CommonQueryAsyncTaskResultParams = dict
+CommonQueryAsyncTaskResultResponse = dict
+CommonQueryCreditsParams = dict
+CommonQueryCreditsResponse = dict
+ImageLosslessEnlargementParams = dict
+ImageLosslessEnlargementResponse = dict
+ImageDistortionCorrectionParams = dict
+ImageDistortionCorrectionResponse = dict
+ImageClarityEnhancementParams = dict
+ImageClarityEnhancementResponse = dict
+ImageContrastEnhancementParams = dict
+ImageContrastEnhancementResponse = dict
+ImageColorEnhancementParams = dict
+ImageColorEnhancementResponse = dict
+ImageColoringParams = dict
+ImageColoringResponse = dict
+ImageStyleTransferParams = dict
+ImageStyleTransferResponse = dict
+ImageStyleMigrationParams = dict
+ImageStyleMigrationResponse = dict
+ImageColorMigrationParams = dict
+ImageColorMigrationResponse = dict
+ImageHdColorMigrationParams = dict
+ImageHdColorMigrationResponse = dict
+ImageIntelligentCompositionParams = dict
+ImageIntelligentCompositionResponse = dict
+ImagePhotoEditingParams = dict
+ImagePhotoEditingResponse = dict
+ImageInvisibleImageWatermarkParams = dict
+ImageInvisibleImageWatermarkResponse = dict
+ImageInvisibleTextWatermarkParams = dict
+ImageInvisibleTextWatermarkResponse = dict
+ImageErasureParams = dict
+ImageErasureResponse = dict
+ImageRemoveObjectsParams = dict
+ImageRemoveObjectsResponse = dict
+ImageRemoveObjectsAdvancedParams = dict
+ImageRemoveObjectsAdvancedResponse = dict
+ImageRemoveObjectsProParams = dict
+ImageRemoveObjectsProResponse = dict
+ImageAiObjectReplacerParams = dict
+ImageAiObjectReplacerResponse = dict
+ImageAiImageExtenderParams = dict
+ImageAiImageExtenderResponse = dict
+ImageExposureRatingParams = dict
+ImageExposureRatingResponse = dict
+PortraitFaceFusionParams = dict
+PortraitFaceFusionResponse = dict
+PortraitIntelligentBeautificationParams = dict
+PortraitIntelligentBeautificationResponse = dict
+PortraitIntelligentSkinRetouchingParams = dict
+PortraitIntelligentSkinRetouchingResponse = dict
+PortraitIntelligentFaceSlimmingParams = dict
+PortraitIntelligentFaceSlimmingResponse = dict
+PortraitFacialBeautificationParams = dict
+PortraitFacialBeautificationResponse = dict
+PortraitFacialBeautificationAdvancedParams = dict
+PortraitFacialBeautificationAdvancedResponse = dict
+PortraitFacialBeautificationProParams = dict
+PortraitFacialBeautificationProResponse = dict
+PortraitFacialMakeupParams = dict
+PortraitFacialMakeupResponse = dict
+PortraitFacialFiltersParams = dict
+PortraitFacialFiltersResponse = dict
+PortraitFacialShapingParams = dict
+PortraitFacialShapingResponse = dict
+PortraitExpressionEditingParams = dict
+PortraitExpressionEditingResponse = dict
+PortraitHairstyleEditingParams = dict
+PortraitHairstyleEditingResponse = dict
+PortraitHairstyleEditingProParams = dict
+PortraitHairstyleEditingProResponse = dict
+PortraitLivePhotosParams = dict
+PortraitLivePhotosResponse = dict
+PortraitHitchcockEffectParams = dict
+PortraitHitchcockEffectResponse = dict
+PortraitFacialBlurringParams = dict
+PortraitFacialBlurringResponse = dict
+PortraitFaceAttributeEditingParams = dict
+PortraitFaceAttributeEditingResponse = dict
+PortraitFaceRestorationEnhancementParams = dict
+PortraitFaceRestorationEnhancementResponse = dict
+PortraitLipsColorChangerParams = dict
+PortraitLipsColorChangerResponse = dict
+PortraitSkinDiseaseDetectionParams = dict
+PortraitSkinDiseaseDetectionResponse = dict
+PortraitSkinAnalysisBasicParams = dict
+PortraitSkinAnalysisBasicResponse = dict
+PortraitSkinAnalysisAdvancedParams = dict
+PortraitSkinAnalysisAdvancedResponse = dict
+PortraitSkinAnalysisProfessionalParams = dict
+PortraitSkinAnalysisProfessionalResponse = dict
+PortraitFaceAnalyzerParams = dict
+PortraitFaceAnalyzerResponse = dict
+PortraitFaceAnalyzerAdvancedParams = dict
+PortraitFaceAnalyzerAdvancedResponse = dict
+PortraitTryOnClothesProParams = dict
+PortraitTryOnClothesProResponse = dict
+CutoutFoodBackgroundRemovalParams = dict
+CutoutFoodBackgroundRemovalResponse = dict
+CutoutProductBackgroundRemovalParams = dict
+CutoutProductBackgroundRemovalResponse = dict
+CutoutClothingBackgroundRemovalParams = dict
+CutoutClothingBackgroundRemovalResponse = dict
+CutoutHdHumanBodyBackgroundRemovalParams = dict
+CutoutHdHumanBodyBackgroundRemovalResponse = dict
+CutoutAvatarExtractionParams = dict
+CutoutAvatarExtractionResponse = dict
+CutoutHairExtractionParams = dict
+CutoutHairExtractionResponse = dict

ailabtools/errors.py

@@ -0,0 +1,17 @@
+from dataclasses import dataclass
+from typing import Optional, Any
+
+@dataclass
+class ErrorDetail:
+    status_code: Optional[int] = None
+    code: Optional[str] = None
+    code_message: Optional[str] = None
+    message: Optional[str] = None
+
+class AILabApiError(Exception):
+    def __init__(self, message: str, detail: Optional[ErrorDetail] = None, request_id: Optional[str] = None, log_id: Optional[str] = None, raw: Any = None):
+        super().__init__(message)
+        self.detail = detail
+        self.request_id = request_id
+        self.log_id = log_id
+        self.raw = raw