Update app.py

app.py

@@ -11,12 +11,16 @@ app = FastAPI()
 async def process_document_base64(request: Request):
     project_id = os.getenv('PROJECT_ID')
     processor_id = os.getenv('PROCESSOR_ID')
+    document_entities = {}
     request_data = await request.json()
     # print(request_data)
     if request_data.get('base64_content') is None:
         return {"error":"base64 data not present"}
     message_id = request_data.get('message_id')
     filename = request_data.get('filename')
+    file_type = filename.split('.')
+    
+        
     payload = {
         "skipHumanReview": True,
         "rawDocument": {
@@ -39,6 +43,14 @@ async def process_document_base64(request: Request):
         headers=headers,
         json=payload
     )
-    print(response.json())
-    return {"message_id":message_id , "filename":filename , "entities_data":response.json()}
+    response_json = response.json()
+    entities = response_json.get('entities' , None)
+    if entities is not None:
+        for ent in entities:
+            if ent.get('type') is not None:
+                mention_text = ent.get('mention_text')
+                normalised_values = ent.get('normalizedValue') if 'normalizedValue' in entities else None
+                document_entities[ent.get('type')] = {"mention_text":ent.get('mention_text'),"normalizedValue":normalised_values}
+    
+    return {"message_id":message_id , "filename":filename , "entities_data":document_entities}