Spaces:
Runtime error
Runtime error
feat(docker): update docker config
Browse files- README.md +25 -0
- docker-compose.yml +11 -0
- dockerfile +19 -0
- grpc.sh +1 -0
- predictor/__init__.py +42 -9
- protos/resume.proto +48 -0
- protos/resume_pb2.py +37 -0
- protos/resume_pb2.pyi +74 -0
- protos/resume_pb2_grpc.py +100 -0
- requirements.grpc.txt +2 -0
- server.py +153 -0
README.md
CHANGED
@@ -11,3 +11,28 @@ license: apache-2.0
|
|
11 |
---
|
12 |
|
13 |
Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
11 |
---
|
12 |
|
13 |
Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
|
14 |
+
|
15 |
+
### 环境
|
16 |
+
|
17 |
+
1. 最新的Docker Engine - Community(Linux版本)
|
18 |
+
version: 20.0.4
|
19 |
+
|
20 |
+
### 如何部署
|
21 |
+
|
22 |
+
0. 登陆Huggingface申请模型下载权限
|
23 |
+
- minskiter/resume-token-classification
|
24 |
+
https://huggingface.co/minskiter/resume-token-classification
|
25 |
+
|
26 |
+
- minskiter/resume-token-classification-name-0708
|
27 |
+
https://huggingface.co/minskiter/resume-token-classification-name-0708
|
28 |
+
|
29 |
+
1. 首先设置环境变量
|
30 |
+
```bash
|
31 |
+
export HF_Token={Your Huggingface Token}
|
32 |
+
```
|
33 |
+
|
34 |
+
2. 执行部署
|
35 |
+
```
|
36 |
+
sudo docker compose up -d --build
|
37 |
+
```
|
38 |
+
新版使用`docker compose`,旧版使用`docker-compose`
|
docker-compose.yml
ADDED
@@ -0,0 +1,11 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
version: '3'
|
2 |
+
|
3 |
+
services:
|
4 |
+
ner:
|
5 |
+
build: .
|
6 |
+
restart: always
|
7 |
+
ports:
|
8 |
+
- "50050:50051"
|
9 |
+
environment:
|
10 |
+
- HF_Token=${HF_Token}
|
11 |
+
|
dockerfile
ADDED
@@ -0,0 +1,19 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
FROM python:3.8
|
2 |
+
|
3 |
+
# Path: /app
|
4 |
+
WORKDIR /app
|
5 |
+
|
6 |
+
# CACHE
|
7 |
+
COPY requirements.txt ./
|
8 |
+
RUN pip install -r requirements.txt
|
9 |
+
|
10 |
+
# CACHE GRPC
|
11 |
+
COPY requirements.grpc.txt ./
|
12 |
+
RUN pip install -r requirements.grpc.txt
|
13 |
+
|
14 |
+
COPY . ./
|
15 |
+
RUN ./grpc.sh
|
16 |
+
|
17 |
+
EXPOSE 50051
|
18 |
+
|
19 |
+
ENTRYPOINT [ "python", "./server.py" ]
|
grpc.sh
ADDED
@@ -0,0 +1 @@
|
|
|
|
|
1 |
+
python -m grpc_tools.protoc -I./ --python_out=./ --pyi_out=./ --grpc_python_out=./ ./protos/resume.proto
|
predictor/__init__.py
CHANGED
@@ -7,6 +7,8 @@ import pandas as pd
|
|
7 |
import math
|
8 |
import queue
|
9 |
from datetime import date
|
|
|
|
|
10 |
|
11 |
class Predictor():
|
12 |
|
@@ -22,6 +24,7 @@ class Predictor():
|
|
22 |
raise ValueError("'common' pipeline is None")
|
23 |
self.pipelines = pipelines
|
24 |
self.today = today
|
|
|
25 |
self.__init_split_data()
|
26 |
self.__init_schools_data(paths)
|
27 |
self.__init_patterns()
|
@@ -111,7 +114,7 @@ class Predictor():
|
|
111 |
raise
|
112 |
if datestr=="至今":
|
113 |
return self.today
|
114 |
-
return
|
115 |
|
116 |
def split_to_blocks(
|
117 |
self,
|
@@ -158,7 +161,7 @@ class Predictor():
|
|
158 |
text: str,
|
159 |
start: int,
|
160 |
end: int,
|
161 |
-
max_expand_length=
|
162 |
):
|
163 |
expand_l,expand_r = start,end
|
164 |
for l in range(max(start-max_expand_length,0), start):
|
@@ -200,15 +203,15 @@ class Predictor():
|
|
200 |
}
|
201 |
# 获取名字,先过滤所有空白字符,防止名字中间有空格
|
202 |
remove_blanks_text, index_mapper = self.remove_blanks(text, re.compile(r' '))
|
203 |
-
|
204 |
-
|
205 |
-
|
206 |
-
entities = self.pipelines['name'](
|
207 |
for entity in entities:
|
208 |
if entity['entity']=='NAME' and self.name_pattern.match(entity['word']) is not None:
|
209 |
obj = {
|
210 |
-
'start': index_mapper[
|
211 |
-
'end': index_mapper[
|
212 |
'entity': 'NAME',
|
213 |
'text': entity['word']
|
214 |
}
|
@@ -220,7 +223,10 @@ class Predictor():
|
|
220 |
if not repeat:
|
221 |
obj['origin'] = text[obj['start']:obj['end']]
|
222 |
return_obj['name'].append(obj)
|
|
|
|
|
223 |
# 获取年龄
|
|
|
224 |
for age_match in self.age_patterns[0].finditer(remove_blanks_text):
|
225 |
age = None
|
226 |
s,e = -1,-1
|
@@ -266,6 +272,9 @@ class Predictor():
|
|
266 |
'entity': 'AGE',
|
267 |
'origin': text[index_mapper[s]:index_mapper[e-1]+1]
|
268 |
})
|
|
|
|
|
|
|
269 |
# 获取学校
|
270 |
for school_match in self.school_pattern.finditer(remove_blanks_text):
|
271 |
start,end = school_match.span()
|
@@ -309,6 +318,9 @@ class Predictor():
|
|
309 |
obj['origin'] = text[obj['start']:obj['end']]
|
310 |
obj['level'] = self.schools[obj['text']]
|
311 |
return_obj['schools'].append(obj)
|
|
|
|
|
|
|
312 |
# 获取学历
|
313 |
for i,pattern in enumerate(self.edu_patterns):
|
314 |
for edu_match in pattern.finditer(remove_blanks_text):
|
@@ -331,16 +343,23 @@ class Predictor():
|
|
331 |
if not repeat:
|
332 |
obj['origin'] = text[obj['start']:obj['end']]
|
333 |
return_obj['edus'].append(obj)
|
|
|
|
|
|
|
334 |
# 如果有工作经历
|
335 |
if self.works_key_pattern.search(remove_blanks_text) is not None:
|
336 |
for job_time_match in self.job_time_patterns.finditer(remove_blanks_text):
|
337 |
origin_start,origin_end = job_time_match.span()
|
338 |
# convert_to_date
|
339 |
fr = self.to_date(job_time_match.group(1))
|
|
|
|
|
340 |
fs,fe = job_time_match.span(1)
|
341 |
to = self.to_date(job_time_match.group(3))
|
|
|
|
|
342 |
ts,te = job_time_match.span(3)
|
343 |
-
expand_text, start, end = self.get_expand_span(remove_blanks_text, origin_start, origin_end, max_expand_length=
|
344 |
entities = self.pipelines['common'](expand_text)
|
345 |
objs = []
|
346 |
for entity in entities:
|
@@ -390,6 +409,9 @@ class Predictor():
|
|
390 |
work_month += diff_y * 12 + diff_m
|
391 |
last_end = end
|
392 |
return_obj['work_time'] = math.ceil(work_month/12)
|
|
|
|
|
|
|
393 |
# 获取手机号码
|
394 |
for phone_match in self.phone_pattern.finditer(text):
|
395 |
start,end = phone_match.span()
|
@@ -400,6 +422,9 @@ class Predictor():
|
|
400 |
'origin': text[start:end],
|
401 |
'text': re.sub('\s','',text[start:end])
|
402 |
})
|
|
|
|
|
|
|
403 |
for email_match in self.email_pattern.finditer(text):
|
404 |
start,end = email_match.span()
|
405 |
return_obj['email'].append({
|
@@ -409,6 +434,9 @@ class Predictor():
|
|
409 |
'origin': text[start:end],
|
410 |
'text': re.sub('\s','',text[start:end])
|
411 |
})
|
|
|
|
|
|
|
412 |
for gender_match in self.gender_pattern.finditer(text):
|
413 |
start,end = gender_match.span(2)
|
414 |
return_obj['gender'].append({
|
@@ -418,6 +446,9 @@ class Predictor():
|
|
418 |
'word': text[start:end],
|
419 |
'text': text[start:end]
|
420 |
})
|
|
|
|
|
|
|
421 |
for block in self.split_to_blocks(remove_blanks_text):
|
422 |
entities = self.pipelines["common"](block["text"])
|
423 |
for entity in entities:
|
@@ -436,6 +467,8 @@ class Predictor():
|
|
436 |
break
|
437 |
if not repeat:
|
438 |
return_obj['titles'].append(obj)
|
|
|
|
|
439 |
return return_obj
|
440 |
|
441 |
def __call__(self, *args: Any, **kwds: Any) -> Any:
|
|
|
7 |
import math
|
8 |
import queue
|
9 |
from datetime import date
|
10 |
+
import time
|
11 |
+
import logging
|
12 |
|
13 |
class Predictor():
|
14 |
|
|
|
24 |
raise ValueError("'common' pipeline is None")
|
25 |
self.pipelines = pipelines
|
26 |
self.today = today
|
27 |
+
self.logger = logging.getLogger(__name__)
|
28 |
self.__init_split_data()
|
29 |
self.__init_schools_data(paths)
|
30 |
self.__init_patterns()
|
|
|
114 |
raise
|
115 |
if datestr=="至今":
|
116 |
return self.today
|
117 |
+
return None
|
118 |
|
119 |
def split_to_blocks(
|
120 |
self,
|
|
|
161 |
text: str,
|
162 |
start: int,
|
163 |
end: int,
|
164 |
+
max_expand_length=10,
|
165 |
):
|
166 |
expand_l,expand_r = start,end
|
167 |
for l in range(max(start-max_expand_length,0), start):
|
|
|
203 |
}
|
204 |
# 获取名字,先过滤所有空白字符,防止名字中间有空格
|
205 |
remove_blanks_text, index_mapper = self.remove_blanks(text, re.compile(r' '))
|
206 |
+
start_time = time.perf_counter()
|
207 |
+
for block in self.split_to_blocks(remove_blanks_text):
|
208 |
+
block_text,block_l = block['text'],block['start']
|
209 |
+
entities = self.pipelines['name'](block_text)
|
210 |
for entity in entities:
|
211 |
if entity['entity']=='NAME' and self.name_pattern.match(entity['word']) is not None:
|
212 |
obj = {
|
213 |
+
'start': index_mapper[block_l+entity['start']],
|
214 |
+
'end': index_mapper[block_l+entity['end']-1]+1,
|
215 |
'entity': 'NAME',
|
216 |
'text': entity['word']
|
217 |
}
|
|
|
223 |
if not repeat:
|
224 |
obj['origin'] = text[obj['start']:obj['end']]
|
225 |
return_obj['name'].append(obj)
|
226 |
+
end_time = time.perf_counter()
|
227 |
+
self.logger.info(f"process name time: {end_time-start_time}")
|
228 |
# 获取年龄
|
229 |
+
start_time = time.perf_counter()
|
230 |
for age_match in self.age_patterns[0].finditer(remove_blanks_text):
|
231 |
age = None
|
232 |
s,e = -1,-1
|
|
|
272 |
'entity': 'AGE',
|
273 |
'origin': text[index_mapper[s]:index_mapper[e-1]+1]
|
274 |
})
|
275 |
+
end_time = time.perf_counter()
|
276 |
+
self.logger.info(f"process age time: {end_time-start_time}")
|
277 |
+
start_time = time.perf_counter()
|
278 |
# 获取学校
|
279 |
for school_match in self.school_pattern.finditer(remove_blanks_text):
|
280 |
start,end = school_match.span()
|
|
|
318 |
obj['origin'] = text[obj['start']:obj['end']]
|
319 |
obj['level'] = self.schools[obj['text']]
|
320 |
return_obj['schools'].append(obj)
|
321 |
+
end_time = time.perf_counter()
|
322 |
+
self.logger.info(f"process school time: {end_time-start_time}")
|
323 |
+
start_time = time.perf_counter()
|
324 |
# 获取学历
|
325 |
for i,pattern in enumerate(self.edu_patterns):
|
326 |
for edu_match in pattern.finditer(remove_blanks_text):
|
|
|
343 |
if not repeat:
|
344 |
obj['origin'] = text[obj['start']:obj['end']]
|
345 |
return_obj['edus'].append(obj)
|
346 |
+
end_time = time.perf_counter()
|
347 |
+
self.logger.info(f"process edu time: {end_time-start_time}")
|
348 |
+
start_time = time.perf_counter()
|
349 |
# 如果有工作经历
|
350 |
if self.works_key_pattern.search(remove_blanks_text) is not None:
|
351 |
for job_time_match in self.job_time_patterns.finditer(remove_blanks_text):
|
352 |
origin_start,origin_end = job_time_match.span()
|
353 |
# convert_to_date
|
354 |
fr = self.to_date(job_time_match.group(1))
|
355 |
+
if fr is None:
|
356 |
+
continue
|
357 |
fs,fe = job_time_match.span(1)
|
358 |
to = self.to_date(job_time_match.group(3))
|
359 |
+
if to is None:
|
360 |
+
continue
|
361 |
ts,te = job_time_match.span(3)
|
362 |
+
expand_text, start, end = self.get_expand_span(remove_blanks_text, origin_start, origin_end, max_expand_length=50)
|
363 |
entities = self.pipelines['common'](expand_text)
|
364 |
objs = []
|
365 |
for entity in entities:
|
|
|
409 |
work_month += diff_y * 12 + diff_m
|
410 |
last_end = end
|
411 |
return_obj['work_time'] = math.ceil(work_month/12)
|
412 |
+
end_time = time.perf_counter()
|
413 |
+
self.logger.info(f"process work time: {end_time-start_time}")
|
414 |
+
start_time = time.perf_counter()
|
415 |
# 获取手机号码
|
416 |
for phone_match in self.phone_pattern.finditer(text):
|
417 |
start,end = phone_match.span()
|
|
|
422 |
'origin': text[start:end],
|
423 |
'text': re.sub('\s','',text[start:end])
|
424 |
})
|
425 |
+
end_time = time.perf_counter()
|
426 |
+
self.logger.info(f"process phone time: {end_time-start_time}")
|
427 |
+
start_time = time.perf_counter()
|
428 |
for email_match in self.email_pattern.finditer(text):
|
429 |
start,end = email_match.span()
|
430 |
return_obj['email'].append({
|
|
|
434 |
'origin': text[start:end],
|
435 |
'text': re.sub('\s','',text[start:end])
|
436 |
})
|
437 |
+
end_time = time.perf_counter()
|
438 |
+
self.logger.info(f"process email time: {end_time-start_time}")
|
439 |
+
start_time = time.perf_counter()
|
440 |
for gender_match in self.gender_pattern.finditer(text):
|
441 |
start,end = gender_match.span(2)
|
442 |
return_obj['gender'].append({
|
|
|
446 |
'word': text[start:end],
|
447 |
'text': text[start:end]
|
448 |
})
|
449 |
+
end_time = time.perf_counter()
|
450 |
+
self.logger.info(f"process gender time: {end_time-start_time}")
|
451 |
+
start_time = time.perf_counter()
|
452 |
for block in self.split_to_blocks(remove_blanks_text):
|
453 |
entities = self.pipelines["common"](block["text"])
|
454 |
for entity in entities:
|
|
|
467 |
break
|
468 |
if not repeat:
|
469 |
return_obj['titles'].append(obj)
|
470 |
+
end_time = time.perf_counter()
|
471 |
+
self.logger.info(f"process title time: {end_time-start_time}")
|
472 |
return return_obj
|
473 |
|
474 |
def __call__(self, *args: Any, **kwds: Any) -> Any:
|
protos/resume.proto
ADDED
@@ -0,0 +1,48 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
syntax = "proto3";
|
2 |
+
|
3 |
+
service Resume{
|
4 |
+
// Get Resume Info
|
5 |
+
rpc GetInfo(TextRequest) returns (ResumeEntitiesResponse);
|
6 |
+
rpc Health(PingRequest) returns (PongResponse);
|
7 |
+
}
|
8 |
+
|
9 |
+
message PingRequest{
|
10 |
+
string ping = 1;
|
11 |
+
}
|
12 |
+
|
13 |
+
message PongResponse{
|
14 |
+
string done = 1;
|
15 |
+
}
|
16 |
+
|
17 |
+
message TextRequest{
|
18 |
+
string text = 1;
|
19 |
+
}
|
20 |
+
|
21 |
+
message Entity{
|
22 |
+
string entity = 1;
|
23 |
+
int32 start = 2;
|
24 |
+
int32 end = 3;
|
25 |
+
optional string text = 4;
|
26 |
+
string origin = 5;
|
27 |
+
optional string level = 6;
|
28 |
+
}
|
29 |
+
|
30 |
+
message JobEntity {
|
31 |
+
Entity start = 1;
|
32 |
+
Entity end = 2;
|
33 |
+
Entity company = 3;
|
34 |
+
}
|
35 |
+
|
36 |
+
message ResumeEntitiesResponse{
|
37 |
+
repeated Entity names = 1;
|
38 |
+
repeated Entity emails = 2;
|
39 |
+
repeated Entity phones = 3;
|
40 |
+
repeated Entity schools = 4;
|
41 |
+
repeated Entity edus = 5;
|
42 |
+
repeated Entity ages = 6;
|
43 |
+
repeated Entity genders = 7;
|
44 |
+
int32 work_years = 8;
|
45 |
+
repeated JobEntity jobs = 9;
|
46 |
+
repeated Entity titles = 10;
|
47 |
+
}
|
48 |
+
|
protos/resume_pb2.py
ADDED
@@ -0,0 +1,37 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
# -*- coding: utf-8 -*-
|
2 |
+
# Generated by the protocol buffer compiler. DO NOT EDIT!
|
3 |
+
# source: protos/resume.proto
|
4 |
+
"""Generated protocol buffer code."""
|
5 |
+
from google.protobuf.internal import builder as _builder
|
6 |
+
from google.protobuf import descriptor as _descriptor
|
7 |
+
from google.protobuf import descriptor_pool as _descriptor_pool
|
8 |
+
from google.protobuf import symbol_database as _symbol_database
|
9 |
+
# @@protoc_insertion_point(imports)
|
10 |
+
|
11 |
+
_sym_db = _symbol_database.Default()
|
12 |
+
|
13 |
+
|
14 |
+
|
15 |
+
|
16 |
+
DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x13protos/resume.proto\"\x1b\n\x0bPingRequest\x12\x0c\n\x04ping\x18\x01 \x01(\t\"\x1c\n\x0cPongResponse\x12\x0c\n\x04\x64one\x18\x01 \x01(\t\"\x1b\n\x0bTextRequest\x12\x0c\n\x04text\x18\x01 \x01(\t\"~\n\x06\x45ntity\x12\x0e\n\x06\x65ntity\x18\x01 \x01(\t\x12\r\n\x05start\x18\x02 \x01(\x05\x12\x0b\n\x03\x65nd\x18\x03 \x01(\x05\x12\x11\n\x04text\x18\x04 \x01(\tH\x00\x88\x01\x01\x12\x0e\n\x06origin\x18\x05 \x01(\t\x12\x12\n\x05level\x18\x06 \x01(\tH\x01\x88\x01\x01\x42\x07\n\x05_textB\x08\n\x06_level\"S\n\tJobEntity\x12\x16\n\x05start\x18\x01 \x01(\x0b\x32\x07.Entity\x12\x14\n\x03\x65nd\x18\x02 \x01(\x0b\x32\x07.Entity\x12\x18\n\x07\x63ompany\x18\x03 \x01(\x0b\x32\x07.Entity\"\x8b\x02\n\x16ResumeEntitiesResponse\x12\x16\n\x05names\x18\x01 \x03(\x0b\x32\x07.Entity\x12\x17\n\x06\x65mails\x18\x02 \x03(\x0b\x32\x07.Entity\x12\x17\n\x06phones\x18\x03 \x03(\x0b\x32\x07.Entity\x12\x18\n\x07schools\x18\x04 \x03(\x0b\x32\x07.Entity\x12\x15\n\x04\x65\x64us\x18\x05 \x03(\x0b\x32\x07.Entity\x12\x15\n\x04\x61ges\x18\x06 \x03(\x0b\x32\x07.Entity\x12\x18\n\x07genders\x18\x07 \x03(\x0b\x32\x07.Entity\x12\x12\n\nwork_years\x18\x08 \x01(\x05\x12\x18\n\x04jobs\x18\t \x03(\x0b\x32\n.JobEntity\x12\x17\n\x06titles\x18\n \x03(\x0b\x32\x07.Entity2a\n\x06Resume\x12\x30\n\x07GetInfo\x12\x0c.TextRequest\x1a\x17.ResumeEntitiesResponse\x12%\n\x06Health\x12\x0c.PingRequest\x1a\r.PongResponseb\x06proto3')
|
17 |
+
|
18 |
+
_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, globals())
|
19 |
+
_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'protos.resume_pb2', globals())
|
20 |
+
if _descriptor._USE_C_DESCRIPTORS == False:
|
21 |
+
|
22 |
+
DESCRIPTOR._options = None
|
23 |
+
_PINGREQUEST._serialized_start=23
|
24 |
+
_PINGREQUEST._serialized_end=50
|
25 |
+
_PONGRESPONSE._serialized_start=52
|
26 |
+
_PONGRESPONSE._serialized_end=80
|
27 |
+
_TEXTREQUEST._serialized_start=82
|
28 |
+
_TEXTREQUEST._serialized_end=109
|
29 |
+
_ENTITY._serialized_start=111
|
30 |
+
_ENTITY._serialized_end=237
|
31 |
+
_JOBENTITY._serialized_start=239
|
32 |
+
_JOBENTITY._serialized_end=322
|
33 |
+
_RESUMEENTITIESRESPONSE._serialized_start=325
|
34 |
+
_RESUMEENTITIESRESPONSE._serialized_end=592
|
35 |
+
_RESUME._serialized_start=594
|
36 |
+
_RESUME._serialized_end=691
|
37 |
+
# @@protoc_insertion_point(module_scope)
|
protos/resume_pb2.pyi
ADDED
@@ -0,0 +1,74 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from google.protobuf.internal import containers as _containers
|
2 |
+
from google.protobuf import descriptor as _descriptor
|
3 |
+
from google.protobuf import message as _message
|
4 |
+
from typing import ClassVar as _ClassVar, Iterable as _Iterable, Mapping as _Mapping, Optional as _Optional, Union as _Union
|
5 |
+
|
6 |
+
DESCRIPTOR: _descriptor.FileDescriptor
|
7 |
+
|
8 |
+
class Entity(_message.Message):
|
9 |
+
__slots__ = ["end", "entity", "level", "origin", "start", "text"]
|
10 |
+
END_FIELD_NUMBER: _ClassVar[int]
|
11 |
+
ENTITY_FIELD_NUMBER: _ClassVar[int]
|
12 |
+
LEVEL_FIELD_NUMBER: _ClassVar[int]
|
13 |
+
ORIGIN_FIELD_NUMBER: _ClassVar[int]
|
14 |
+
START_FIELD_NUMBER: _ClassVar[int]
|
15 |
+
TEXT_FIELD_NUMBER: _ClassVar[int]
|
16 |
+
end: int
|
17 |
+
entity: str
|
18 |
+
level: str
|
19 |
+
origin: str
|
20 |
+
start: int
|
21 |
+
text: str
|
22 |
+
def __init__(self, entity: _Optional[str] = ..., start: _Optional[int] = ..., end: _Optional[int] = ..., text: _Optional[str] = ..., origin: _Optional[str] = ..., level: _Optional[str] = ...) -> None: ...
|
23 |
+
|
24 |
+
class JobEntity(_message.Message):
|
25 |
+
__slots__ = ["company", "end", "start"]
|
26 |
+
COMPANY_FIELD_NUMBER: _ClassVar[int]
|
27 |
+
END_FIELD_NUMBER: _ClassVar[int]
|
28 |
+
START_FIELD_NUMBER: _ClassVar[int]
|
29 |
+
company: Entity
|
30 |
+
end: Entity
|
31 |
+
start: Entity
|
32 |
+
def __init__(self, start: _Optional[_Union[Entity, _Mapping]] = ..., end: _Optional[_Union[Entity, _Mapping]] = ..., company: _Optional[_Union[Entity, _Mapping]] = ...) -> None: ...
|
33 |
+
|
34 |
+
class PingRequest(_message.Message):
|
35 |
+
__slots__ = ["ping"]
|
36 |
+
PING_FIELD_NUMBER: _ClassVar[int]
|
37 |
+
ping: str
|
38 |
+
def __init__(self, ping: _Optional[str] = ...) -> None: ...
|
39 |
+
|
40 |
+
class PongResponse(_message.Message):
|
41 |
+
__slots__ = ["done"]
|
42 |
+
DONE_FIELD_NUMBER: _ClassVar[int]
|
43 |
+
done: str
|
44 |
+
def __init__(self, done: _Optional[str] = ...) -> None: ...
|
45 |
+
|
46 |
+
class ResumeEntitiesResponse(_message.Message):
|
47 |
+
__slots__ = ["ages", "edus", "emails", "genders", "jobs", "names", "phones", "schools", "titles", "work_years"]
|
48 |
+
AGES_FIELD_NUMBER: _ClassVar[int]
|
49 |
+
EDUS_FIELD_NUMBER: _ClassVar[int]
|
50 |
+
EMAILS_FIELD_NUMBER: _ClassVar[int]
|
51 |
+
GENDERS_FIELD_NUMBER: _ClassVar[int]
|
52 |
+
JOBS_FIELD_NUMBER: _ClassVar[int]
|
53 |
+
NAMES_FIELD_NUMBER: _ClassVar[int]
|
54 |
+
PHONES_FIELD_NUMBER: _ClassVar[int]
|
55 |
+
SCHOOLS_FIELD_NUMBER: _ClassVar[int]
|
56 |
+
TITLES_FIELD_NUMBER: _ClassVar[int]
|
57 |
+
WORK_YEARS_FIELD_NUMBER: _ClassVar[int]
|
58 |
+
ages: _containers.RepeatedCompositeFieldContainer[Entity]
|
59 |
+
edus: _containers.RepeatedCompositeFieldContainer[Entity]
|
60 |
+
emails: _containers.RepeatedCompositeFieldContainer[Entity]
|
61 |
+
genders: _containers.RepeatedCompositeFieldContainer[Entity]
|
62 |
+
jobs: _containers.RepeatedCompositeFieldContainer[JobEntity]
|
63 |
+
names: _containers.RepeatedCompositeFieldContainer[Entity]
|
64 |
+
phones: _containers.RepeatedCompositeFieldContainer[Entity]
|
65 |
+
schools: _containers.RepeatedCompositeFieldContainer[Entity]
|
66 |
+
titles: _containers.RepeatedCompositeFieldContainer[Entity]
|
67 |
+
work_years: int
|
68 |
+
def __init__(self, names: _Optional[_Iterable[_Union[Entity, _Mapping]]] = ..., emails: _Optional[_Iterable[_Union[Entity, _Mapping]]] = ..., phones: _Optional[_Iterable[_Union[Entity, _Mapping]]] = ..., schools: _Optional[_Iterable[_Union[Entity, _Mapping]]] = ..., edus: _Optional[_Iterable[_Union[Entity, _Mapping]]] = ..., ages: _Optional[_Iterable[_Union[Entity, _Mapping]]] = ..., genders: _Optional[_Iterable[_Union[Entity, _Mapping]]] = ..., work_years: _Optional[int] = ..., jobs: _Optional[_Iterable[_Union[JobEntity, _Mapping]]] = ..., titles: _Optional[_Iterable[_Union[Entity, _Mapping]]] = ...) -> None: ...
|
69 |
+
|
70 |
+
class TextRequest(_message.Message):
|
71 |
+
__slots__ = ["text"]
|
72 |
+
TEXT_FIELD_NUMBER: _ClassVar[int]
|
73 |
+
text: str
|
74 |
+
def __init__(self, text: _Optional[str] = ...) -> None: ...
|
protos/resume_pb2_grpc.py
ADDED
@@ -0,0 +1,100 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
|
2 |
+
"""Client and server classes corresponding to protobuf-defined services."""
|
3 |
+
import grpc
|
4 |
+
|
5 |
+
from protos import resume_pb2 as protos_dot_resume__pb2
|
6 |
+
|
7 |
+
|
8 |
+
class ResumeStub(object):
|
9 |
+
"""Missing associated documentation comment in .proto file."""
|
10 |
+
|
11 |
+
def __init__(self, channel):
|
12 |
+
"""Constructor.
|
13 |
+
|
14 |
+
Args:
|
15 |
+
channel: A grpc.Channel.
|
16 |
+
"""
|
17 |
+
self.GetInfo = channel.unary_unary(
|
18 |
+
'/Resume/GetInfo',
|
19 |
+
request_serializer=protos_dot_resume__pb2.TextRequest.SerializeToString,
|
20 |
+
response_deserializer=protos_dot_resume__pb2.ResumeEntitiesResponse.FromString,
|
21 |
+
)
|
22 |
+
self.Health = channel.unary_unary(
|
23 |
+
'/Resume/Health',
|
24 |
+
request_serializer=protos_dot_resume__pb2.PingRequest.SerializeToString,
|
25 |
+
response_deserializer=protos_dot_resume__pb2.PongResponse.FromString,
|
26 |
+
)
|
27 |
+
|
28 |
+
|
29 |
+
class ResumeServicer(object):
|
30 |
+
"""Missing associated documentation comment in .proto file."""
|
31 |
+
|
32 |
+
def GetInfo(self, request, context):
|
33 |
+
"""Get Resume Info
|
34 |
+
"""
|
35 |
+
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
36 |
+
context.set_details('Method not implemented!')
|
37 |
+
raise NotImplementedError('Method not implemented!')
|
38 |
+
|
39 |
+
def Health(self, request, context):
|
40 |
+
"""Missing associated documentation comment in .proto file."""
|
41 |
+
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
42 |
+
context.set_details('Method not implemented!')
|
43 |
+
raise NotImplementedError('Method not implemented!')
|
44 |
+
|
45 |
+
|
46 |
+
def add_ResumeServicer_to_server(servicer, server):
|
47 |
+
rpc_method_handlers = {
|
48 |
+
'GetInfo': grpc.unary_unary_rpc_method_handler(
|
49 |
+
servicer.GetInfo,
|
50 |
+
request_deserializer=protos_dot_resume__pb2.TextRequest.FromString,
|
51 |
+
response_serializer=protos_dot_resume__pb2.ResumeEntitiesResponse.SerializeToString,
|
52 |
+
),
|
53 |
+
'Health': grpc.unary_unary_rpc_method_handler(
|
54 |
+
servicer.Health,
|
55 |
+
request_deserializer=protos_dot_resume__pb2.PingRequest.FromString,
|
56 |
+
response_serializer=protos_dot_resume__pb2.PongResponse.SerializeToString,
|
57 |
+
),
|
58 |
+
}
|
59 |
+
generic_handler = grpc.method_handlers_generic_handler(
|
60 |
+
'Resume', rpc_method_handlers)
|
61 |
+
server.add_generic_rpc_handlers((generic_handler,))
|
62 |
+
|
63 |
+
|
64 |
+
# This class is part of an EXPERIMENTAL API.
|
65 |
+
class Resume(object):
|
66 |
+
"""Missing associated documentation comment in .proto file."""
|
67 |
+
|
68 |
+
@staticmethod
|
69 |
+
def GetInfo(request,
|
70 |
+
target,
|
71 |
+
options=(),
|
72 |
+
channel_credentials=None,
|
73 |
+
call_credentials=None,
|
74 |
+
insecure=False,
|
75 |
+
compression=None,
|
76 |
+
wait_for_ready=None,
|
77 |
+
timeout=None,
|
78 |
+
metadata=None):
|
79 |
+
return grpc.experimental.unary_unary(request, target, '/Resume/GetInfo',
|
80 |
+
protos_dot_resume__pb2.TextRequest.SerializeToString,
|
81 |
+
protos_dot_resume__pb2.ResumeEntitiesResponse.FromString,
|
82 |
+
options, channel_credentials,
|
83 |
+
insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
|
84 |
+
|
85 |
+
@staticmethod
|
86 |
+
def Health(request,
|
87 |
+
target,
|
88 |
+
options=(),
|
89 |
+
channel_credentials=None,
|
90 |
+
call_credentials=None,
|
91 |
+
insecure=False,
|
92 |
+
compression=None,
|
93 |
+
wait_for_ready=None,
|
94 |
+
timeout=None,
|
95 |
+
metadata=None):
|
96 |
+
return grpc.experimental.unary_unary(request, target, '/Resume/Health',
|
97 |
+
protos_dot_resume__pb2.PingRequest.SerializeToString,
|
98 |
+
protos_dot_resume__pb2.PongResponse.FromString,
|
99 |
+
options, channel_credentials,
|
100 |
+
insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
|
requirements.grpc.txt
ADDED
@@ -0,0 +1,2 @@
|
|
|
|
|
|
|
1 |
+
grpcio==1.54.2
|
2 |
+
grpcio-tools==1.54.2
|
server.py
ADDED
@@ -0,0 +1,153 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import logging
|
2 |
+
import grpc
|
3 |
+
from concurrent import futures
|
4 |
+
import protos.resume_pb2
|
5 |
+
import protos.resume_pb2_grpc
|
6 |
+
from huggingface_hub import login
|
7 |
+
import os
|
8 |
+
from predictor import Predictor
|
9 |
+
from transformers import pipeline
|
10 |
+
from datetime import date
|
11 |
+
|
12 |
+
HF_TOKEN = os.environ["HF_Token"]
|
13 |
+
PORT = os.environ.get("PORT", "50051")
|
14 |
+
login(HF_TOKEN)
|
15 |
+
|
16 |
+
class Resume(protos.resume_pb2_grpc.ResumeServicer):
|
17 |
+
|
18 |
+
def __init__(self):
|
19 |
+
self.done = False
|
20 |
+
self.logger = logging.getLogger(__name__)
|
21 |
+
self.predictor = Predictor(
|
22 |
+
pipelines={
|
23 |
+
"name": pipeline("nerpipe", model="minskiter/resume-token-classification-name-0708",trust_remote_code=True,use_auth_token=True),
|
24 |
+
"common": pipeline("nerpipe",model="minskiter/resume-token-classification",trust_remote_code=True,use_auth_token=True)
|
25 |
+
},
|
26 |
+
paths=[
|
27 |
+
"data/W020230619818476939351.xls",
|
28 |
+
"data/W020230619818476975218.xls"
|
29 |
+
],
|
30 |
+
today=date(2023,4,1)
|
31 |
+
)
|
32 |
+
self.done = True
|
33 |
+
|
34 |
+
def Health(self, request, context):
|
35 |
+
self.logger.info("Health check")
|
36 |
+
if request.ping=="PING":
|
37 |
+
if self.done:
|
38 |
+
return protos.resume_pb2.PongResponse(done="OK")
|
39 |
+
else:
|
40 |
+
return protos.resume_pb2.PongResponse(done="Pending")
|
41 |
+
return protos.resume_pb2.PongResponse(done="PING request is not valid")
|
42 |
+
|
43 |
+
def GetInfo(self, request, context):
|
44 |
+
entities = self.predictor(request.text)
|
45 |
+
logging.info(entities)
|
46 |
+
res = protos.resume_pb2.ResumeEntitiesResponse()
|
47 |
+
for name in entities['name']:
|
48 |
+
res.names.append(protos.resume_pb2.Entity(
|
49 |
+
entity=name['entity'],
|
50 |
+
start=name['start'],
|
51 |
+
end=name['end'],
|
52 |
+
text=name.get('text',None),
|
53 |
+
origin=name["origin"]
|
54 |
+
))
|
55 |
+
for age in entities['age']:
|
56 |
+
res.ages.append(protos.resume_pb2.Entity(
|
57 |
+
entity=age['entity'],
|
58 |
+
start=age['start'],
|
59 |
+
end=age['end'],
|
60 |
+
text=age.get('text',None),
|
61 |
+
origin=age["origin"]
|
62 |
+
))
|
63 |
+
for gender in entities['gender']:
|
64 |
+
res.genders.append(protos.resume_pb2.Entity(
|
65 |
+
entity=gender['entity'],
|
66 |
+
start=gender['start'],
|
67 |
+
end=gender['end'],
|
68 |
+
text=gender.get('text',None),
|
69 |
+
origin=gender["origin"]
|
70 |
+
))
|
71 |
+
for email in entities['email']:
|
72 |
+
res.emails.append(protos.resume_pb2.Entity(
|
73 |
+
entity=email['entity'],
|
74 |
+
start=email['start'],
|
75 |
+
end=email['end'],
|
76 |
+
text=email.get('text',None),
|
77 |
+
origin=email["origin"]
|
78 |
+
))
|
79 |
+
for phone in entities['phone']:
|
80 |
+
res.phones.append(protos.resume_pb2.Entity(
|
81 |
+
entity=phone['entity'],
|
82 |
+
start=phone['start'],
|
83 |
+
end=phone['end'],
|
84 |
+
text=phone.get('text',None),
|
85 |
+
origin=phone["origin"]
|
86 |
+
))
|
87 |
+
for edu in entities['edus']:
|
88 |
+
res.edus.append(protos.resume_pb2.Entity(
|
89 |
+
entity=edu['entity'],
|
90 |
+
start=edu['start'],
|
91 |
+
end=edu['end'],
|
92 |
+
text=edu.get('text',None),
|
93 |
+
origin=edu["origin"]
|
94 |
+
))
|
95 |
+
for school in entities['schools']:
|
96 |
+
res.schools.append(protos.resume_pb2.Entity(
|
97 |
+
entity=school['entity'],
|
98 |
+
start=school['start'],
|
99 |
+
end=school['end'],
|
100 |
+
text=school.get('text',None),
|
101 |
+
origin=school["origin"],
|
102 |
+
level=school.get('level',None)
|
103 |
+
))
|
104 |
+
for company,start,end in entities['jobs']:
|
105 |
+
jobEntity = protos.resume_pb2.JobEntity(
|
106 |
+
start=protos.resume_pb2.Entity(
|
107 |
+
entity=start['entity'],
|
108 |
+
start=start['start'],
|
109 |
+
end=start['end'],
|
110 |
+
text=start.get('text',None),
|
111 |
+
origin=start["origin"]
|
112 |
+
),
|
113 |
+
end=protos.resume_pb2.Entity(
|
114 |
+
entity=end['entity'],
|
115 |
+
start=end['start'],
|
116 |
+
end=end['end'],
|
117 |
+
text=end.get('text',None),
|
118 |
+
origin=end["origin"]
|
119 |
+
),
|
120 |
+
company=protos.resume_pb2.Entity(
|
121 |
+
entity=company['entity'],
|
122 |
+
start=company['start'],
|
123 |
+
end=company['end'],
|
124 |
+
text=company.get('text',None),
|
125 |
+
origin=company["origin"]
|
126 |
+
)
|
127 |
+
)
|
128 |
+
res.jobs.append(jobEntity)
|
129 |
+
for title in entities['titles']:
|
130 |
+
res.titles.append(protos.resume_pb2.Entity(
|
131 |
+
entity=title['entity'],
|
132 |
+
start=title['start'],
|
133 |
+
end=title['end'],
|
134 |
+
text=title.get('text',None),
|
135 |
+
origin=title["origin"]
|
136 |
+
))
|
137 |
+
res.work_years = entities['work_time']
|
138 |
+
return res
|
139 |
+
|
140 |
+
|
141 |
+
def serve(port = "50051"):
|
142 |
+
logger = logging.getLogger(__name__)
|
143 |
+
server = grpc.server(futures.ThreadPoolExecutor(max_workers=10))
|
144 |
+
protos.resume_pb2_grpc.add_ResumeServicer_to_server(Resume(), server)
|
145 |
+
server.add_insecure_port('[::]:' + port)
|
146 |
+
logger.info("Starting server on port %s", port)
|
147 |
+
server.start()
|
148 |
+
logger.info("Running..")
|
149 |
+
server.wait_for_termination()
|
150 |
+
|
151 |
+
if __name__ == '__main__':
|
152 |
+
logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
|
153 |
+
serve(PORT)
|