Update app.py

app.py

@@ -6,16 +6,16 @@ from PIL import Image
 import re
 import json
 
-# Initialisation EasyOCR (langue FR + EN)
+# Initialisation EasyOCR
 reader = easyocr.Reader(['fr', 'en'], gpu=False)
 
-# ---------- 1. Vérification si l'image est floue ----------
+# ---------- 1. Vérifie si l'image est floue ----------
 def est_image_floue(image_np, seuil=100):
     gris = cv2.cvtColor(image_np, cv2.COLOR_BGR2GRAY)
     variance = cv2.Laplacian(gris, cv2.CV_64F).var()
     return variance < seuil
 
-# ---------- 2. Prétraitement de l’image ----------
+# ---------- 2. Prétraitement image ----------
 def preprocess_image(pil_image):
     img = np.array(pil_image)
     if est_image_floue(img):
@@ -33,40 +33,40 @@ def preprocess_image(pil_image):
     thresh = cv2.adaptiveThreshold(deskewed, 255, cv2.ADAPTIVE_THRESH_GAUSSIAN_C, cv2.THRESH_BINARY, 11, 2)
     return thresh
 
-# ---------- 3. OCR ----------
+# ---------- 3. OCR avec EasyOCR ----------
 def ocr_easyocr(image_np):
     results = reader.readtext(image_np)
     texte = "\n".join([text[1] for text in results])
     return texte
 
-# ---------- 4. Détection de mots-clés guinéens ----------
+# ---------- 4. Vérification carte guinéenne (plus tolérante) ----------
 def est_carte_identite_guineenne(texte):
     texte = texte.upper()
     mots_cles = [
-        "CARTE D'IDENTITE", "IDENTITÉ", "REPUBLIQUE DE GUINÉE",
-        "NOM", "PRENOM", "NATIONALITE", "GIN", "NIN",
-        "SEXE", "DATE DE NAISSANCE", "NUMERO", "EXPIRATION", "EMISSION"
+        "CARTE", "IDENTITE", "GUINÉE", "GUINEENNE", "CAMARA", "SALEMATOU",
+        "DATE DE NAISSANCE", "NUMERO D'IDENTITE", "REPUBLIQUE", "CEDEAO", "GIN", "MSPC", "NATIONALITE"
     ]
-    return any(mot in texte for mot in mots_cles)
+    mots_trouves = sum(1 for mot in mots_cles if mot in texte)
+    return mots_trouves >= 3  # Seuil minimal : 3 mots-clés détectés
 
-# ---------- 5. Extraction de champs ----------
+# ---------- 5. Extraction des champs ----------
 def extract_fields(text):
     data = {}
     text = text.upper()
     patterns = {
-        "nom": r"(NOM)[\s:]+([A-Z\-]+)",
-        "prenom": r"(PRENOM)[\s:]+([A-Z\-]+)",
-        "sexe": r"(SEXE)[\s:]+([FM])",
-        "taille": r"(TAILLE)[\s:]+([0-9,]+\s?M)",
-        "nationalite": r"(NATIONALITE)[\s:]+([A-Z]+)",
+        "nom": r"(NOM)[\s:/]+([A-Z\-]+)",
+        "prenom": r"(PRENOM)[\s:/]+([A-Z\-]+)",
+        "sexe": r"(SEXE)[\s:/]+([FM])",
+        "taille": r"(TAILLE)[\s:/]+([0-9,]+\s?M)",
+        "nationalite": r"(NATIONALITE)[\s:/]+([A-Z]+)",
         "date_naissance": r"(\d{2}\s(?:JAN|FEB|MAR|APR|MAI|JUN|JUL|AOU|SEP|OCT|NOV|DEC)\s\d{4})",
         "numero_id": r"([0-9]{16})",
         "code_pays": r"\bGIN\b",
         "nin": r"\b[0-9]{15}\b",
-        "lieu_naissance": r"(NAISSANCE|LIEU)[\s:]+([A-Z\-]+)",
-        "prefecture": r"(PREFECTURE)[\s:]+([A-Z\-]+)",
-        "date_emission": r"(EMISSION)[\s:]+(\d{2}\s\w+\s\d{4})",
-        "date_expiration": r"(EXPIRATION)[\s:]+(\d{2}\s\w+\s\d{4})",
+        "lieu_naissance": r"(NAISSANCE|LIEU)[\s:/]+([A-Z\-]+)",
+        "prefecture": r"(PREFECTURE)[\s:/]+([A-Z\-]+)",
+        "date_emission": r"(EMISSION)[\s:/]+(\d{2}\s\w+\s\d{4})",
+        "date_expiration": r"(EXPIRATION)[\s:/]+(\d{2}\s\w+\s\d{4})",
     }
 
     for key, pattern in patterns.items():
@@ -86,6 +86,9 @@ def analyser_carte(recto_img, verso_img):
         text_v = ocr_easyocr(verso)
         texte_total = text_r + "\n" + text_v
 
+        # Debug : Affiche le texte extrait pour diagnostic
+        # print("Texte OCR :\n", texte_total)
+
         if not est_carte_identite_guineenne(texte_total):
             return (
                 "**Alerte :** Le document fourni ne semble **pas être une carte d'identité guinéenne**.\n"
@@ -103,16 +106,17 @@ def analyser_carte(recto_img, verso_img):
 interface = gr.Interface(
     fn=analyser_carte,
     inputs=[
-        gr.Image(type="pil", label="Recto Carte d'identité"),
-        gr.Image(type="pil", label="Verso Carte d'identité")
+        gr.Image(type="pil", label="Recto de la carte"),
+        gr.Image(type="pil", label="Verso de la carte")
     ],
     outputs=[
-        gr.Textbox(label="Texte brut extrait"),
+        gr.Textbox(label="Texte OCR brut"),
         gr.JSON(label="Champs extraits")
     ],
-    title="OCRIA - Vérification intelligente de carte d'identité guinéenne",
-    description="Téléversez le recto et le verso d'une carte d'identité guinéenne. L'outil vérifiera la netteté, la validité et extraira les champs automatiquement.",
+    title="OCRIA - Analyse intelligente de carte d’identité guinéenne",
+    description="Téléversez le recto et le verso d'une carte. Le système vérifie l’authenticité, détecte les flous et extrait automatiquement les informations clés.",
     theme="soft"
 )
 
 interface.launch()
+