documentExtractionag051/ExtractDocument
0
1�
2 �`i� � � � S SK r S SKJr \ R " \ R S9 \ R3 " \5 r S\/0r SS jr S r4g)� N)�extract_invoice_tables)�level�Invoicec � � [ U 5 n[ /n[ R S[ U5 S35 U H2 n [ R SU R5 35 U " UUUUUUUS9nM4 U$ ! [ aD n6[ R SU R7 SU8 35 SSKn U R 5 Sn9A10M� Sn11A12ff = f) a� 13Main entry point for layout-based post-processing.14 15Args:16 input_data (dict): Original input JSON data17 validation_data (dict): Validation data structure18 ocr_elems (list): List of TextRect objects19 column_headers (dict): Optional custom column headers20 anchor_column (str): Optional anchor column name21 anchor_pattern (str): Optional anchor pattern regex22 advanced_options (dict): Optional advanced options23 24Returns:25 dict: Updated validation data26zRunning z handler(s)zRunning handler: )� ocr_elems�validation_data�page_dimensions�column_headers�
anchor_column�anchor_pattern�advanced_optionszHandler z failed: r N)27�_extract_page_dimensionsr �logger�info�len�__name__� Exception�error� traceback� print_exc) �28input_datar r r29 r r r
r � handler_list�handler�er s �hC:\Users\poona\OneDrive\Documents\nextlifyai\Document_Automation\Document_Extraction\custom_processor.py�do_custom_post_processingr , s� � �2 /�z�:�O� +�+�L�30�K�K�(�3�|�,�-�[�9�:� ��� � �'��(8�(8�'9�:�;�
"� %�#� /� /� -� +� -�!1� �O� �$ ��� � "� �L�L�8�G�$4�$4�#5�Y�q�c�B� C� � � � � !� !�� "�s � A/�/31B=�99B8�8B=c 32 �n � SU ; a� [ U R S5 [ 5 ( ak U S nU( a_ SUS ; aV [ U5 VVs0 s H= u p#UR SUS- 5 UR SS5 UR SS5 4_M? snn$ SU ; ax U R S0 5 R S0 5 R S/ 5 n[ U5 VVs0 s H- u p#US- UR SS5 UR SS5 4_M/ snn$ [ R S 5 0 $ s snnf s snnf )33zn34Extract page dimensions from input data.35 36Supports both simplified schema and original engine output format.37�pages�pageNumr � �width�height�38engineData�imagePreprocessingResultz!Could not extract page dimensions)�39isinstance�get�list� enumerater �warning)�datar �idx�ps r r r b s0 � � �$��:�d�h�h�w�&7��>�>��W�
�� �Y�%��(�*� (�� .�� .�F�C� ���i��q��)�A�E�E�'�1�,=�q�u�u�X�q�?Q�+R�R� .��
� �t�� �H�H�\�2� &�
�S�+�R�
0�
�S��"�
� � $�E�*�40�*���
�!�G�a�e�e�G�Q�'����x��);�<� <�*�41� 42�43 �N�N�6�7�
�I��%��44s
�AD+� 4D1)NNNN) �logging�invoice_post_processingr � basicConfig�INFO� getLoggerr r �LAYOUT_PROCESSORSr r � � r �<module>r5 s^ �� � :� � � �'�,�,� '� � � �8� $���< �&�'�� � ����3�l r4