refactor

app.py

@@ -4,7 +4,7 @@ from fastapi import FastAPI, File, UploadFile, HTTPException
 from fastapi.responses import HTMLResponse, JSONResponse
 import os
 from fastapi.middleware.cors import CORSMiddleware
-from utils import crop_images_from_detections, detect_licensePlate, licence_dect, lookup_user, lookup_user_metadata, ner_recog, read_text_img, upload_to_s3, vehicle_dect, write_to_upload
+from utils import crop_images_from_detections, detect_licensePlate, upload_to_s3
 # from pinecone import Pinecone
 # from deepface import DeepFace
 
@@ -203,14 +203,14 @@ async def upload_files(
         front_url =  upload_to_s3(front_path)
         back_url =  upload_to_s3(back_path)
 
-        front_text =  read_text_img(front_path)
-        back_text =  read_text_img(back_path)
+        # front_text =  read_text_img(front_path)
+        # back_text =  read_text_img(back_path)
         
-        ent_front =  ner_recog(front_text)
-        ent_back =  ner_recog(back_text)
+        # ent_front =  ner_recog(front_text)
+        # ent_back =  ner_recog(back_text)
 
-        print(f"[*] ---- Entity front ----> {ent_front}")
-        print(f"[*] ---- Entity back ----> {ent_back}")
+        # print(f"[*] ---- Entity front ----> {ent_front}")
+        # print(f"[*] ---- Entity back ----> {ent_back}")
 
         if face is not None:
              # Save the face image to disk
@@ -219,9 +219,7 @@ async def upload_files(
                 face_file.write(await face.read())
             face_url = upload_to_s3(face_path)
             serial_number = ''
-            for i in ent_back["entities"]:
-                if 'serial' in i:
-                    serial_number = i['serial']
+            
             
             
             print(f"[*] --- Serial ---> {serial_number}")
@@ -251,7 +249,7 @@ async def upload_files(
 
 
 
-        return JSONResponse(content = {"message": "Upload successful", "data":{"text_front": f'{front_text}', 'front_url': front_url , 'entity_front': front_id, 'text_back': f'{back_text}', 'back_url': back_url,  'entity_back': back_id}}, status_code=200)
+        return JSONResponse(content = {"message": "Upload successful", "data":{'front_url': front_url , 'entity_front': front_id, 'back_url': back_url,  'entity_back': back_id}}, status_code=200)
     except Exception as e:
         return HTTPException(status_code=500, detail=f"Internal server error  {str(e)}")
 

utils.py

@@ -183,19 +183,10 @@ def detect_licensePlate(img: str) -> dict:
 
     return final
 
-    # if len(results > 1):
-    #     for result in results:
-    #         boxes = result.boxes.xyxy
-    #         for box in boxes.numpy():
-    #             x1, y1, x2, y2 = box[0], box[1], box[2], box[3]
-    #             cropped_image = image.crop((x1, y1, x2, y2))
-    #             color = 
 
 
 
 
-    return results
-
 
 def licence_dect(img: str) -> list:
     image = Image.open(img)
@@ -298,11 +289,7 @@ def vehicle_dect(img: str) -> any:
                 logger.info("[*]---- Color detecttion : f{color}")
 
                 logger.info("[*]---- Classes detecttion : f{classes}")
-            
-            # {'color': (70, 69, 74), 'plate': ['LT661HM CMR ']}
-            # ['car', 'car']
-            # {'color': (73, 82, 89), 'plate': []}
-            # ['car', 'car']
+        
 
         for i in range(len(classes)):
             final.append({ "type": classes[i], "car_data" : colors[i]})
@@ -313,12 +300,6 @@ def vehicle_dect(img: str) -> any:
         # raise(e)
         pass
 
-def write_to_upload( file, file_path, folder= "uploads"):
-    path = os.path.join(folder, file_path)
-    with open(path, "wb") as front_file:
-        front_file.write( file.read())
-    return path
-
 # res = licence_dect("cars.jpg")
 # print(res)