Update app.py

app.py

@@ -2,6 +2,7 @@ from fastapi import FastAPI , File , UploadFile , Form
 from datetime import datetime
 from pydantic import BaseModel
 import mrzscanner
+import PIL
 import json
 import uvicorn
 from mrz.checker.td1 import TD1CodeChecker
@@ -63,13 +64,20 @@ def check(lines):
 
 import cv2 
 
-def get_info(scanner,img):
+def get_info(scanner,img,file_type):
 
   s=""
-  image_np=np.frombuffer(img,dtype=np.uint8)
-  img_np=cv2.imdecode(image_np,cv2.IMREAD_COLOR)
-    
-  #image = cv2.cvtColor(img, cv2.COLOR_RGB2BGR)
+  if file_type=="image":
+      image_np=np.frombuffer(img,dtype=np.uint8)
+      img_np=cv2.imdecode(image_np,cv2.IMREAD_COLOR)
+  elif file_type=="pdf":
+       img_np = np.array(img)
+  else:
+       img_np = np.array(img)
+      
+
+
+
   results = scanner.decodeMat(img_np)
 
   lst1=[]
@@ -169,19 +177,50 @@ class encodedImage(BaseModel):
 
 
 @app.post('/')
-async def _file_upload(img:
-    UploadFile = File(...)):
+async def _file_upload(img:UploadFile = File(...)):    
 
-    img.filename = f"{uuid.uuid4()}.jpg"
-    contents = await img.read()  # <-- Important!
-    mrzscanner.initLicense("DLS2eyJoYW5kc2hha2VDb2RlIjoiMTAxNjIyNjAzLVRYbEVaWE5yZEc5d1VISnZhbDlrYkhJIiwib3JnYW5pemF0aW9uSUQiOiIxMDE2MjI2MDMiLCJjaGVja0NvZGUiOjE4ODA3MzYyNzN9")
+    file_type=""
+    if "pdf" in img.filename:
+        
+        
+        file_type="pdf"
+        import pypdfium2 as pdfium
+        img1 = await img.read()  # <-- Important!
+        pdf = pdfium.PdfDocument(img1)
+
+        for x in range(len(pdf)):        
+            page = pdf.get_page(x)
+            pil_image = page.render_to(pdfium.BitmapConv.pil_image,)
+            pil_image.filename = f"{uuid.uuid4()}.jpg"
+            mrzscanner.initLicense("DLS2eyJoYW5kc2hha2VDb2RlIjoiMTAxNjIyNjAzLVRYbEVaWE5yZEc5d1VISnZhbDlrYkhJIiwib3JnYW5pemF0aW9uSUQiOiIxMDE2MjI2MDMiLCJjaGVja0NvZGUiOjE4ODA3MzYyNzN9")
   
-    #mrzscanner.initLicense("DLS2eyJoYW5kc2hha2VDb2RlIjoiMjAwMDAxLTE2NDk4Mjk3OTI2MzUiLCJvcmdhbml6YXRpb25JRCI6IjIwMDAwMSIsInNlc3Npb25QYXNzd29yZCI6IndTcGR6Vm05WDJrcEQ5YUoifQ==")
-    scanner = mrzscanner.createInstance()
-    scanner.loadModel(mrzscanner.get_model_path())
-    lst=get_info(scanner, contents)
+  #          mrzscanner.initLicense("DLS2eyJoYW5kc2hha2VDb2RlIjoiMjAwMDAxLTE2NDk4Mjk3OTI2MzUiLCJvcmdhbml6YXRpb25JRCI6IjIwMDAwMSIsInNlc3Npb25QYXNzd29yZCI6IndTcGR6Vm05WDJrcEQ5YUoifQ==")
+            scanner = mrzscanner.createInstance()
+            scanner.loadModel(mrzscanner.get_model_path())
+            lst=get_info(scanner, pil_image,file_type)
+            if lst:
+                break
+
+            
+    else:
+        file_type="image"
+        img.filename = f"{uuid.uuid4()}.jpg"
+  
+        contents = await img.read()  # <-- Important!
+
+        mrzscanner.initLicense("DLS2eyJoYW5kc2hha2VDb2RlIjoiMTAxNjIyNjAzLVRYbEVaWE5yZEc5d1VISnZhbDlrYkhJIiwib3JnYW5pemF0aW9uSUQiOiIxMDE2MjI2MDMiLCJjaGVja0NvZGUiOjE4ODA3MzYyNzN9")
+  
+  #      mrzscanner.initLicense("DLS2eyJoYW5kc2hha2VDb2RlIjoiMjAwMDAxLTE2NDk4Mjk3OTI2MzUiLCJvcmdhbml6YXRpb25JRCI6IjIwMDAwMSIsInNlc3Npb25QYXNzd29yZCI6IndTcGR6Vm05WDJrcEQ5YUoifQ==")
+        scanner = mrzscanner.createInstance()
+        scanner.loadModel(mrzscanner.get_model_path())
+   
+        lst=get_info(scanner, contents,file_type)
+
+       
+            
+        
     if lst:
- 
+   
         final_lst=extract_result(lst)
         json_file=return_df(final_lst)
         success=False
@@ -219,7 +258,6 @@ def func():
 def main(img):
   # set license.
   mrzscanner.initLicense("DLS2eyJoYW5kc2hha2VDb2RlIjoiMTAxNjIyNjAzLVRYbEVaWE5yZEc5d1VISnZhbDlrYkhJIiwib3JnYW5pemF0aW9uSUQiOiIxMDE2MjI2MDMiLCJjaGVja0NvZGUiOjE4ODA3MzYyNzN9")
-  
   #mrzscanner.initLicense("DLS2eyJoYW5kc2hha2VDb2RlIjoiMjAwMDAxLTE2NDk4Mjk3OTI2MzUiLCJvcmdhbml6YXRpb25JRCI6IjIwMDAwMSIsInNlc3Npb25QYXNzd29yZCI6IndTcGR6Vm05WDJrcEQ5YUoifQ==")
   scanner = mrzscanner.createInstance()
   scanner.loadModel(mrzscanner.get_model_path())