Omkar008 commited on
Commit
d2f7dce
·
verified ·
1 Parent(s): d8cffed

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
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