Update app.py

app.py

@@ -1,12 +1,20 @@
+from fastapi import FastAPI , File , UploadFile , Form
 from datetime import datetime
 import mrzscanner
 import json
-import numpy as np
 from mrz.checker.td1 import TD1CodeChecker
 from mrz.checker.td2 import TD2CodeChecker
 from mrz.checker.td3 import TD3CodeChecker
 from mrz.checker.mrva import MRVACodeChecker
 from mrz.checker.mrvb import MRVBCodeChecker
+import cv2
+import uuid
+import numpy as np
+import gradio as gr
+
+CUSTOM_PATH = "/gradio"
+
+app = FastAPI()
 
 def check(lines):
     try:
@@ -57,14 +65,10 @@ def get_info(scanner,img):
 
   s=""
   image_np=np.frombuffer(img,dtype=np.uint8)
-  print("ok")
-  print(image_np)
-    
   img_np=cv2.imdecode(image_np,cv2.IMREAD_COLOR)
-  print("ok2")
-  print(img_np)
-  image = cv2.cvtColor(img, cv2.COLOR_RGB2BGR)
-  results = scanner.decodeMat(image)
+    
+  #image = cv2.cvtColor(img, cv2.COLOR_RGB2BGR)
+  results = scanner.decodeMat(img_np)
 
   lst1=[]
   lst=[]
@@ -145,6 +149,28 @@ def return_df(final_lst):
 
 
 
+class encodedImage(BaseModel):
+    base64img:str
+
+
+@app.post('/file')
+async def _file_upload(my_file:
+    UploadFile = File(...)):
+
+    my_file.filename = f"{uuid.uuid4()}.jpg"
+    contents = await my_file.read()  # <-- Important!
+
+    mrzscanner.initLicense("DLS2eyJoYW5kc2hha2VDb2RlIjoiMjAwMDAxLTE2NDk4Mjk3OTI2MzUiLCJvcmdhbml6YXRpb25JRCI6IjIwMDAwMSIsInNlc3Npb25QYXNzd29yZCI6IndTcGR6Vm05WDJrcEQ5YUoifQ==")
+    scanner = mrzscanner.createInstance()
+    scanner.loadModel(mrzscanner.get_model_path())
+    lst=get_info(scanner, contents)
+ 
+    final_lst=extract_result(lst)
+    json_file=return_df(final_lst)
+    return json_file
+
+
+
 def main(img):
   # set license
   mrzscanner.initLicense("DLS2eyJoYW5kc2hha2VDb2RlIjoiMjAwMDAxLTE2NDk4Mjk3OTI2MzUiLCJvcmdhbml6YXRpb25JRCI6IjIwMDAwMSIsInNlc3Npb25QYXNzd29yZCI6IndTcGR6Vm05WDJrcEQ5YUoifQ==")