o
    6Hf5!                     @   s  d dl Z d dlZd dlZd dlZd dlmZmZ d dlm	Z	 d dl
mZ d dlmZmZmZmZmZ d dlmZ e  Ze jeddZe jesOe e eeZeejd	< ed
dd Zejddgddd Z ejddgddd Z!ejddgddd Z"dS )    N)Flaskrequest)Image)BytesIO)rotate_imageresponse_formatterallowed_file	face_evalface_eval_regis)transusers
attendanceUPLOAD_FOLDER/c                   C   s   dS )Nzhello world r   r   r   /var/www/html/face-recog/app.py
helloWorld   s   r   z/compare_facesPOST)methodsc               
   C   s  zjt jdd} dt jvsdt jvrt|  d }t|dW S t jd }t jd }t|jr3t|js?t|  d }t|dW S t	|}t	|}t
|}t
|}|slt|}t
|}|slt|  d }t|dW S |st|}t
|}|st|  d	 }t|dW S t|d
krt||dd| d}|d dkrt|d |d W S nt|d
krtt|  d ddW S tj|ddd }	tj|ddd }
tj|	g|
dd}tjtjd |jdd }tj|st| dd t|D }|d r!t|dk rtj|tdd }|d || d}t|  d  }n@d}t|  d! }t|dkra|D ].}t	tj||}tj|ddd }	tj|	g|
dd}|d r_d}t|  d  } nq2t||t|d W S  ty } ztt|dW  Y d }~S d }~ww )"NzAccept-Languageenimage_1image_2req_file_namefailedallowed_extensionface_1_not_foundface_2_not_found   :  U   )min_wide_face_arealaplacian_thresoldlangresultFmessagestatusonly_one_facelargemodelr   g      ?	tolerancer   .c                 S   s   g | ]	}| d r|qS ).jpg)endswith).0fr   r   r   
<listcomp>W   s    z$match_attendance.<locals>.<listcomp>   z%Y%m%d_%H%M%Sr-   success
face_matchface_not_match) r   headersgetfilesr   r   r   filenameface_recognitionload_image_fileface_locationsr   lenr	   face_encodingscompare_facesospathjoinappconfigsplitexistsmakedirslistdirtimestrftimeseeksavebool	Exceptionstr)r"   r$   req_image_1req_image_2r   r   face_locations_1face_locations_2eval_attendanceimage_1_encodingsimage_2_encodingsmatch_resultfolder_pathexisting_filesrA   r%   r0   er   r   r   match_attendance   s   







 




r[   z/find_facesc               
   C   s  zmt jdd} t jd }t|}t|}|s9|d t|	 }t
|d}|d tt|}t|dkrTt||dd	d
}t|d |d |d W S t|dkrdtt|  d dW S tt|  d dW S  ty } ztt|dW  Y d }~S d }~ww )Nr"   r   imager   Z   RGBr   r   r   )r    r!   r$   r%   r#   r&   r   face_not_found)r   formr7   r8   r:   r;   r<   rK   r   readr   openrotateconvertnparrayr=   r
   r   r   rN   rO   )r"   	req_imageloadedImagefaces
file_bytesr\   datarZ   r   r   r   
find_facesw   s*   




rl   z/testingc                  C   s  dt jvs
dt jvrd} t| dS t jd }t jd }t|jr%t|js,d} t| dS t|}t|}t|}t|}|sId} t| dS |sRd} t| dS tj|dd	d
 }tj|dd	d
 }tj	|g|dd}	t
|	 |	d
 rzd}
d} nd}
d} t| |
t|	d
 S )Nr   r   z/request file must have name image_1 and image_2r   z?Hanya ekstensi berikut yang diperbolehkan: jpg, jpeg, png, webpz'Faces not detected in the source image.z*Faces not detected in the requested image.r'   r(   r   g(\?r*   r3   zFacial recognition is match.z"Facial recognition does not match.)r   r8   r   r   r9   r:   r;   r<   r>   r?   printrM   )r$   rP   rQ   r   r   rR   rS   rU   rV   rW   r%   r   r   r   testing   s8   









rn   )#r@   rI   r:   numpyre   flaskr   r   PILr   ior   utilsr   r   r   r	   r
   translationsr   getcwddirnamerA   rB   attendance_folderrF   rG   __name__rC   rD   router   r[   rl   rn   r   r   r   r   <module>   s(     



^
&