Update app.py

app.py

@@ -6,11 +6,20 @@ from PIL import Image
 import re
 import json
 
-# Initialisation d'EasyOCR
-reader = easyocr.Reader(['fr', 'en'])
+# Initialisation EasyOCR (langue FR + EN)
+reader = easyocr.Reader(['fr', 'en'], gpu=False)
 
+# ---------- 1. Vérification 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 ----------
 def preprocess_image(pil_image):
     img = np.array(pil_image)
+    if est_image_floue(img):
+        raise ValueError("L'image semble floue. Veuillez fournir une image plus nette.")
     img = cv2.cvtColor(img, cv2.COLOR_RGB2BGR)
     gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
     gray = cv2.equalizeHist(gray)
@@ -24,11 +33,23 @@ def preprocess_image(pil_image):
     thresh = cv2.adaptiveThreshold(deskewed, 255, cv2.ADAPTIVE_THRESH_GAUSSIAN_C, cv2.THRESH_BINARY, 11, 2)
     return thresh
 
+# ---------- 3. OCR ----------
 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 ----------
+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"
+    ]
+    return any(mot in texte for mot in mots_cles)
+
+# ---------- 5. Extraction de champs ----------
 def extract_fields(text):
     data = {}
     text = text.upper()
@@ -55,18 +76,30 @@ def extract_fields(text):
 
     return data
 
+# ---------- 6. Fonction principale ----------
 def analyser_carte(recto_img, verso_img):
     try:
         recto = preprocess_image(recto_img)
         verso = preprocess_image(verso_img)
+
         text_r = ocr_easyocr(recto)
         text_v = ocr_easyocr(verso)
         texte_total = text_r + "\n" + text_v
+
+        if not est_carte_identite_guineenne(texte_total):
+            return (
+                "**Alerte :** Le document fourni ne semble **pas être une carte d'identité guinéenne**.\n"
+                "Merci de vérifier l'image ou d'en fournir une autre.",
+                {}
+            )
+
         champs = extract_fields(texte_total)
         return texte_total, champs
+
     except Exception as e:
         return f"Erreur : {str(e)}", {}
 
+# ---------- 7. Interface Gradio ----------
 interface = gr.Interface(
     fn=analyser_carte,
     inputs=[
@@ -77,8 +110,9 @@ interface = gr.Interface(
         gr.Textbox(label="Texte brut extrait"),
         gr.JSON(label="Champs extraits")
     ],
-    title="VisionIdentIA - Lecture intelligente de carte d'identité guinéenne",
-    description="Téléversez le recto et le verso d'une carte d'identité pour une extraction automatique des informations par IA.",
+    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.",
+    theme="soft"
 )
 
 interface.launch()