Update main.py

main.py

@@ -1,9 +1,82 @@
+import re
+import cv2
+import json
+import uvicorn
+import numpy as np
+from fastapi import FastAPI, File, UploadFile, HTTPException
+from fastapi.responses import Response
+from fastapi.middleware.cors import CORSMiddleware
+from paddleocr import PaddleOCR
+
+# ✅ CREATE APP FIRST (VERY IMPORTANT)
+app = FastAPI(title="AI Aadhaar Masking API")
+
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+# ✅ OCR INIT
+ocr = PaddleOCR(use_angle_cls=True, lang='en')
+
+# -------------------------------
+# Enhance Image
+# -------------------------------
+def enhance(img):
+    gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
+    return cv2.createCLAHE(clipLimit=2.0, tileGridSize=(8,8)).apply(gray)
+
+# -------------------------------
+# Clean Digits
+# -------------------------------
+def clean_digits(text):
+    text = text.upper() \
+        .replace('O','0') \
+        .replace('I','1') \
+        .replace('L','1') \
+        .replace('S','5') \
+        .replace('B','8')
+    return re.sub(r'[^0-9]', '', text)
+
+# -------------------------------
+# Mask Function
+# -------------------------------
+def mask(img, bbox, ratio):
+    p1 = tuple(map(int, bbox[0]))
+    p3 = tuple(map(int, bbox[2]))
+    width = p3[0] - p1[0]
+    mask_w = int(width * ratio)
+    cv2.rectangle(img, p1, (p1[0]+mask_w, p3[1]), (0,0,0), -1)
+    return img
+
+# -------------------------------
+# QR Mask
+# -------------------------------
+def mask_qr(img):
+    detector = cv2.QRCodeDetector()
+    data, bbox, _ = detector.detectAndDecode(img)
+    if bbox is not None:
+        bbox = bbox[0].astype(int)
+        x1, y1 = bbox[0]
+        x2, y2 = bbox[2]
+        cv2.rectangle(img, (x1,y1), (x2,y2), (0,0,0), -1)
+    return img
+
+# -------------------------------
+# MAIN API
+# -------------------------------
 @app.post("/v1/aadhaar/process")
 async def process(file: UploadFile = File(...)):
     try:
         contents = await file.read()
         img = cv2.imdecode(np.frombuffer(contents, np.uint8), cv2.IMREAD_COLOR)
 
+        if img is None:
+            raise HTTPException(status_code=400, detail="Invalid image")
+
         # QR Mask
         img = mask_qr(img)
 
@@ -13,17 +86,19 @@ async def process(file: UploadFile = File(...)):
 
         img = mask_qr(img)
 
+        # Enhance
         processed = enhance(img)
 
-        # OCR (AI)
+        # OCR
         ocr_result = ocr.ocr(processed)
 
-        # ✅ FIX 1: MUST BE INSIDE TRY
+        # ✅ FIX: Safe check
         if not ocr_result:
             raise HTTPException(status_code=400, detail="No text detected")
 
         results = []
 
+        # ✅ SAFE LOOP
         for line in ocr_result:
             if not line:
                 continue
@@ -37,11 +112,12 @@ async def process(file: UploadFile = File(...)):
                 except:
                     continue
 
-        # ✅ FIX 2: outside loop
+        # ✅ OUTSIDE LOOP
         extracted = {"aadhaar": None, "vid": None}
 
         for i, (bbox, text, conf) in enumerate(results):
 
+            # Ignore low confidence
             if conf < 0.6:
                 continue
 
@@ -69,6 +145,7 @@ async def process(file: UploadFile = File(...)):
                     img = mask(img, bbox, 1.0)
                     continue
 
+        # Encode image
         _, buffer = cv2.imencode('.jpg', img)
 
         return Response(
@@ -78,4 +155,11 @@ async def process(file: UploadFile = File(...)):
         )
 
     except Exception as e:
-        raise HTTPException(status_code=500, detail=str(e))
+        raise HTTPException(status_code=500, detail=str(e))
+
+
+# -------------------------------
+# RUN
+# -------------------------------
+if __name__ == "__main__":
+    uvicorn.run(app, host="0.0.0.0", port=8000)