syntax = "proto3";

package ai;
option go_package = "./pb";


// 人脸识别
message FaceRecognitionRequest {
  bytes  face = 1;
  string user_id = 2;
}
message FaceRecognitionResponse {
  int64 face_id = 3;
}
// tf分类识别
message TfClassificationRequest {
  bytes  image = 1;
}
message TfClassificationResponse {
  string class_name = 2;
  float  score = 3;
}
// caffe分类识别
message CaffeClassificationRequest {
  bytes  image = 1;
}
message CaffeClassificationResponse {
  string class_name = 2;
  float  score = 3;
}
// 查询人脸样本库
message QueryFaceLibraryRequest {
  string user_id = 1;
  int64 type = 2;
}
message FaceLibrary {
  int64 id = 1;
  string face_name = 2;
  string face_image = 3;
}
message QueryFaceLibraryResponse {
  repeated FaceLibrary faces = 1;
}
// 添加或修改人脸样本名称
message ModifyFaceNameRequest {
  string user_id = 1;
  int64 face_id = 2;
  string face_name = 3;
}
message ModifyFaceNameResponse {
  int64 face_id = 1;
  string face_name = 2;
}
// 修改人脸类型
message ModifyFaceTypeRequest {
  string user_id = 1;
  repeated int64 face_id = 2;
  int64 type = 3;
}
message ModifyFaceTypeResponse {
  string result = 1;
}

// 模糊图片搜索
message ImageClarityRequest {
  bytes  image = 1;
}
message ImageClarityResponse {
  bool   is_blurred = 1;
  float  confidence = 2;
}


service AiService {
  // FaceRecognition
  rpc FaceRecognition (FaceRecognitionRequest) returns (FaceRecognitionResponse);
  // TfClassification
  rpc TfClassification (TfClassificationRequest) returns (TfClassificationResponse);
  // CaffeClassification
  rpc CaffeClassification (CaffeClassificationRequest) returns (CaffeClassificationResponse);
  // QueryFaceLibrary
  rpc QueryFaceLibrary (QueryFaceLibraryRequest) returns (QueryFaceLibraryResponse);
  // ModifyFaceName
  rpc ModifyFaceName (ModifyFaceNameRequest) returns (ModifyFaceNameResponse);
  // ModifyFaceType
  rpc ModifyFaceType (ModifyFaceTypeRequest) returns (ModifyFaceTypeResponse);
  // FuzzySearch
  rpc ImageClarity (ImageClarityRequest) returns (ImageClarityResponse);
}