Probleme de Localisation et nouvel attribut Role Media

Check_claim_gemini.py

@@ -281,6 +281,18 @@ ZONE_FR_TO_PART = {
     "pare-choc arriere": "back-bumper",
     "pare choc arriere": "back-bumper",
 
+    # Pare-choc(s) SANS précision avant/arrière (ex: "le pare-chocs est
+    # abîmé") -> on ne devine PAS lequel des deux : on couvre les DEUX
+    # (avant ET arrière) plutôt que de tomber sur "zone non spécifiée"
+    # (qui élargirait la vérification à toute la voiture au lieu de la
+    # restreindre aux pare-chocs). Placées après les variantes avant/arrière
+    # ci-dessus : peu importe, le tri par longueur de clé (_ZONE_KEYS_SORTED)
+    # garantit que "pare-choc avant"/"pare-choc arriere" sont essayées avant.
+    "pare-choc": ["front-bumper", "back-bumper"],
+    "pare choc": ["front-bumper", "back-bumper"],
+    "pare-chocs": ["front-bumper", "back-bumper"],
+    "pare chocs": ["front-bumper", "back-bumper"],
+
     # Capot / coffre / toit
     "capot": "hood",
     "coffre": "trunk",
@@ -302,6 +314,13 @@ ZONE_FR_TO_PART = {
     "portiere arriere droite": "right_back-door",
     "porte arriere droite": "right_back-door",
 
+    # Porte(s)/portière(s) SANS précision de côté NI d'axe avant/arrière
+    # (ex: "une porte est enfoncée") -> comme pour le pare-choc, on ne
+    # devine pas laquelle : on couvre les 4 portes plutôt que d'élargir la
+    # vérification à toute la voiture.
+    "portiere": ["left_front-door", "right_front-door", "left_back-door", "right_back-door"],
+    "porte": ["left_front-door", "right_front-door", "left_back-door", "right_back-door"],
+
     # Rétroviseurs
     "retroviseur gauche": "left_mirror",
     "retro gauche": "left_mirror",
@@ -365,6 +384,47 @@ ZONE_FR_TO_PART = {
 # Tri par longueur de clé décroissante : on veut matcher "aile arriere droite" avant un éventuel motif plus générique qui s'y trouverait inclus.
 _ZONE_KEYS_SORTED = sorted(ZONE_FR_TO_PART.keys(), key=len, reverse=True)
 
+
+# ─── CORRESPONDANCE CODE PIÈCE (nomenclature CV) -> LIBELLÉ FRANÇAIS ─────────
+# Utilisé pour afficher un libellé lisible (colonne AppSheet "Localisation
+# Carrosserie") plutôt que le code technique interne (ex: "front-bumper").
+CODE_TO_FR_LABEL = {
+    "front-bumper": "Pare-choc avant",
+    "back-bumper": "Pare-choc arrière",
+    "hood": "Capot",
+    "trunk": "Coffre",
+    "roof": "Toit",
+    "license-plate": "Plaque d'immatriculation",
+    "grille": "Calandre",
+    "windshield": "Pare-brise avant",
+    "back-windshield": "Pare-brise arrière",
+    "left_front-door": "Portière avant gauche",
+    "right_front-door": "Portière avant droite",
+    "left_back-door": "Portière arrière gauche",
+    "right_back-door": "Portière arrière droite",
+    "left_mirror": "Rétroviseur gauche",
+    "right_mirror": "Rétroviseur droit",
+    "left_headlight": "Phare avant gauche",
+    "right_headlight": "Phare avant droit",
+    "left_tail-light": "Feu arrière gauche",
+    "right_tail-light": "Feu arrière droit",
+    "left_front-wheel": "Roue avant gauche",
+    "right_front-wheel": "Roue avant droite",
+    "left_back-wheel": "Roue arrière gauche",
+    "right_back-wheel": "Roue arrière droite",
+    "left_front-window": "Vitre avant gauche",
+    "right_front-window": "Vitre avant droite",
+    "left_back-window": "Vitre arrière gauche",
+    "right_back-window": "Vitre arrière droite",
+    "left_rocker-panel": "Bas de caisse gauche",
+    "right_rocker-panel": "Bas de caisse droit",
+    "left_fender": "Aile avant gauche",
+    "right_fender": "Aile avant droite",
+    "left_quarter-panel": "Aile arrière gauche",
+    "right_quarter-panel": "Aile arrière droite",
+}
+
+
 EXTRACTION_PROMPT_TEMPLATE = """Tu es un extracteur d'informations strict. On te donne une phrase décrivant une réclamation client sur un véhicule.
 
 Extrait UNIQUEMENT deux informations :
@@ -1283,6 +1343,74 @@ def save_claim_result(result, vehicle_id, output_dir):
     return out_path
 
 
+# ─── RÔLE DES MÉDIAS (table AppSheet "Medias Cameras", colonne "Role Media") ──
+# Valeurs possibles de l'enum "Role Media" :
+#   - "Dommages"         : le dommage réclamé est confirmé visible sur ce média.
+#   - "Parties Visibles" : ce média montre la pièce réclamée, mais sans dommage
+#                          confirmé dessus (utile pour visualiser l'état du
+#                          véhicule à l'entrée).
+#   - "Inutile"          : ce média ne montre pas la pièce réclamée.
+ROLE_DOMMAGES = "Dommages"
+ROLE_PARTIES_VISIBLES = "Parties Visibles"
+ROLE_INUTILE = "Inutile"
+
+
+def build_media_role_resolver(claim_result: dict):
+    """
+    À partir du résultat de process_claim()/process_multi_claim(), construit
+    une fonction `resolver(file_id, is_image) -> role` qui donne le rôle
+    ("Dommages" / "Parties Visibles" / "Inutile") d'un média d'entrée à
+    partir de son ID Drive (utilisé comme nom de fichier local par
+    drive_enrichment.py / drive_lookup.py, ex: "abcd1234.png") et de son
+    type (photo ou vidéo).
+
+    Règles :
+      - Photo : "Dommages" si un dommage y a ��té confirmé (damage_visible=True
+        sur au moins un "check" qui ne vient pas d'une frame vidéo) ; sinon
+        "Parties Visibles" si elle couvre la pièce réclamée (présente dans
+        candidate_images d'au moins un dommage) ; sinon "Inutile" (ne montre
+        pas la pièce réclamée par la réclamation).
+      - Vidéo : suit la même logique que leur upload dans Preuves / Parties
+        Visibles (finalize_claim.executer_bilan_final /
+        report_builder.construire_bilan_final) : "Dommages" si le verdict
+        global est OUI, "Parties Visibles" si la zone n'a pas pu être
+        précisée (mapped_part=None -> toutes les vidéos sont ajoutées pour
+        vérification à l'oeil nu), "Inutile" sinon.
+    """
+    sous_resultats = claim_result.get("damages") or [claim_result]
+
+    photos_avec_dommage = set()
+    photos_visibles = set()
+    for r in sous_resultats:
+        for check in r.get("checks", []):
+            if check.get("from_video"):
+                continue
+            file_id = Path(check["image_path"]).stem
+            photos_visibles.add(file_id)
+            if check.get("damage_visible") is True:
+                photos_avec_dommage.add(file_id)
+        for chemin_image in r.get("candidate_images", []):
+            photos_visibles.add(Path(chemin_image).stem)
+
+    if claim_result.get("verdict") == "OUI":
+        role_video = ROLE_DOMMAGES
+    elif claim_result.get("mapped_part") is None:
+        role_video = ROLE_PARTIES_VISIBLES
+    else:
+        role_video = ROLE_INUTILE
+
+    def resolver(file_id: str, is_image: bool) -> str:
+        if not is_image:
+            return role_video
+        if file_id in photos_avec_dommage:
+            return ROLE_DOMMAGES
+        if file_id in photos_visibles:
+            return ROLE_PARTIES_VISIBLES
+        return ROLE_INUTILE
+
+    return resolver
+
+
 # ─── TRAITEMENT EN LOT, MODÈLES CHARGÉS UNE SEULE FOIS ──────────
 
 def process_claims_batch(claims, output_dir):

appsheet_client.py

@@ -129,6 +129,13 @@ def ajouter_medias(rows: list):
     appsheet_action(TABLE_MEDIAS, "Add", rows)
 
 
+def edit_media(row_id: str, id_media: str, champs: dict):
+    """Met à jour une ligne existante de Medias Cameras (ex: colonne
+    "Role Media"). `champs` est un dict {nom_colonne_appsheet: valeur}."""
+    row = {"Row ID": row_id, "ID Media": id_media, **champs}
+    appsheet_action(TABLE_MEDIAS, "Edit", [row])
+
+
 def find_rows(table: str, selector: str = None) -> list:
     """Lit des lignes d'une table AppSheet (action Find), avec un filtre
     optionnel au format expression AppSheet, ex:

extract_claim_from_email.py

@@ -182,13 +182,14 @@ valide, sans aucun texte avant ou après, avec exactement ces clés :
   "vehicle_model": "Marque et modèle du véhicule si mentionné, sinon null",
   "claim_message": "Reformulation neutre, factuelle et concise du/des dommage(s), en français naturel, sans code technique. Si PLUSIEURS dommages distincts sont mentionnés (zones et/ou types différents), ils doivent TOUS être repris, sous forme d'une phrase par dommage séparée par un point-virgule (ex: 'Dégradation constatée sur le pare-choc avant.' ou 'Rayure sur le pare-choc avant ; enfoncement sur la portière arrière droite.'). Ne jamais en omettre un au profit d'un seul.",
   "claim_demand": "Ce que le client demande explicitement (ex: dédommagement, remboursement, informations sur les démarches, prise en charge), en une phrase courte. Si rien n'est explicitement demandé, mets null.",
-  "damaged_part": "La pièce endommagée. DOIT être EXACTEMENT l'une des valeurs suivantes : {allowed_parts}. Si aucune partie ne correspond, trouve celle qui se rapproche le plus. Si aucune partie n'est identifiable dans le mail, mets null."
+  "damaged_parts": "Un TABLEAU des pièces endommagées. Chaque élément DOIT être EXACTEMENT l'une des valeurs suivantes : {allowed_parts}. Si le mail précise suffisamment la zone (ex: 'pare-choc avant', 'portière arrière droite'), mets UNIQUEMENT la pièce exacte correspondante (tableau d'un seul élément). Si le mail mentionne une catégorie SANS préciser laquelle des pièces de cette catégorie est concernée (ex: 'pare-chocs' sans dire avant ou arrière, 'une portière' sans dire laquelle), mets TOUTES les pièces de cette catégorie dans le tableau plutôt que d'en deviner une seule au hasard (ex: pour 'pare-chocs' -> [\"front-bumper\", \"back-bumper\"] ; pour 'portière' -> [\"left_front-door\", \"right_front-door\", \"left_back-door\", \"right_back-door\"]). Si aucune partie n'est identifiable dans le mail, mets un tableau vide []."
 }}
 
 Règles :
 - Si une information est absente du mail, mets la valeur JSON `null` (pas une chaîne vide).
 - Ne devine jamais une plaque ou un nom qui ne serait pas explicitement présent dans le texte.
 - "claim_message" doit rester factuel et concis, utilisable directement dans un rapport.
+- Ne choisis JAMAIS arbitrairement une seule pièce quand le mail ne permet pas de trancher entre plusieurs pièces d'une même catégorie (avant/arrière, gauche/droite) : liste-les toutes dans "damaged_parts" (voir consigne ci-dessus).
 """
 
 
@@ -201,8 +202,8 @@ def extract_claim_info_from_email(email_text):
     """
     client = get_genai_client()
     # Codes pièce (nomenclature CV), dédoublonnés en gardant l'ordre —
-    # ce sont ces codes que le modèle doit renvoyer dans "damaged_part",
-    # ensuite traduits en libellé français via CODE_TO_FR_LABEL.
+    # ce sont ces codes que le modèle doit renvoyer dans "damaged_parts",
+    # ensuite traduits en libellés français via CODE_TO_FR_LABEL.
     allowed_parts = list(dict.fromkeys(ZONE_FR_TO_PART.values()))
 
     prompt = EXTRACTION_PROMPT.format(
@@ -232,10 +233,12 @@ def extract_claim_info_from_email(email_text):
         "vehicle_model",
         "claim_message",
         "claim_demand",
-        "damaged_part",
+        "damaged_parts",
     ]
     for key in expected_keys:
         data.setdefault(key, None)
+    if not isinstance(data.get("damaged_parts"), list):
+        data["damaged_parts"] = []
 
     # Sécurise le retraitement : si le client ne précise pas explicitement
     # sa demande, on pose une valeur par défaut plutôt que de laisser le
@@ -243,13 +246,19 @@ def extract_claim_info_from_email(email_text):
     if not data.get("claim_demand"):
         data["claim_demand"] = DEFAULT_CLAIM_DEMAND
 
-    # Traduction du code pièce (nomenclature CV) en libellé français lisible,
-    # pour la colonne AppSheet "Localisation Carrosserie". Si aucune pièce
-    # n'est identifiable dans le mail, on pose "Non Spécifié" plutôt que de
-    # laisser le champ vide.
-    data["localisation_carrosserie"] = (
-        CODE_TO_FR_LABEL.get(data["damaged_part"]) or DEFAULT_LOCALISATION_CARROSSERIE
-    )
+    # Traduction des codes pièce (nomenclature CV) en libellés français
+    # lisibles, pour la colonne AppSheet "Localisation Carrosserie". Une
+    # catégorie ambiguë (ex: "pare-chocs" sans avant/arrière, "portière" sans
+    # précision) donne PLUSIEURS codes dans "damaged_parts" (cf. prompt
+    # d'extraction) : on les liste TOUS plutôt que d'en choisir un seul. Si
+    # aucune pièce n'est identifiable dans le mail, on pose "Non Spécifié"
+    # plutôt que de laisser le champ vide.
+    labels = [
+        CODE_TO_FR_LABEL[code]
+        for code in dict.fromkeys(data["damaged_parts"])
+        if code in CODE_TO_FR_LABEL
+    ]
+    data["localisation_carrosserie"] = " / ".join(labels) if labels else DEFAULT_LOCALISATION_CARROSSERIE
 
     return data
 

finalize_claim.py

@@ -61,11 +61,11 @@ from pathlib import Path
 import gspread
 from googleapiclient.http import MediaFileUpload, MediaIoBaseDownload
 
-from appsheet_client import edit_reclamation, find_rows, TABLE_RECLAMATIONS, TABLE_MEDIAS
+from appsheet_client import edit_reclamation, edit_media, find_rows, TABLE_RECLAMATIONS, TABLE_MEDIAS
 from drive_enrichment import get_drive_service, trouver_dossier_plaque
 from oauth_drive import get_user_drive_service, get_user_credentials
 from Inspect_car_parts import run_parts_inspection
-from Check_claim_gemini import process_multi_claim
+from Check_claim_gemini import process_multi_claim, build_media_role_resolver
 
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger("finalize_claim")
@@ -250,6 +250,24 @@ def executer_bilan_final(id_reclamation: str):
         claim_result = process_multi_claim(claim_message, str(parts_json), videos_dir=videos_dir)
         verdict = claim_result.get("verdict")  # "OUI" / "NON" / None (cas d'erreur)
 
+        # 4bis. Rôle de chaque média (colonne "Role Media" de Medias Cameras) :
+        #   "Dommages" si le dommage réclamé y est confirmé, "Parties Visibles"
+        #   si le média montre la pièce réclamée sans dommage confirmé,
+        #   "Inutile" sinon (cf. Check_claim_gemini.build_media_role_resolver).
+        resolver_role_media = build_media_role_resolver(claim_result)
+        for m in medias:
+            file_id = _extraire_id_drive(m.get("Fichier Media", ""))
+            if not file_id:
+                continue
+            role = resolver_role_media(file_id, m.get("Type Media") == "Image")
+            try:
+                edit_media(m.get("Row ID"), m.get("ID Media"), {"Role Media": role})
+            except Exception:
+                logger.exception(
+                    "Échec de la mise à jour de 'Role Media' pour le média %s (réclamation %s)",
+                    file_id, id_reclamation,
+                )
+
         # 5. Dossiers Drive sous BDD Claim/Plaque/<plaque>/Bilan Final/ -------
         plaque = reclamation.get("Vehicule Plaque", "")
         dossier_plaque = trouver_dossier_plaque(service, plaque)