quyip commited on
Commit
1b4655e
·
1 Parent(s): a680719
Files changed (1) hide show
  1. main.py +23 -0
main.py CHANGED
@@ -1,5 +1,8 @@
 
 
1
  from fastapi import FastAPI
2
  from pydantic import BaseModel
 
3
  from summary import summarize
4
 
5
  KEY = 'J9l#K4wP5h@2'
@@ -22,3 +25,23 @@ async def summary(request: SummaryReq):
22
  if request.key != KEY:
23
  return 'Unauthorized'
24
  return summarize(request.id, request.text)
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from typing import TypedDict, List
2
+
3
  from fastapi import FastAPI
4
  from pydantic import BaseModel
5
+
6
  from summary import summarize
7
 
8
  KEY = 'J9l#K4wP5h@2'
 
25
  if request.key != KEY:
26
  return 'Unauthorized'
27
  return summarize(request.id, request.text)
28
+
29
+
30
+ class Page(TypedDict):
31
+ id: str
32
+ text: str
33
+
34
+
35
+ class SummariesReq(BaseModel):
36
+ key: str
37
+ pages: List[Page]
38
+
39
+
40
+ @app.post("/summaries/")
41
+ async def summaries(request: SummariesReq):
42
+ if request.key != KEY:
43
+ return 'Unauthorized'
44
+ result = []
45
+ for page in request.pages:
46
+ result.append(await summarize(page.id, page.text))
47
+ return result