quyip commited on
Commit
0433163
·
1 Parent(s): d57b505
Files changed (2) hide show
  1. main.py +5 -3
  2. summary.py +0 -5
main.py CHANGED
@@ -1,6 +1,8 @@
1
  from fastapi import FastAPI
2
  from pydantic import BaseModel
 
3
 
 
4
  app = FastAPI()
5
 
6
 
@@ -16,6 +18,6 @@ class SummaryReq(BaseModel):
16
 
17
  @app.post("/summary/")
18
  async def summary(request: SummaryReq):
19
- print('=============')
20
- print(request)
21
- return ''
 
1
  from fastapi import FastAPI
2
  from pydantic import BaseModel
3
+ from summary import summarize
4
 
5
+ KEY = 'J9l#K4wP5h@2'
6
  app = FastAPI()
7
 
8
 
 
18
 
19
  @app.post("/summary/")
20
  async def summary(request: SummaryReq):
21
+ if request.key != KEY:
22
+ return 'Unauthorized'
23
+ return summarize(request.text)
summary.py CHANGED
@@ -17,16 +17,11 @@ def summarize(text: str):
17
  return {
18
  "ver": AiSummaryVersion
19
  }
20
- print('==============0')
21
  summary = get_summarization(text) if len(text) > 100 else text
22
- print('==============1')
23
  translated = get_en_translation(summary)
24
- print('==============2')
25
  tags1 = get_classification(translated)
26
- print('==============3')
27
  tags2 = get_tags(translated)
28
  tags = filter_tags(list(set(tags1 + tags2)))
29
- print('==============4')
30
 
31
  return {
32
  "ver": AiSummaryVersion,
 
17
  return {
18
  "ver": AiSummaryVersion
19
  }
 
20
  summary = get_summarization(text) if len(text) > 100 else text
 
21
  translated = get_en_translation(summary)
 
22
  tags1 = get_classification(translated)
 
23
  tags2 = get_tags(translated)
24
  tags = filter_tags(list(set(tags1 + tags2)))
 
25
 
26
  return {
27
  "ver": AiSummaryVersion,