idvpackage 3.0.10__py3-none-any.whl → 3.0.11__py3-none-any.whl

idvpackage/common.py

@@ -909,8 +909,6 @@ def load_and_process_image_deepface(image_input, country=None):
             # Keep best fallback (just in case)
             
 
-        if country == "QAT":
-            return 0, 0
 
         if best_face_objs is None or best_confidence < CONFIDENCE_THRESHOLD:
             print(f"No valid face found (threshold={CONFIDENCE_THRESHOLD})")

idvpackage/ocr.py

@@ -206,27 +206,12 @@ class IdentityVerification:
             video_capture = cv2.VideoCapture(temp_video_file.name)
 
             if video_capture.isOpened():
-                frame_count = int(video_capture.get(cv2.CAP_PROP_FRAME_COUNT))
-
-                for _ in range(frame_count):
-                    ret, frame = video_capture.read()
-                    #                         if ret:
-                    # frame_count_vid+=1
-                    # if frame_count_vid % 10 == 0:
-                    _, buffer = cv2.imencode(".jpg", frame)
-                    image_data = buffer.tobytes()
-
-                    image = vision_v1.Image(content=image_data)
-
-                    response = self.client.face_detection(image=image)
-                    if len(response.face_annotations) >= 1:
-                        break
-
-            video_capture.set(cv2.CAP_PROP_POS_FRAMES, 0)
+                video_capture.set(cv2.CAP_PROP_POS_FRAMES, 0)
 
             selfie_result = self.extract_selfie_from_video(video_capture)
             if isinstance(selfie_result, dict):
                 video_quality["error"] = selfie_result["error"]
+
             else:
                 (
                     selfie_blurry_result,
@@ -241,22 +226,38 @@ class IdentityVerification:
                     video_quality["selfie"] = selfie_result
                     video_quality["shape"] = selfie_result.shape
 
-            video_capture.release()  # Release the video capture
-
-        # except Exception as e:
-        #     video_quality["error"] = "bad_video"
-
+            return video_quality
+    
+        except Exception as e:
+            logging.exception("check_document_quality failed")
+            video_quality["error"] = "face_not_clear_in_video"
+            return video_quality
+        
         finally:
-            # Ensure the temporary file is deleted
-            if os.path.exists(temp_video_file_path):
-                os.remove(temp_video_file_path)
-                # print(f"Temporary file {temp_video_file_path} has been deleted.")
+              
+            if video_capture is not None:
+                try:
+                    video_capture.release()
+                except Exception:
+                    pass
+                del video_capture
 
-        return video_quality
+            # Remove temp file
+            if temp_video_file_path and os.path.exists(temp_video_file_path):
+                try:
+                    os.remove(temp_video_file_path)
+                except Exception:
+                    logging.warning("Failed to delete temp file %s", temp_video_file_path)
+
+            # Force cleanup of OpenCV / numpy memory
+            import gc
+            gc.collect()
+        
 
     def extract_selfie_from_video(self, video_capture):
         """Extract the best quality selfie from video with speed optimizations for frontal faces."""
         video_dict = {'error': ''}
+        
 
         try:
             # Get rotation metadata from video
@@ -286,12 +287,12 @@ class IdentityVerification:
 
             best_face = None
             best_score = -1
-            best_frame = None
             best_frame_position = None
-            frame_results = []
+            best_frame = None
 
-            print(f"Analyzing video with {total_frames} frames")
-            print(f"Checking {len(frame_positions)} strategic frames")
+            logging.info(f"Analyzing video with {total_frames} frames")
+            logging.info(f"Checking {len(frame_positions)} strategic frames")
+            logging.info(f"Frame positions to analyze: {frame_positions}")
 
             for target_frame in frame_positions:
                 if target_frame >= total_frames:
@@ -299,6 +300,7 @@ class IdentityVerification:
                 if target_frame < 0:
                     target_frame = 0
 
+                logging.info(f"Processing frame {target_frame}")
                 video_capture.set(cv2.CAP_PROP_POS_FRAMES, target_frame)
                 ret, frame = video_capture.read()
                 if not ret or frame is None or frame.size == 0:
@@ -319,8 +321,17 @@ class IdentityVerification:
                     encode_params = [cv2.IMWRITE_JPEG_QUALITY, 90]
                     _, buffer = cv2.imencode(".jpg", small_frame, encode_params)
 
-                    image = vision_v1.Image(content=buffer.tobytes())
+                    # image = vision_v1.Image(content=buffer.tobytes())
+                    # response = self.client.face_detection(image=image, max_results=2)
+                    # faces = response.face_annotations
+
+                    image_bytes = buffer.tobytes()
+                    del buffer
+
+                    image = vision_v1.Image(content=image_bytes)
                     response = self.client.face_detection(image=image, max_results=2)
+                    del image_bytes
+
                     faces = response.face_annotations
 
                     if not faces:
@@ -431,40 +442,46 @@ class IdentityVerification:
                             }
 
                     if frame_best_face is not None:
-                        frame_results.append({
-                            'frame': target_frame,
-                            'face': frame_best_face,
-                            'score': frame_best_score,
-                            'frame_data': frame.copy()
-                        })
+
+                        
+                        # frame_results.append({
+                        #     'frame': target_frame,
+                        #     'face': frame_best_face,
+                        #     'score': frame_best_score,
+                        #     'frame_data': frame.copy()
+                        # })
+                        logging.info(f"Frame {target_frame}: Best face score {frame_best_score:.2f} "
+                                     f"(Frontal: {frame_best_face['frontal_score']:.2f}, "
+                                     f"Center: {frame_best_face['center_score']:.2f}, "
+                                     f"Confidence: {frame_best_face['confidence']:.2f})")
 
                         if frame_best_score > best_score:
+                            logging.info(f"New best face found at frame {target_frame} with score {frame_best_score:.2f}")
                             best_score = frame_best_score
                             best_face = frame_best_face
                             best_frame = frame.copy()
                             best_frame_position = target_frame
 
                 except Exception as e:
+                    logging.info(f"Error processing frame {target_frame}: {e}")
                     continue
 
-            # Process results
-            if len(frame_results) > 0:
-                # Sort faces by score
-                frame_results.sort(key=lambda x: x['score'], reverse=True)
+            # # Process results
+            # if len(frame_results) > 0:
+            #     # Sort faces by score
+            #     frame_results.sort(key=lambda x: x['score'], reverse=True)
 
-                for i, result in enumerate(frame_results[:min(3, len(frame_results))]):
-                    face_info = result['face']
-                    print(f"Rank {i+1}: Frame {face_info['frame']}, "
-                          f"Score: {result['score']:.2f}, "
-                          f"Frontal: {face_info['frontal_score']:.2f}, "
-                          f"Center: {face_info['center_score']:.2f}")
+                # for i, result in enumerate(frame_results[:min(3, len(frame_results))]):
+                #     face_info = result['face']
+                #     print(f"Rank {i+1}: Frame {face_info['frame']}, "
+                #           f"Score: {result['score']:.2f}, "
+                #           f"Frontal: {face_info['frontal_score']:.2f}, "
+                #           f"Center: {face_info['center_score']:.2f}")
 
                 # Use the best frame
-                best_result = frame_results[0]
-                best_face = best_result['face']
-                best_frame = best_result['frame_data']
+                # best_frame = frame_best_face['frame']
 
-                print(f"Selected frame {best_face['frame']} as best selfie")
+                # print(f"Selected frame {best_face['frame']} as best selfie")
 
             if best_face and best_frame is not None:
                 try:
@@ -493,9 +510,11 @@ class IdentityVerification:
                 return video_dict
 
         except Exception as e:
+            logging.info(f"Exception in extract_selfie_from_video: {e}")
             video_dict['error'] = 'video_processing_error'
             return video_dict
 
+        
 
     def is_colored(self, base64_image):
         img = self.image_conversion(base64_image)

{idvpackage-3.0.10.dist-info → idvpackage-3.0.11.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: idvpackage
-Version: 3.0.10
+Version: 3.0.11
 Summary: This repository contains a Python program designed to execute Optical Character Recognition (OCR) and Facial Recognition on images.
 Home-page: https://github.com/NymCard-Payments/project_idv_package
 Classifier: Programming Language :: Python :: 3

{idvpackage-3.0.10.dist-info → idvpackage-3.0.11.dist-info}/RECORD

@@ -1,6 +1,6 @@
 idvpackage/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 idvpackage/blur_detection.py,sha256=CtLqTtg2azE4YCM8AT1jYJq6QxGoniMO0VkQYIcMtRI,1883
-idvpackage/common.py,sha256=4KEG5WIPc2ntI93RLqyuImMjl-v4JbHRsI3BpNCMG7Y,47018
+idvpackage/common.py,sha256=xKnzPpPoXvYMIpJh4Afly6hoO__Ba13nUHODIScoCwI,46965
 idvpackage/constants.py,sha256=XvQy5ORGsXoQGcXcvUBC17vpFMzgzleS9S_mDN72G4c,7212
 idvpackage/ekyc.py,sha256=I7J2NTerBBnlxzgKkq5KVwgNtlAuEBeL3u_KgA1p2V8,2986
 idvpackage/genai_utils.py,sha256=u3n_dP53P6NJaCGu1d221S8DW1g9MoSq56SNh7VQeJQ,9273
@@ -12,7 +12,7 @@ idvpackage/lazy_imports.py,sha256=Bg5BPL2KYTztKu4gGgwt4-gFP9Mo2vZWs_UJ4zrL5Hw,11
 idvpackage/lebanon_id_extraction.py,sha256=eqWSpfb83RRk5TEhJNOuiPLt-wCLY5L4TvrpkOckSPs,21192
 idvpackage/lebanon_passport_extraction.py,sha256=AEOMRcFx1vVRjI2MeFRWdPHVj-jXSC5WZdpYEsQwn7U,7057
 idvpackage/liveness_spoofing_v2.py,sha256=6rp7R4n66ogiiQO7d0ZQzUQ7dIKGXWfD6OjXxq_DSg8,8822
-idvpackage/ocr.py,sha256=-1mRuEe3822oPRBkcfZUGtIMyQKq0xw8PZi5g-2k1L4,202435
+idvpackage/ocr.py,sha256=MYQBcJ7yWk5HQGCyAD_KRJWp8w2xVMlJxaSYAXbs8B8,203315
 idvpackage/ocr_utils.py,sha256=NmeGj7rhOzjg9sb_TSFcBBl8fJn4-INuHSvmijsLpiM,89845
 idvpackage/pse_passport_extraction.py,sha256=pI5eG9yxeh_q88IISP1QkEj8BqcsXmhCa48VS0N7El8,20369
 idvpackage/qatar_id_extraction.py,sha256=zIgKBLSTc-rKDy2b4CqM25JtLxQ9bvlyXeH-NZPh6_E,56718
@@ -35,8 +35,8 @@ idvpackage/spoof_resources/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJ
 idvpackage/spoof_resources/functional.py,sha256=164aGMvBDHU6HWRyDBr-2EUPeJ4rc4cXl10NJyGB7eE,20138
 idvpackage/spoof_resources/generate_patches.py,sha256=ANvJ5rDl-SXARchS8JwOxuKLhUuabbxLwvz4zsYbADE,1722
 idvpackage/spoof_resources/transform.py,sha256=3Ft9S6g6N0SU24f3feHXquh5qRc85JFt2gNKnCYQwjo,11311
-idvpackage-3.0.10.dist-info/licenses/LICENSE,sha256=JTmNGOPPvG2XBgkW2R2xwzJeR_OEjaFKeePN1jGYVt8,1068
-idvpackage-3.0.10.dist-info/METADATA,sha256=wGIDz5v2JPoSMEVnA2gR5rWfFKKCnkfxX-vmGFkRHLA,4800
-idvpackage-3.0.10.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-idvpackage-3.0.10.dist-info/top_level.txt,sha256=8jL1PsYvKBIWWLGe8jEm7kU5yZ5sld_OCaz0Dzvi6qY,11
-idvpackage-3.0.10.dist-info/RECORD,,
+idvpackage-3.0.11.dist-info/licenses/LICENSE,sha256=JTmNGOPPvG2XBgkW2R2xwzJeR_OEjaFKeePN1jGYVt8,1068
+idvpackage-3.0.11.dist-info/METADATA,sha256=xyf8tPQb68CZ1duGyrjcyHKvKFwaVv_acXOIzpwLKug,4800
+idvpackage-3.0.11.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+idvpackage-3.0.11.dist-info/top_level.txt,sha256=8jL1PsYvKBIWWLGe8jEm7kU5yZ5sld_OCaz0Dzvi6qY,11
+idvpackage-3.0.11.dist-info/RECORD,,