cgdshgdv

controllers/main.py

@@ -1,6 +1,5 @@
 import os
 import sys
-import threading # <-- Đã thêm thư viện threading
 from pathlib import Path
 
 # ─── 1. CẤU HÌNH ĐƯỜNG DẪN TUYỆT ĐỐI (TRÁNH LẠC ĐƯỜNG) ──────────────────────
@@ -49,52 +48,34 @@ logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
 
-# ─── KHỞI TẠO AI CHẠY NGẦM ───────────────────────────────────────────────────
-ocr_predictor = None
-read_info = None
-is_ai_ready = False
+# ─── KHỞI TẠO AI: YOLOv7 + VietOCR ────────────────────────────────────────────
+logger.info("[Startup] Nạp mô hình VietOCR (VGG-seq2seq)...")
+vocr_config_path = os.path.join(current_dir, 'Vocr', 'config', 'vgg-seq2seq.yml')
+config_vietocr = Cfg_vietocr.load_config_from_file(vocr_config_path)
 
-def load_ai_background():
-    global ocr_predictor, read_info, is_ai_ready
-    try:
-        logger.info("[AI_LOADER] Bắt đầu nạp mô hình AI chạy ngầm (Chống sập Render)...")
-        
-        # Nạp VietOCR
-        vocr_config_path = os.path.join(current_dir, 'Vocr', 'config', 'vgg-seq2seq.yml')
-        config_vietocr = Cfg_vietocr.load_config_from_file(vocr_config_path)
-        config_vietocr['weights'] = os.path.join(current_dir, 'Models', 'seq2seqocr.pth')
-        config_vietocr['device']  = 'cpu'
-        ocr_predictor = Predictor(config_vietocr)
-
-        # Nạp YOLOv7
-        get_dictionary = Detect(opt)
-        scan_weight = os.path.join(current_dir, 'Models', 'cccdYoloV7.pt')
-        imgsz, stride, device, half, model, names = get_dictionary.load_model(scan_weight)
-        
-        read_info = ReadInfo(imgsz, stride, device, half, model, names, ocr_predictor)
-        
-        is_ai_ready = True
-        logger.info("[AI_LOADER] HOÀN TẤT! Hệ thống YOLO + VietOCR đã sẵn sàng nhận diện.")
-    except Exception as e:
-        logger.error(f"[AI_LOADER] Lỗi khi nạp AI: {e}")
+config_vietocr['weights'] = os.path.join(current_dir, 'Models', 'seq2seqocr.pth')
+config_vietocr['device']  = 'cpu'  # Đổi thành 'cuda:0' nếu máy có Card rời NVIDIA
+ocr_predictor = Predictor(config_vietocr)
+
+logger.info("[Startup] Nạp mô hình YOLOv7 (Phát hiện vùng thông tin)...")
+get_dictionary = Detect(opt)
+scan_weight = os.path.join(current_dir, 'Models', 'cccdYoloV7.pt')
+imgsz, stride, device, half, model, names = get_dictionary.load_model(scan_weight)
+
+read_info = ReadInfo(imgsz, stride, device, half, model, names, ocr_predictor)
+logger.info("[Startup] Hệ thống YOLO + VietOCR đã sẵn sàng!")
 
 
-# ─── Startup (Vượt qua vòng kiểm duyệt của Render) ────────────────────────────
+# ─── Startup ──────────────────────────────────────────────────────────────────
 @asynccontextmanager
 async def lifespan(app: FastAPI):
     logger.info("[Startup] Khởi tạo cấu trúc Database (nếu chưa có)...")
     init_database()
-    
     logger.info("[Startup] Nạp embedding vào RAM...")
     _load_embeddings_to_ram()
     logger.info(f"[Startup] {face_memory_store.count} khuôn mặt trên RAM")
-
-    # BẬT LUỒNG CHẠY NGẦM: Server mở cổng ngay lập tức, AI từ từ nạp
-    threading.Thread(target=load_ai_background, daemon=True).start()
-    
     yield
     logger.info("[Shutdown] Bye!")
-
 def _load_embeddings_to_ram():
     conn = None
     cursor = None
@@ -122,11 +103,10 @@ def _load_embeddings_to_ram():
                 })
             except Exception as e:
                 logger.warning(f"[Startup] Bỏ qua khuôn mặt lỗi: {e}")
-                
-        face_memory_store.load_all(parsed)
+                face_memory_store.load_all(parsed)
         
     except Exception as e:
-        logger.error(f"[Startup] Lỗi kết nối DB khi nạp dữ liệu: {e}")
+        logger.error(f"[Startup]  Lỗi kết nối DB khi nạp dữ liệu: {e}")
         face_memory_store.load_all([]) 
         
     finally:
@@ -147,11 +127,13 @@ app.add_middleware(
 
 app.mount("/uploads", StaticFiles(directory=UPLOAD_DIR), name="uploads")
 
+
 class PersonUpdate(BaseModel):
     name: str
     role: str
     department: str
 
+
 def save_log_to_db(log_queries: list) -> None:
     if not log_queries:
         return
@@ -171,10 +153,6 @@ def save_log_to_db(log_queries: list) -> None:
 
 @app.post("/api/face/ocr")
 async def extract_ocr_local(file: UploadFile = File(...), side: str = Form(...)):
-    # BẢO VỆ: Chặn request nếu AI chưa nạp xong
-    if not is_ai_ready:
-        return {"success": False, "message": "Hệ thống AI đang khởi động, vui lòng thử lại sau 1-2 phút!"}
-
     temp_path = ""
     try:
         temp_filename = f"temp_cccd_{uuid.uuid4().hex}.jpg"
@@ -199,7 +177,7 @@ async def extract_ocr_local(file: UploadFile = File(...), side: str = Form(...))
                 "expiry_date":      raw.get("date_of_expiry", ""),
             }
 
-        else:  # back
+        else:  # back — ĐÃ NÂNG CẤP: dùng get_back_info như mặt trước
             raw = read_info.get_back_info(temp_path)
             logger.info(f"[OCR] Mặt sau raw: {raw}")
             mapped_data = {
@@ -218,7 +196,7 @@ async def extract_ocr_local(file: UploadFile = File(...), side: str = Form(...))
         logger.error(f"[OCR] Lỗi: {e}", exc_info=True)
         if os.path.exists(temp_path):
             os.remove(temp_path)
-        return {"success": False, "message": str(e), "data": {}}
+        return {"success": True, "data": {}}
 
 
 # ═════════════════════════════════════════════════════════════════════════════
@@ -307,7 +285,6 @@ async def register(
     new_encodings: list[tuple] = []
     avatar_path = ""
     saved_files = [] # Lưu danh sách file đã tạo để xóa nếu có lỗi
-    COSINE_THRESHOLD = 0.5 # Bổ sung biến ngưỡng vì thấy bạn nhắc tới trong code
 
     try:
         cccd       = json.loads(cccd_info) if cccd_info else {}
@@ -345,8 +322,8 @@ async def register(
                 avatar_path = saved_path
                 cursor.execute(
                     """INSERT INTO persons 
-                          (id, name, role, department, status, img_path, work_expiry_date)
-                        VALUES (%s, %s, %s, %s, 'active', %s, %s)""",
+                         (id, name, role, department, status, img_path, work_expiry_date)
+                       VALUES (%s, %s, %s, %s, 'active', %s, %s)""",
                     (person_id, name, role, department, avatar_path, expiry_val),
                 )
 
@@ -431,7 +408,6 @@ async def register(
     finally:
         cursor.close()
         conn.close()
-
 # ═════════════════════════════════════════════════════════════════════════════
 # DANH SÁCH NGƯỜI DÙNG
 # ═════════════════════════════════════════════════════════════════════════════
@@ -467,7 +443,7 @@ async def get_persons():
 
 
 # ═════════════════════════════════════════════════════════════════════════════
-# CẬP NHẬT & XÓA & LOGS & THỐNG KÊ
+# CẬP NHẬT & XÓA & LOGS & THỐNG KÊ (Giữ nguyên)
 # ═════════════════════════════════════════════════════════════════════════════
 @app.put("/api/face/persons/{id}")
 async def update_person(id: str, person_data: PersonUpdate):