File size: 1,037 Bytes
df6525b
 
77fbfd7
df6525b
 
 
 
2f6db8c
 
 
 
df6525b
 
 
 
2f6db8c
df6525b
 
 
 
 
 
 
 
 
 
 
2f6db8c
df6525b
 
 
2f6db8c
53f56c0
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
from fastapi import FastAPI, HTTPException,Query
import json
from authenticate import get_access_token
app = FastAPI()

@app.post("/process-document-base64")
async def process_document_base64(request_data: dict):
    project_id = os.getenv('PROJECT_ID')
    processor_id = os.getenv('PROCESSOR_ID')
    if request_data.get('base64_content') is None:
        return {"error":"base64 data not present"}
    payload = {
        "skipHumanReview": True,
        "rawDocument": {
            "mimeType": "application/pdf",
            "content": request_data.get('base64_content')
        }
    }
    
    access_token = get_access_token()
    
    headers = {
        'Authorization': f'Bearer {access_token}',
        'Content-Type': 'application/json; charset=utf-8'
    }
    
    response = requests.post(
        f'https://us-documentai.googleapis.com/v1/projects/{project_id}/locations/us/processors/{processor_id}:process',
        headers=headers,
        json=payload
    )
    print(response.json())
    return response.json()