Update app.py

app.py

@@ -1,4 +1,5 @@
-from fastapi import FastAPI, UploadFile, File
+from fastapi import FastAPI
+from pydantic import BaseModel
 import easyocr
 import cv2
 import numpy as np
@@ -13,13 +14,23 @@ app = FastAPI()
 
 reader = easyocr.Reader(['en'])
 
+# =========================
+# REQUEST MODEL
+# =========================
+
+class ImageRequest(BaseModel):
+    image_path: str
+
 # =========================
 # IMAGE QUALITY CHECKS
 # =========================
 
 def is_blurry(image):
 
-    gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
+    gray = cv2.cvtColor(
+        image,
+        cv2.COLOR_BGR2GRAY
+    )
 
     variance = cv2.Laplacian(
         gray,
@@ -62,7 +73,7 @@ def detect_document(text):
 
     # REMOVE EXTRA SYMBOLS
     cleaned_text = re.sub(
-        r'[^a-zA-Z0-9\\s-]',
+        r'[^a-zA-Z0-9\s-]',
         ' ',
         text
     )
@@ -200,7 +211,7 @@ def detect_document(text):
         }
 
     # =========================
-    # ELECTRICITY COMPANIES
+    # ELECTRICITY / UTILITY BILL
     # =========================
 
     electricity_keywords = [
@@ -235,7 +246,7 @@ def detect_document(text):
         "yedc",
         "yola electricity",
 
-        # Common Nigerian utility terms
+        # Common terms
         "prepaid",
         "postpaid",
         "disco",
@@ -421,9 +432,9 @@ def detect_document(text):
 
     plate_patterns = [
 
-        r"[A-Z]{3}-?\\d{3}[A-Z]{2}",
-        r"[A-Z]{2}\\d{3}[A-Z]{3}",
-        r"[A-Z]{3}\\s\\d{3}\\s[A-Z]{2}"
+        r"[A-Z]{3}-?\d{3}[A-Z]{2}",
+        r"[A-Z]{2}\d{3}[A-Z]{3}",
+        r"[A-Z]{3}\s\d{3}\s[A-Z]{2}"
     ]
 
     detected_plate = None
@@ -470,6 +481,7 @@ def detect_document(text):
 
     return None
 
+
 # =========================
 # HOME ROUTE
 # =========================
@@ -499,19 +511,27 @@ def home():
 
 @app.post("/validate")
 async def validate_document(
-    file: UploadFile = File(...)
+    request: ImageRequest
 ):
 
     try:
 
+        image_path = request.image_path
+
         # =========================
-        # SAVE FILE
+        # CHECK FILE EXISTS
         # =========================
 
-        image_path = "temp.jpg"
+        if not os.path.exists(image_path):
 
-        with open(image_path, "wb") as f:
-            f.write(await file.read())
+            return {
+                "success": False,
+                "message": "Image not found",
+                "reason": (
+                    "The provided image path "
+                    "does not exist."
+                )
+            }
 
         # =========================
         # READ IMAGE
@@ -524,8 +544,13 @@ async def validate_document(
             return {
                 "success": False,
                 "message": "Invalid image",
-                "reason": "The uploaded file could not be read as an image.",
-                "suggestion": "Upload a valid JPG or PNG image."
+                "reason": (
+                    "The file could not be "
+                    "read as an image."
+                ),
+                "suggestion": (
+                    "Provide a valid JPG or PNG image."
+                )
             }
 
         # =========================
@@ -537,8 +562,11 @@ async def validate_document(
             return {
                 "success": False,
                 "message": "Image rejected",
-                "reason": "The uploaded image is blurry.",
-                "suggestion": "Retake the photo with better focus."
+                "reason": "The image is blurry.",
+                "suggestion": (
+                    "Retake the photo with "
+                    "better focus."
+                )
             }
 
         # =========================
@@ -550,8 +578,10 @@ async def validate_document(
             return {
                 "success": False,
                 "message": "Image rejected",
-                "reason": "The uploaded image is too dark.",
-                "suggestion": "Take the photo in a brighter environment."
+                "reason": "The image is too dark.",
+                "suggestion": (
+                    "Use better lighting."
+                )
             }
 
         # =========================
@@ -569,8 +599,13 @@ async def validate_document(
             return {
                 "success": False,
                 "message": "Document rejected",
-                "reason": "No readable text was detected in the image.",
-                "suggestion": "Ensure the document is clear and fully visible."
+                "reason": (
+                    "No readable text was detected."
+                ),
+                "suggestion": (
+                    "Ensure the document is "
+                    "clear and fully visible."
+                )
             }
 
         # =========================
@@ -589,8 +624,8 @@ async def validate_document(
                 "success": False,
                 "message": "Document rejected",
                 "reason": (
-                    "The uploaded image does not match any "
-                    "supported document type."
+                    "The uploaded image does not "
+                    "match any supported document type."
                 ),
                 "supported_documents": [
                     "National ID (NIN)",
@@ -621,8 +656,12 @@ async def validate_document(
         return {
             "success": True,
             "message": "Document verified successfully",
-            "document_type": document_result["document_type"],
-            "confidence": document_result["confidence"],
+            "document_type": (
+                document_result["document_type"]
+            ),
+            "confidence": (
+                document_result["confidence"]
+            ),
             "matched_keywords": (
                 document_result["matched_keywords"]
             ),
@@ -637,11 +676,17 @@ async def validate_document(
             "reason": str(e)
         }
 
-    finally:
 
-        # =========================
-        # CLEAN TEMP FILE
-        # =========================
+# =========================
+# RUN SERVER
+# =========================
+
+if __name__ == "__main__":
+
+    import uvicorn
 
-        if os.path.exists("temp.jpg"):
-            os.remove("temp.jpg")
+    uvicorn.run(
+        app,
+        host="0.0.0.0",
+        port=7860
+    )