FERMAT_TEST_REST_SERVER/GLASSES_AI_SERVER
4
0
Fork 0
Code Issues Pull Requests Projects Releases 1 Wiki Activity

edit : 이미지생성 api 생성갯수 삭제, 이미지생성 api 데이터를 return해주는 api 추가, 입력이미지 입력받아 검색하는 api 추가

Browse Source
This commit is contained in:
김재웅
2025-08-07 16:54:46 +09:00
parent 5c20b9ccb5
commit 4ef360448a
5 changed files with 151 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
main_rest/app/models.py
View File

@@ -587,9 +587,7 @@ class ImageGenerateReq(BaseModel):
"""
### [Request] image generate request
"""
prompt : str = Field(description='프롬프트', example='검은색 안경')
downloadCount : int = Field(1, description='이미지 생성 갯수', example=1)
prompt : str = Field(description='프롬프트', example='검은색 안경')
class BingCookieSetReq(BaseModel):
"""
@@ -615,6 +613,16 @@ class VectorImageSearchVitReq(BaseModel):
modelType : str = Field(VitModelType.l14, description='pretrained model 타입', example=VitModelType.l14)
indexType : str = Field(VitIndexType.l2, description='인덱스 타입', example=VitIndexType.l2)
searchNum : int = Field(4, description='검색결과 이미지 갯수', example=4)
class VectorImageSearchVitInputImgReq(BaseModel):
"""
### [Request] vector image search vit - input image
"""
inputImage : str = Field(description='base64 이미지', example='')
modelType : str = Field(VitModelType.l14, description='pretrained model 타입', example=VitModelType.l14)
indexType : str = Field(VitIndexType.l2, description='인덱스 타입', example=VitIndexType.l2)
searchNum : int = Field(4, description='검색결과 이미지 갯수', example=4)
class VectorImageSearchVitDataReq(VectorImageSearchVitReq):
@@ -623,6 +631,7 @@ class VectorImageSearchVitDataReq(VectorImageSearchVitReq):
"""
querySend: bool = Field(True, description='쿼리 이미지 전송 여부', example=True)
class VectorImageSearchVitReportReq(BaseModel):
"""
### [Request] vector image search vit request
@@ -631,6 +640,7 @@ class VectorImageSearchVitReportReq(BaseModel):
modelType : str = Field(VitModelType.l14, description='pretrained model 타입', example=VitModelType.l14)
indexType : str = Field(VitIndexType.l2, description='인덱스 타입', example=VitIndexType.l2)
class VectorImageResult(BaseModel):
image : str = Field("", description='이미지 데이터', example='')
percents: float = Field(0.0, description='percents 값', example='')
@@ -665,7 +675,7 @@ class ImageGenerateDataRes(ResponseBase):
"""
### image generate Data response
"""
imageData : int = Field(0, description='실제 이미지 생성 갯수', example=1)
imageData : str = Field('', description='이미지 데이터', example='')
@staticmethod
def set_error(error,img_data=''):