fix bug

pipeline_main.py

@@ -148,9 +148,32 @@ class TechnicalDrawingPipeline:
                     if conf_list:
                         ocr_confidence = round(sum(conf_list) / len(conf_list), 3)
                 except Exception as e:
-                    print(f" -> [CẢNH BÁO] Lỗi khi OCR Note đối tượng {i+1}: {e}. Đã bỏ qua để tránh crash.")
-                    ocr_content = "[OCR Error]"
-                    ocr_confidence = None
+                    print(f" -> [CẢNH BÁO] Lỗi khi OCR Note đối tượng {i+1} ({e}). Đang khởi tạo lại Backend...")
+                    try:
+                        del self.text_ocr
+                        import gc
+                        gc.collect()
+                        self.text_ocr = PaddleOCR(use_angle_cls=True, lang='en', use_gpu=False, ocr_version='PP-OCRv4', show_log=False, use_mkldnn=False, enable_mkldnn=False, cpu_threads=1, ir_optim=False, use_tensorrt=False)
+                        
+                        ocr_crop = np.copy(crop_img)
+                        ocr_result = self.text_ocr.ocr(ocr_crop, cls=True)
+                        lines = []
+                        conf_list = []
+                        if ocr_result and len(ocr_result) > 0 and ocr_result[0] is not None:
+                            for box_info in ocr_result[0]:
+                                text_val = box_info[1][0]
+                                conf_val = float(box_info[1][1])
+                                if conf_val > 0.6:
+                                    lines.append(text_val)
+                                    conf_list.append(conf_val)
+                        ocr_content = "\n".join(lines)
+                        if conf_list:
+                            ocr_confidence = round(sum(conf_list) / len(conf_list), 3)
+                        print(f" -> Đã tự phục hồi và OCR Note thành công đối tượng {i+1}!")
+                    except Exception as reset_e:
+                        print(f" -> [THẤT BẠI] Đã thử lại nhưng vẫn lỗi Note đối tượng {i+1}: {reset_e}")
+                        ocr_content = "[OCR Error]"
+                        ocr_confidence = None
                 
             elif class_name == "Table":
                 # => Nhận dạng Bảng & Xuất HTML
@@ -164,9 +187,26 @@ class TechnicalDrawingPipeline:
                     ocr_content = "\n".join(html_strings)
                     ocr_confidence = None # Tạm để Null cho Table
                 except Exception as e:
-                    print(f" -> [CẢNH BÁO] Lỗi khi xử lý cấu trúc Table đối tượng {i+1}: {e}. Đã bỏ qua để tránh crash.")
-                    ocr_content = "[Table Extractor Error]"
-                    ocr_confidence = None
+                    print(f" -> [CẢNH BÁO] Lỗi cấu trúc Table đối tượng {i+1} ({e}). Đang khởi tạo lại Backend...")
+                    try:
+                        del self.table_ocr
+                        import gc
+                        gc.collect()
+                        self.table_ocr = PPStructure(layout=False, table=True, ocr=True, show_log=False, recovery=True, lang='en', use_gpu=False, use_mkldnn=False, enable_mkldnn=False, cpu_threads=1, ir_optim=False, use_tensorrt=False)
+                        
+                        tab_crop = np.copy(crop_img)
+                        tab_results = self.table_ocr(tab_crop)
+                        html_strings = []
+                        for region in tab_results:
+                            if region.get('type') == 'table' and 'res' in region:
+                                html_strings.append(region['res'].get('html', ''))
+                        ocr_content = "\n".join(html_strings)
+                        ocr_confidence = None
+                        print(f" -> Đã tự phục hồi và xử lý Table thành công đối tượng {i+1}!")
+                    except Exception as reset_e:
+                        print(f" -> [THẤT BẠI] Đã thử lại nhưng vẫn lỗi Table đối tượng {i+1}: {reset_e}")
+                        ocr_content = "[Table Extractor Error]"
+                        ocr_confidence = None
                 
             elif class_name == "PartDrawing":
                 # Không ocr part drawing