U bh@sddlZddlmZmZmZmZmZmZmZddl m Z ddl m Z ddl mZmZddlmZmZmZmZe eeZedejd<d ejd <d ejd <d ejd<ejejd ddejejd ddddddhZddZddZedddZedddZedd d!Z ed"d#d$Z!ejd%d&gd'd(d)Z"ejd*d&gd'd+d,Z#ed-kr|ej$dd.d/dS)0N)Flaskrequestrender_templateredirecturl_forjsonifysend_from_directory)secure_filename) load_dotenv)db OcrResult)combine_images_to_pdfdetect_and_extract_facecall_gemini_ocrconvert_pdf_first_page_to_image SECRET_KEYZuploads UPLOAD_FOLDERZextracted_facesEXTRACTED_FACES_FOLDERiMAX_CONTENT_LENGTHT)exist_okpngjpgjpegpdfcCs d|ko|dddtkS)N.)rsplitlowerALLOWED_EXTENSIONSfilenamer"*/var/www/html/id_scanner_app_1-main/app.py allowed_file'sr$cCs|d}|d}|d}d}d}d}|r|jdkrt|jrt|j}tjtjd|}| ||}|}| ddd } | dkr|}n,| d krt d t |tjd}|st d n|rn|rnt|jrnt|jrnt|j} t|j} tjtjd| } tjtjd| } | | | | | }t| | tjd\}}|r`|}|}n d d idfSn d didfS|sd didfSt|}d |kr|dfSd}|rt|tjd}t|||d}tj|tjd|||jddfS)N single_file side1_file side2_filerrr)rrrrzDPDF detected, attempting to convert first page for face detection...z4Could not extract image from PDF for face detection.errorz!Failed to combine images into PDFzIInvalid file submission. Please provide a single file or two side images.z(File to process could not be determined.ir)original_filenameextracted_dataZface_image_pathT)ZsuccessdataZ face_imageid)getr!r$r ospathjoinappconfigsaverrprintrr rrr r sessionaddZcommitr.)filesr%r&r'Zfile_to_processr+Zface_source_image_pathr!filepathZfile_extensionZ s1_filenameZ s2_filenameZ s1_filepathZ s2_filepathZpdf_pathZ pdf_filenamer,Z face_pathZ new_resultr"r"r# process_files+sj      $         r</cCstdS)Nz index.html)rr"r"r"r#indexrsr>z/historycCs"tjtj}td|dS)Nz history.htmlresults)r queryZorder_byZ processed_atZdescallrr?r"r"r#historyvsrCz/uploads/cCsttjd|S)Nrrr4r5r r"r"r# serve_upload{srEz /extracted_faces/cCsttjd|S)NrrDr r"r"r# serve_facesrFz/uploadPOST)methodscCstjs dStjdtjdtjdd}t|\}}|dkrNttdSd}t|trnd |krn|d }nt|tr||}||fSdS) N)z No file partr*r%r&r'r%r&r'r/rCzAn unknown error occurred.r)) rr:r0r<rr isinstancedictstr)r:result status_codeZ error_messager"r"r#upload_from_forms       rOz/api/ocrcCsRtjstddidfStjdtjdtjdd}t|\}}t||fS)Nr)zNo file part in the requestr*fileZside1Zside2rI)rr:rr0r<)r:rMrNr"r"r#upload_from_apis    rQ__main__z0.0.0.0)debughost)%r1flaskrrrrrrrZwerkzeug.utilsr dotenvr Zmodelsr r utilsr rrr__name__r4urandomr5makedirsrr$r<router>rCrErFrOrQrunr"r"r"r#s<$      G