base64 image

app.py

@@ -5,8 +5,10 @@ import os
 # os.system('apt-get install tesseract-ocr -y')
 # os.system('pip install -q pytesseract')
 
-from base64 import b64decode, b64encode
-from io import BytesIO
+#from base64 import b64decode, b64encode
+#from io import BytesIO
+import base64
+import io
 
 
 import tesserocr
@@ -28,19 +30,26 @@ pipe = pipeline("document-question-answering", model="impira/layoutlm-document-q
 
 
 
-@app.get("/hello_2")
-def read_root():
-    image = 'MicrosoftTeams-image (23).png'
+@app.post("/image")
+def read_root(imageBase64:str):
+    
+
+    img = Image.open(io.BytesIO(base64.decodebytes(bytes(imageBase64, "utf-8"))))
+    print(img)
+    image = img
 
     question_1 = "What is the Total amount?"
     question_2 = "What is Total VAT amount?"
     question_3 = "What is the Date?"
+    question_4 = "What is the total amount currency?"
     output_1 = pipe(image, question_1)
     output_2 = pipe(image, question_2)
     output_3 = pipe(image, question_3)
+    output_4 = pipe(image, question_4)
     
     response = {}
     response['total amount'] = output_1[0]['answer']
     response['toal vat'] = output_2[0]['answer']
     response['date'] = output_3[0]['answer']
+    response['currency'] = output_4[0]['answer']
     return response