Explicite/api-test
0
1�
2 g�id � �� � d dl mZ d dlZd dlmZ d dlZd dlmZmZ d dl m3Z4 dede ddfd �Z defd5�Z
d � Zd � Zddej defd�Zd� Zdededefd�Zdefd�Zdedefd�ZdS )� )�BytesION)�Image)�
HTTPException�6UploadFile)�settings�upload�7field_name�returnc � � | st dd|� ��� � �| j r| j � d� � st dd|� d��� � �d S )N� zFalta el archivo: �� status_code�detailzimage/�'z' debe ser una imagen.)r � content_type�8startswith)r r s �]/Users/alexiseduardojuarezjaimes/Desktop/Test api-face-extractor/app/services/face_service.py� ensure_imager sw � � � W���4U��4U�4U�V�V�V�V� � � \�f�&9�&D�&D�X�&N�&N� \���4Z�9�4Z�4Z�4Z�[�[�[�[�\� \� c � �� K � | � � � � d {V ��}|st dd�� � � t j t |� � � � S # t10 $ r t dd�� � �w xY w)Nr u La imagen está vacía.r
u* No pude leer la imagen. ¿Formato válido?)�readr �face_recognition�load_image_filer � Exception)r �raws r �read_as_np_imager s� � � � �� � �
�
�11 �12 �13 �14 �15 �16 �C� � O���4M�N�N�N�N�b� �/��� � �=�=�=�� � b� b� b���4`�a�a�a�a�b���s � A � A.c �� � t j | t j �� � }t |� � dk rt dd�� � �t |� � dk rt dd�� � �|d S )N)�modelr � u6 No se detectó ningún rostro en una de las imágenes.r
� u9 Se detectaron múltiples rostros en una de las imágenes.)r �face_locationsr �17face_model�lenr )�np_img�locss r �detect_single_facer& si � � � *�6��9L� M� M� M�D�18�4�y�y�A�~�~���4l�m�m�m�m�19�4�y�y�1�}�}���4o�p�p�p�p� ��7�Nr c � � |\ }}}}t j | � � }|j \ }}t d|� � }t d|� � }t ||� � }t ||� � }|� ||||f� � } ||||d�}20| |21fS )Nr )�top�right�bottom�left)r � fromarray�size�max�min�crop) r$ �locr( r) r* r+ �pil_img�w�h� face_crop�bboxs r � crop_facer7 s� � � "� �C�����o�f�%�%�G� �<�D�A�q�22
�a��+�+�C�c�!�T�l�l�t�
��F�^�^�F�S��E�]�]�U�� � �d�C��� 7�8�8�I���&�$� G� G�D� �d�?�r �_ r2 �qualityc �� � t � � }| � |d|�� � t j |� � � � � � d� � }|d|� �fS )N�JPEG)�formatr9 zutf-8zdata:image/jpeg;base64,)r �save�base64� b64encode�getvalue�decode)r2 r9 �buf�b64s r �img_to_base64_jpegrD , s] � �23�)�)�C� �L�L��V�W�L�5�5�5�24�25�3�<�<�>�>�26*�27*�281�291�'�30:�31:�C� �/�#�/�/� /�/r c �f � t j | |g�� � }|st dd�� � �|d S )N)�known_face_locationsr uM No pude generar el embedding del rostro (imagen borrosa / mala iluminación).r
r )r �face_encodingsr )r$ r1 �encss r � encode_facerI 2 sI � � � *�6��� N� N� N�D� � 32���b�33� 34� 35� 36� ��7�Nr �distance� tolerancec �H � | |k rdS d| |z z37 }t |dz d� � S )Ng g �?g Y@� )�round)rJ rK �scores r �distance_to_percentrP ; s5 � ��9� � ��s� �8�i�'� (�E� ��� �� "� "�"r �imagec � �� K � t | d� � t | � � � d {V ��}t |� � }t ||� � \ }}t |� � \ }}|d||d�S )NrQ �jpeg)r6 r< � face_base64�data_uri)r r r& r7 rD )rQ r$ r1 r5 r6 �face_b64rU s r �extract_face_from_uploadrW B s� � � � ���� � � �#�E�*�*�
*�
*�
*�
*�
*�
*�F�38 �V�39$�40$�C� ���,�,�O�I�t�+�I�6�6��H�h� �� � � � � r �image1�image2c � � K � t | d� � t |d� � t | � � � d {V ��}t |� � � d {V ��}t |� � }t |� � }t ||� � \ }}t |� � \ }} t ||� � }41t ||� � } t
t j � � } t
t j42 |43g| � � d � � }
t t j |44g| | �� � d � � }t |
| � � }| t |
d� � |||| d�S )NrX rY r )rK � )rK rJ �is_match�similarity_percent� image1_bbox�image1_face_data_uri)r r r& r7 rD rI �floatr rK r �
face_distance�bool�
compare_facesrP rN )rX rY �np1�np2�loc1�loc2�45face1_crop�bbox1�_�face1_data_uri�enc1�enc2rK rJ r\ r] s r �compare_two_uploadsrn Q s[ � � � ����"�"�"����"�"�"� ��(�(�46(�47(�48(�49(�50(�51(�C� ��(�(�52(�53(�54(�55(�56(�57(�C� �c� "� "�D� �c� "� "�D� "�#�t�,�,��J��*�:�6�6��A�~� �s�D� !� !�D� �s�D� !� !�D��h�(�)�)�I��%�3�T�F�D�A�A�!�D�E�E�H��$�2�D�6�4�9�U�U�U�VW�X�Y�Y�H�,�X�y�A�A�� ��(�A�&�&� � 0� � .�
� � r )r8 )�ior r> �PILr r �fastapir r �app.core.configr �strr r r& r7 �intrD rI r` rP rW rn � r r �<module>rv s~ �� � � � � � �
�
�
�
� � � � � � � � � � � -� -� -� -� -� -� -� -� $� $� $� $� $� $�\�� \�� \�� \� \� \� \� b�:� b� b� b� b�� � �58� 59� 60�0� 0�� � 0�c� 0� 0� 0� 0� � � �#�%� #�E� #�e� #� #� #� #�
�*�
�
�
�
� �j� �*� � � � � � r