Spaces:
Runtime error
Runtime error
quyip
commited on
Commit
·
1b4655e
1
Parent(s):
a680719
fix
Browse files
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
|