j<|ddlmZddlZddlZddlZddlmZddlmZej e Z GddZ dS)) annotationsN)Path)AnycBeZdZ ddd Z dddZ dddZdS)DrawingPipelineRunnerMixinN file_pathrfilenamestr project_id document_id drawing_hintreturndictc ddlm}t|}||||ds|dr|nd|t|d|jd|d}|j||S)Nruuid4zDWG-DWG_z /storage/pending)idr r drawing_coder file_urltakeoff_statusr)uuidrr startswithname _erp_clientcreate_drawing_takeoff) selfr r r r rr takeoff_idpayloads Ec:\Projects\DSCons\app\modules\takeoff\application\pipeline_runner.pycreate_pending_takeoffz1DrawingPipelineRunnerMixin.create_pending_takeoff s \\ $&##F++#/7/B/B6/J/J#HH"""Y4IN44'(    //888r attached_hsmt_path str | Noneattached_hsmt_filenamec()Kddlm} |j|}|sdddS|j|ddi|d((rt j(s"|j|d d d d diSt(j }|d kr!tj |j(d{V}n|dkr!tj |j(d{V}ng|dkr1tj |jt(dd|d{V}n0tj |jt(dd|d{V}|r|dni} d} r|t jr] ddl))fd} | tj | d{Vz } n#t$$r'} t&d| Yd} ~ nd} ~ wwxYw|dkr] ddl)()fd} | tj | d{Vz } nJ#t$$r'} t&d| Yd} ~ nd} ~ wwxYw| dd} | dd}|r6|ddr|ddn|rd|nd}d|D}|||dd| ||(|d{V}|dg}|dp|dpd }t-|d!r||||"}n|}|d#d$}d%}d%}d%}d%}d%} |j|n4#t$$r'} t&d&| Yd} ~ nd} ~ wwxYwg}t3|D]\}}t5|}t7|d'pd}t7|d(p|d)pd} t7|d*p&|d+pt9|| z}!|d,d-}"|d.pd }#d/|#vsd0|#vrBd1}"|d2t5|d3d vr|nd%z }nd4|#vsd5|#vrd6}"|d7t5|d3d vs7d8t5|d3d vr>d8t5|d3d vr|n|d9z nd%z }nd:|#vr}d;}"|dd=t5|d3d vr|d>zn|nd%z }nPd?|#vs d@|#vsdA|#vsdB|#vr@dC}"|d2t5|d3d vr|nd%z }||||dDz|dDz|dEpdF|dDzdG|dHpdI|dDzdG|d.pdJ|dDz|"|dKpdL|dM|d3d|d3d|| |!|dNp5dO|d.pdddPdMdQdR||!z }t-|jdSr|j|n|D]}$|j |$|dTkrdUndV}%dWtC|dX|dYdZ}&r|nd}'tE|tFr|d[r|d[}%|d#r|d#}|d\r|d\}&|d]r|d]}'|j|d^|||%||'|&d_tC|d`|t9|dat9|dt9|dat9|dadb t&$dc|ddtC|de|dYdfdg|tC||dhdiS#t$$r_} t&%dj|dd| dkl|j|d dm| d dt5| dcYd} ~ Sd} ~ wwxYw)nNrrerrorzTakeoff not found)statusdetailr processingr erroredzFile not found)rai_analysis_notesr*z.dwgz.dxfz.pdfrcd}t}ttdt |D]#}|||ddzz }$||SNr2text openr rangeminlenget_textclose)t_layerhsmt_docir%pymupdfs r"extract_hsmt_textzJDrawingPipelineRunnerMixin.run_takeoff_pipeline..extract_hsmt_text^s"$#*<<4F0G0G#H#H!&s2s8}}'='=!>!>KKA#x{';';F'C'Cd'JJGG (((&r$z"Failed to extract text from HSMT: cd}t}ttdt |D]#}|||ddzz }$||Sr2r6)r=docr? file_path_strr@s r"extract_pdf_textzIDrawingPipelineRunnerMixin.run_takeoff_pipeline..extract_pdf_textms}"$%ll3}+=+=>>!&s2s3xx'8'8!9!9FFA#s1vv'>'>'EEGG &r$z!Failed to extract text from PDF: text_layer image_base64 image_urldata:image/png;base64,cg|]I}|d|d|dp!|dr d|dnddJS) page_numbertitlerHrGrIr)rKrLrH)get).0ps r" zCDrawingPipelineRunnerMixin.run_takeoff_pipeline..s   $%]#3wZ!"{!3!3"5500 D>1BDDD     r$r)rGr rFr ext_pagesrDr%r'items drawing_typegeneral$apply_learned_rules_and_state_prices)rSr confidence_level ai_predictedgz$Could not delete old takeoff items: quantityunit_price_vnd unit_pricetotal_amount_vndtotalcategorycivil item_nameu bê tôngubt lótconcretem3unitu cốt théputhéprebarkgutấng@@u ván khuônformworkm2100m2duđàouđắpuvận chuyểnubùn earthworkwbs_codezWBS-02d norm_codezTT.NORM.u Hạng mục dimension_formulauKhối lượng bóc tách:  price_source_urlz'/dashboard/material-price-comparison?q=+)rr item_index item_orderrkrmr_r]rnrbrXrYr[rp"create_drawing_takeoff_items_batch hsmt_verifiedc\u*AI Quỳnh QS đã hoàn tất bóc tách uO hạng mục thi công thành công. Tổng chi phí dự toán ước tính: z,.0fu VNĐ.confidence_scoreai_report_messagehsmt_reference_file completeduFĐã hoàn tất phân tích bóc tách khối lượng tự động (u hạng mục).) rpreview_image_url pages_dataryrVr{rzr.total_estimated_cost_vndtotal_concrete_volume_m3total_rebar_weight_tonstotal_formwork_area_m2total_earthwork_volume_m3u8[TAKEOFF] Hoàn tất bóc tách siêu tốc takeoff_id=z: u hạng mục, tổng chi phí=u VNĐsuccessu3Hoàn tất bóc tách khối lượng thành công)r*r countrmessageu&[TAKEOFF] Lỗi bóc tách takeoff_id=T)exc_infozError: )&rrrget_drawing_takeoffupdate_drawing_takeoffrMospathexistsrsuffixlowerasyncio to_threadextract_dwg_text_and_previewrender_dxf_to_imagesrender_pdf_to_imagesr@ Exceptionloggerwarninganalyze_drawing_with_visionhasattrrUdelete_drawing_takeoff_items enumerater floatroundappendreplacerucreate_drawing_takeoff_itemr: isinstancerinfor))*rr rr%r'rtakeoffextrQ first_pagerFrAerEpreview_base64 first_img_urlpages_metadata vision_result raw_itemsrSrR source_type total_rebartotal_formworktotal_earthworktotal_concrete total_costitems_to_batchr?itemitem_idqtyu_pricetot_vndcatnl it_payload conf_score report_msg hsmt_ref_filerDr@s* ` @@r"run_takeoff_pipelinez/DrawingPipelineRunnerMixin.run_takeoff_pipeline's  J 9&:::FFG J")5HIII   3 3-|<   $KK 44M  +}(E(E + 77*3-=!'**}%%,2244Cf}}")"35}## ")"3-}## ")"3-tM/B/BAsJ## #*"3-tM/B/BAsJ## *3:1JJ! >bgnn5G&H&H > M"NNN''''''(9:K(L(L"L"L"L"L"L"LLJJ MMMNN#K#K#KLLLLLLLLM L"NNN''''''(9:J(K(K"K"K"K"K"K"KKJJ LLLNN#Jq#J#JKKKKKKKKL(^^L"== '^^NB??N"+A,"2"2;"?"? !   ---BPW=^===UW   #   N#'"B"B+ ^R88%)#+#5'=#C # #      M&))'266I!!.11;;~..  tCDD "AALZB"'++,>OOKK N!O NJ K ==jIIII K K KIaIIJJJJJJJJ K N$U++C &C &4eegg,,DHHZ005A66HH-..M$((<2H2HMA HH/00,xx((,S7]++ hhz733hh{++1r88::"$$ R$C"#s488FB+?+?'@'@'F'F'H'HHHcNN#b((GrMM!C  3txx';';#<#<#B#B#D#DDD"c$((62*>*>&?&?&E&E&G&GGG '#dhhvr.B.B*C*C*I*I*K*KKK C!$v! KK#b(($C"  3txx';';#<#<#B#B#D#DDD '#dhhvr.B.B*C*C*I*I*K*KKK #II!$!NN]]h"nn8HB8N8NRX\^R^R^%C##s488FB+?+?'@'@'F'F'H'HHHcO%%%&0&'!e&'!e$(HHZ$8$8$NTQ>T>T>T%)XXk%:%:%U>UaRSe>U>U$'-1XX6I-J-J.WV#VVQS@T@TVV $ 4 4$'*1,3,0HH5G,H,H-|{dhh{F[F[Fa_acfdfcfEgEoEopsuxEyEy{{!(g% t')MNN M CCNSSSS"0MMJ$@@LLLL*o==2JTSZZTTi i str | Path | Nonedict[str, Any]c K||||||}|d} || ||rt|nd|d{V} |j| p|} |j| pg} t | dpdt | dpdt | dpdt | d pdt | d pdd } d | | | d S)ziUnified end-to-end entrypoint: creates takeoff record, runs pipeline, and returns full structured result.)r r r r rrN)r rr%r'rrrrrr)rrrrrr)r*rrRmetrics)r#rr rrlist_drawing_takeoff_itemsrrM)rr r r r rr%r'createdr pipeline_resultrrRrs r"process_drawing_filez/DrawingPipelineRunnerMixin.process_drawing_file<s--!#% .  T] $ 9 9!%:LVs#5666RV#9 !:! !       "66zBBMg ;;JGGM2). 677<1))). 677<1))(- 566;!((',GKK8P,Q,Q,VUV&W&W). 788=A**  "     r$)NNr) r rr r r r r r rr rr)rNN) r r rr r%r&r'r&rr)NNrNN)r rr r r r r r rr r%rr'r&rr)__name__ __module__ __qualname__r#rrr$r"rr s  :)--1 S9S9S9S9S9r04-12 2 2 2 2 2 2 r$r) __future__rrloggingrpathlibrtypingr getLoggerrrrrr$r"rs""""""   8 $ $b b b b b b b b b b r$