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
|