Update server.py

server.py

@@ -119,11 +119,46 @@ class OCRData(BaseModel):
 class ResponseData(BaseModel):
     documentName: str
     errorCode: int
-    mrz: MRZData
-    ocr: OCRData
+    mrz: Optional[MRZData]
+    ocr: Optional[OCRData]
     status: str
 
 
+def create_response_data(mrz, ocr_data):
+    if not mrz and not ocr_data:
+        return ResponseData(
+            documentName="Passport",
+            errorCode=1,
+            mrz=None,
+            ocr=None,
+            status="No MRZ or OCR data available"
+        )
+    elif not mrz:
+        return ResponseData(
+            documentName="Passport",
+            errorCode=2,
+            mrz=None,
+            ocr=OCRData(**ocr_data),
+            status="PassportEYE did not find an MRZ"
+        )
+    elif not ocr_data:
+        return ResponseData(
+            documentName="Passport",
+            errorCode=3,
+            mrz=MRZData(**mrz),
+            ocr=None,
+            status="OCR result not available"
+        )
+    else:
+        return ResponseData(
+            documentName="Passport",
+            errorCode=0,
+            mrz=MRZData(**mrz),
+            ocr=OCRData(**ocr_data),
+            status="ok"
+        )
+
+
 def create_response_data(mrz, ocr_data):
     return ResponseData(
         documentName="Passport",
@@ -166,7 +201,7 @@ async def recognize_passport(image: UploadFile = File(...)):
 
         ocr_data = json.loads(response_str)
 
-        return create_response_data(mrz.to_dict(), ocr_data)
+        return create_response_data(mrz.to_dict() if mrz else None, ocr_data)
 
     except HTTPException as e:
         raise e