@@ -877,8 +877,11 @@ def process_segments(
877877 for page_num in page_indices
878878 for kv in pagespecific_details ['key_value_details' ].get (str (page_num ), [])
879879 ) or float (bp_confidence ) < confidence_threshold
880- logger .info (f"HITL Status Low confidence { low_confidence } " )
880+ else :
881+ low_confidence = None
881882
883+ logger .info (f"low_confidence: { low_confidence } " )
884+
882885 item .update ({
883886 "page_array" : page_indices ,
884887 "hitl_triggered" : low_confidence ,
@@ -925,6 +928,10 @@ def process_segments(
925928 "hitl_corrected_result" : custom_decimal_output
926929 })
927930 else :
931+ if enable_hitl == 'true' :
932+ std_hitl = 'true'
933+ else :
934+ std_hitl = None
928935 # Process standard output if no custom output match
929936 std_bucket , std_key = parse_s3_path (segment ['standard_output_path' ])
930937 std_output = download_decimal (std_bucket , std_key )
@@ -934,7 +941,7 @@ def process_segments(
934941 page_array = list (range (start_page , end_page + 1 ))
935942 item .update ({
936943 "page_array" : page_array ,
937- "hitl_triggered" : enable_hitl ,
944+ "hitl_triggered" : std_hitl ,
938945 "extraction_bp_name" : "None" ,
939946 "extracted_result" : std_output
940947 })
@@ -944,12 +951,12 @@ def process_segments(
944951 record_number = record_number ,
945952 bp_match = segment .get ('custom_output_status' ),
946953 extraction_bp_name = "None" ,
947- hitl_triggered = enable_hitl ,
954+ hitl_triggered = std_hitl ,
948955 page_array = page_array ,
949956 review_portal_url = SAGEMAKER_A2I_REVIEW_PORTAL_URL
950957 )
951958
952- hitl_triggered = enable_hitl
959+ hitl_triggered = std_hitl
953960 if enable_hitl == 'true' :
954961 for page_number in range (start_page , end_page + 1 ):
955962 ImageUri = f"s3://{ output_bucket } /{ object_key } /pages/{ page_number } /image.jpg"
0 commit comments