Explicite/api-test
0
1�
2 �_�i� � �. � d dl mZ d dlZd dlZd dlmZ d dlmZm Z d dl3Z4d dl m Z d dl
mZmZmZmZmZ d dlmZ d dlmZ d e d5ed dfd �Zd e d ej2 fd
�Zdej2 defd�Z ddej de fd�Z de de de d e fd�Z de de fd�Z!y) � )�BytesION)�Image)�
HTTPException�6UploadFile)�settings)� get_face_app�pick_best_face� to_bbox_dict�cosine_similarity� augment_rgb)�classify_document)�assess_quality�upload�7field_name�returnc � � | st dd|� ��� �| j r| j j d� st dd|� d��� �y )N� zFalta el archivo: �� status_code�detailzimage/�'z' debe ser una imagen.)r � content_type�8startswith)r r s �X/Users/alexiseduardojuarezjaimes/Desktop/api-face-extractor/app/services/face_service.py� ensure_imager sS � � ���6H�� �4U�V�V� � � �f�&9�&9�&D�&D�X�&N���a�9�|�CY�4Z�[�[� 'O� c � � K � | j � � d { ��� }|s
t dd�� � t j t |� � j d� }t
j |� S 7 �V# t $ r t dd�� �w xY w�w)Nr u La imagen está vacía.r �RGBu* No pude leer la imagen. ¿Formato válido?) �readr r �openr �convert�np�array� Exception)r �raw�pils r �read_as_np_rgbr' s{ � �� �� � �
�10 �C� ���4M�N�N�b��j�j��� �&�.�.�u�5���x�x��}� � �� � b���4`�a�a�b�s' �B �A-�B �AA/ �,B �/B�B �np_rgb�bboxc � � t j | � }|j \ }}|d |d |d |d f\ }}}}t d|� }t d|� }t ||� }t ||� }|j ||||f� S )N�top�right�bottom�leftr )r � fromarray�size�max�min�crop) r( r) �pil_img�width�heightr+ r, r- r. s r � crop_facer7 s� � ��o�o�f�%�G��L�L�M�E�6� #�E�{�D��M�4��>�4�PV�<� W� �C����11
�a��+�C�c�!�T�l�t�
���
�F�#�e�U�*;�%� �<�<��s�E�6�2� 3�3r r4 �qualityc � � t � }| j |d|�� t j |j � � j d� }|d|� �fS )N�JPEG)�formatr8 zutf-8zdata:image/jpeg;base64,)r �save�base64� b64encode�getvalue�decode)r4 r8 �buf�b64s r �img_to_base64_jpegrC * sO � �12�)�C� �L�L��V�W�L�5�13�14�15�3�<�<�>�16*�171�181�'�19:�C� �)�#��/� /�/r �sim�mid�kc � � t dt dt | � � � } | |z20 |z }ddt j | � z z }t |dz d� S )z�21 sim suele estar entre ~0.0 y ~1.0 (cosine similarity).22 NO es probabilidad. Esto es un mapeo UI-friendly con curva sigmoid.23 g � �?g Y@� )r1 r2 �float�math�exp�round)rD rE rF �x�ys r � similarity_to_percent_sigmoidrP 0 sR � �24 �c�3�s�E�#�J�'�25(�C� �s��a��A� �s�T�X�X�q�b�\�!�"�A� ��U��A� � r �image1�image2c � � K � t | d� t |d� t | � � d { ��� }t |� � d { ��� }t |� }t � }|j |� }|j |� }t |� }t |� } |�| �
t
dd�� �t |� }26| j } |j } t ||27� }
t |
� \ }}t || t j t j t j t j �� }t || t j t j t j t j �� }|j" }| j" }t% ||� }|g}t j& r�t j& dkD r�t) t j& dz28 � D ]� }t+ |d|z � � }t+ |d29|z � � }t |j |� � }t |j |� � }|�|��\t% |j" |j" � }|j- |� �� t j. r t1 |� n0t3 t5 j6 t5 j8 |� � � } t; | t j< t j> � } t j@ } |jB r |jB sd } d } n | | k\ } | rd
nd} t3 | � tE dt3 | � z30 d� tG | � t3 | � tE t3 | � d� | tI tK |� � |D �!cg c] }!tE t3 |!� d� �� c}!|31|||jB tE |jL d� tE |jN d� tE |jP d� |jR d�|jB tE |jL d� tE |jN d� tE |jP d� |jR d�d�
S 7 ��7 ��32c c}!w �w)NrQ rR i� u. No se detectó rostro en una de las imágenes.r )�min_face_area_ratio� blur_var_min�brightness_min�brightness_max� i� )�seedi� � low_qualityF�match�no_matchrH � rI )�ok�blur_var�brightness_mean�face_area_ratio�signals)
� tolerance�distance�is_match�similarity_percent�raw_similarity�decision� samples_used�samples� image1_bbox�image1_face_data_uri� document1�quality1�quality2)*r r' r
r �getr r r33 r) r7 rC r r rT rU rV rW �normed_embeddingr �
multi_samples�ranger �append�use_best_of_samplesr1 rJ r" �medianr# rP � score_mid�score_k�
sim_thresholdr^ rM �bool�int�lenr_ r` ra rb )"rQ rR �np1�np2�doc1�face_app�faces1�faces2�f1�f2�bbox1�34bbox2_xyxy�35bbox1_xyxy�36face1_crop�_�face1_data_uri�q1�q2�emb1�emb2�base_sim�sims�i�np1_aug�np2_aug�fa1�fa2�s�raw_simrf � thresholdrh re rN s" r �compare_two_uploadsr� : s� � �� ����"����"� �v�&�37&�C� �v�&�38&�C� �S� !�D��~�H� �\�\�#�
�F�
�\�\�#�
�F� �� �B� �� �B� �z�R�Z���4d�e�e�
�� �E����J����J� �3��&�J�*�:�6��A�~� 39� �Z� $� 8� 8� �*�*� �.�.� �.�.� 40�B� 41� �Z� $� 8� 8� �*�*� �.�.� �.�.� 42�B� � � �D�
� � �D� ��t�,�H�
�:�D�� � �(�"8�"8�1�"<��x�-�-��1�2�A�!�#�D�1�H�5�G�!�#�D�1�H�5�G� �� � �g�!6�7�C� �� � �g�!6�7�C��{�c�k��!�#�"6�"6��8L�8L�M�A� �K�K��N� 3� $�7�7�c�$�i�U�2�9�9�R�X�X�VZ�^�C\�=]�G� 7�w��@R�@R�T\�Td�Td�e�� �&�&�I� �E�E�b�e�e� �����i�'�� &�7�J�� �9�%��#��g��.��2���N� #�$6� 7� ��g���2� ��C��I��04�5��1�E�%��(�A�&��5� � .�� �%�%� �b�k�k�1�-� $�R�%7�%7�� ;� $�R�%7�%7�� ;��z�z� 43� �%�%� �b�k�k�1�-� $�R�%7�%7�� ;� $�R�%7�%7�� ;��z�z� 44�- � �] '��45&��j 6�s3 �'Q�Q�Q�Q�L4Q�1 Q �
B7Q�Q� Q)�_ )"�ior r= �numpyr" �PILr �fastapir r rK �app.core.configr � app.services.insightface_enginer r r46 r r �app.services.doc_classifierr
�app.services.face_qualityr �strr �ndarrayr' �dictr7 r{ rC rJ rP r� � r r �<module>r� s� �� �
� � � -� � $�� � :� 4�\�� \�� \�� \� b�� b��47�48� b�4�b�j�j� 4�� 4�0�� � � 0�c� 0� �u� �5� �U� �u� �o�j� o�*� or