Create app.py

app.py

@@ -0,0 +1,193 @@
+from fastapi import FastAPI, UploadFile, File
+from PIL import Image
+import easyocr
+import cv2
+import numpy as np
+import re
+import os
+
+app = FastAPI()
+
+reader = easyocr.Reader(['en'])
+
+# =========================
+# IMAGE QUALITY CHECKS
+# =========================
+
+def is_blurry(image):
+    gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
+    variance = cv2.Laplacian(gray, cv2.CV_64F).var()
+
+    if variance < 100:
+        return True
+
+    return False
+
+
+def is_too_dark(image):
+    brightness = np.mean(image)
+
+    if brightness < 50:
+        return True
+
+    return False
+
+
+# =========================
+# OCR EXTRACTION
+# =========================
+
+def extract_text(image_path):
+
+    results = reader.readtext(image_path)
+
+    text = " ".join([r[1] for r in results]).lower()
+
+    return text
+
+
+# =========================
+# DOCUMENT VALIDATION
+# =========================
+
+def detect_document(text):
+
+    # NIN
+    if (
+        "national identification number" in text
+        or "nin" in text
+    ):
+        return "nin"
+
+    # Passport
+    elif (
+        "passport" in text
+        or "federal republic of nigeria" in text
+    ):
+        return "passport"
+
+    # Driver License
+    elif (
+        "driver" in text
+        and "license" in text
+    ):
+        return "drivers_license"
+
+    # Voter Card
+    elif (
+        "voter" in text
+        or "inec" in text
+    ):
+        return "voters_card"
+
+    # Utility Bill
+    elif (
+        "electricity" in text
+        or "water bill" in text
+        or "eko electric" in text
+        or "ikeja electric" in text
+    ):
+        return "utility_bill"
+
+    # Bank Statement
+    elif (
+        "account statement" in text
+        or "transaction" in text
+        or "balance" in text
+    ):
+        return "bank_statement"
+
+    # Tenancy Agreement
+    elif (
+        "tenancy agreement" in text
+        or "landlord" in text
+        or "tenant" in text
+    ):
+        return "tenancy_agreement"
+
+    # Vehicle Plate Number
+    elif re.search(r"[A-Z]{3}-?\d{3}[A-Z]{2}", text.upper()):
+        return "vehicle_plate"
+
+    return None
+
+
+# =========================
+# MAIN API ENDPOINT
+# =========================
+
+@app.post("/validate")
+
+async def validate_document(
+    file: UploadFile = File(...)
+):
+
+    try:
+
+        # SAVE IMAGE
+        temp_path = "temp.jpg"
+
+        with open(temp_path, "wb") as f:
+            f.write(await file.read())
+
+        # READ IMAGE
+        image = cv2.imread(temp_path)
+
+        if image is None:
+            return {
+                "success": False,
+                "message": "Invalid image"
+            }
+
+        # =========================
+        # IMAGE QUALITY CHECK
+        # =========================
+
+        if is_blurry(image):
+            return {
+                "success": False,
+                "message": "Image is blurry"
+            }
+
+        if is_too_dark(image):
+            return {
+                "success": False,
+                "message": "Image is too dark"
+            }
+
+        # =========================
+        # OCR
+        # =========================
+
+        text = extract_text(temp_path)
+
+        if len(text.strip()) == 0:
+            return {
+                "success": False,
+                "message": "No readable text found"
+            }
+
+        # =========================
+        # DOCUMENT DETECTION
+        # =========================
+
+        document_type = detect_document(text)
+
+        if document_type is None:
+            return {
+                "success": False,
+                "message": "Rejected: Unsupported document"
+            }
+
+        return {
+            "success": True,
+            "document_type": document_type,
+            "extracted_text": text[:300]
+        }
+
+    except Exception as e:
+
+        return {
+            "success": False,
+            "message": str(e)
+        }