Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -7,9 +7,11 @@ app = FastAPI()
|
|
7 |
async def process_document_base64(request: Request):
|
8 |
project_id = os.getenv('PROJECT_ID')
|
9 |
processor_id = os.getenv('PROCESSOR_ID')
|
10 |
-
request_data = request.json()
|
11 |
if request_data.get('base64_content') is None:
|
12 |
return {"error":"base64 data not present"}
|
|
|
|
|
13 |
payload = {
|
14 |
"skipHumanReview": True,
|
15 |
"rawDocument": {
|
@@ -31,5 +33,5 @@ async def process_document_base64(request: Request):
|
|
31 |
json=payload
|
32 |
)
|
33 |
print(response.json())
|
34 |
-
return response.json()
|
35 |
|
|
|
7 |
async def process_document_base64(request: Request):
|
8 |
project_id = os.getenv('PROJECT_ID')
|
9 |
processor_id = os.getenv('PROCESSOR_ID')
|
10 |
+
request_data = await request.json()
|
11 |
if request_data.get('base64_content') is None:
|
12 |
return {"error":"base64 data not present"}
|
13 |
+
message_id = request_data.get('message_id')
|
14 |
+
filename = request_data.get('filename')
|
15 |
payload = {
|
16 |
"skipHumanReview": True,
|
17 |
"rawDocument": {
|
|
|
33 |
json=payload
|
34 |
)
|
35 |
print(response.json())
|
36 |
+
return {"message_id":message_id , "filename":filename , "entities_data":response.json()}
|
37 |
|