Ng ddlmZddlmZmZddlmZddlmZm Z ddl m Z m Z ddl mZerddlmZ d%d&dZ d'd(dZ d)d*d!Zd+d$ZdS),) annotations) TYPE_CHECKINGOptional)Source) TextRegion TextRegions) LayoutElementpartition_groups_from_regions) ElementType) RectangleNx1 int | floaty1x2y2text Optional[str]sourceOptional[Source]returnrc6tj||||||S))rr)r from_coords)r rrrrrs l/var/www/html/ai-engine/env/lib/python3.11/site-packages/unstructured/partition/pdf_image/inference_utils.pybuild_text_region_from_coordsrs"  !"b"btF K K KKbbox 'Rectangle' element_typer c(t||||S)r)rrrtype)r rrrrs rbuild_layout_elementr#s df< P P PPrF ocr_regionslist[TextRegion]ocr_textgroup_by_ocr_textboollist[LayoutElement]c|r|d}g}|D]}g}|dd}|D]>}|sn9|j|vr/||||j?|sp|D]} || |t j|n!tt j|}d|D} d| DS)z. Get layout elements from OCR regions z   c,g|]}t|S)merge_text_regions).0groups r z:build_layout_elements_from_ocr_regions..Is!MMME(//MMMrcfg|].}t|j|j|jtj/S)r")r#rrrr UNCATEGORIZED_TEXT)r0rs rr2z:build_layout_elements_from_ocr_regions..JsK     afQXKDb      r)splitreplacerappendremover from_listr ) r$r&r' text_sectionsgrouped_regions text_sectionregionswords ocr_regionr5merged_regionss r&build_layout_elements_from_ocr_regionsrB)sS\ v.. ) C CLG ((s3399;;E) 2 2 E?e++NN:...LL111  & &""1%%%%  " ";#8#A#A B B B B! C$8 8Mk8Z8Z[[MM_MMMN      rr>rc4|std|jt}|jt}|jt}|jt}d d|j D}|j }tj ||||||S)z Merge a list of TextRegion objects into a single TextRegion. Parameters: - group (TextRegions): A group of TextRegion objects to be merged. Returns: - TextRegion: A single merged TextRegion object. z/The text regions to be merged must be provided.r,cg|]}||Sr.r.)r0rs rr2z&merge_text_regions..esCCCTdCDCCCr) ValueErrorr minastypefloatrrmaxrjointextsrrr)r>min_x1min_y1max_x2max_y2 merged_textrs rr/r/Rs LJKKK Z^^   $ $U + +F Z^^   $ $U + +F Z^^   $ $U + +F Z^^   $ $U + +F((CCW]CCCDDK ^F  !&&&&+v V VVr)NN)r rrrrrrrrrrrrr)NNN) rrrrrrrrrr )NF)r$r%r&rr'r(rr))r>rrr) __future__rtypingrr unstructured_inference.constantsr)unstructured_inference.inference.elementsrr.unstructured_inference.inference.layoutelementr r unstructured.documents.elementsr r rr#rBr/r.rrrWsE""""""********333333MMMMMMMM 877777DCCCCCC# L L L L L#"& QQQQQ##&&&&&RWWWWWWr