Update app.py

app.py

@@ -320,54 +320,28 @@ def find_float_after_label(text: str, label: str, max_numbers: int = 2, window:
 
 
 def extract_name_from_filename(filename: str):
-    """
-    Ex:
-    19850515_ERIC_TEVANE_124522_Analyse de la marche FDM-T_Report (1).pdf
-    -> ERIC TEVANE
-    """
     if not filename:
         return None
 
     base = filename.rsplit("/", 1)[-1]
     base = base.rsplit(".", 1)[0]
 
-    parts = base.split("_")
-    name_parts = []
-
-    # on saute le premier token s'il est purement numérique (date)
-    started = False
-    for p in parts:
-        p_clean = p.strip()
-        if not p_clean:
-            continue
-
-        if not started:
-            if p_clean.isdigit():
-                continue
-            if re.fullmatch(r"[A-ZÀ-Ÿ\-]+", p_clean):
-                started = True
-                name_parts.append(p_clean)
-            continue
-
-        # quand on a commencé à lire le nom, on continue tant que c'est alpha MAJ
-        if re.fullmatch(r"[A-ZÀ-Ÿ\-]+", p_clean):
-            name_parts.append(p_clean)
-        else:
-            break
-
-    if name_parts:
-        return " ".join(name_parts)
+    # ex: 19850515_ERIC_TEVANE_124522_Analyse...
+    m = re.search(r"\d{8}_([A-Z]+)_([A-Z]+)_", base.upper())
+    if m:
+        first_name = m.group(1).strip()
+        last_name = m.group(2).strip()
+        return f"{first_name} {last_name}"
 
     return None
 
 
 def extract_pdf_name(text: str, source_pdf: str = None):
-    # 1) fallback robuste via nom de fichier
+    # priorité au nom du fichier
     name_from_file = extract_name_from_filename(source_pdf) if source_pdf else None
     if name_from_file:
         return name_from_file
 
-    # 2) fallback texte si jamais le PDF est mieux extrait
     if not text:
         return None
 
@@ -381,7 +355,7 @@ def extract_pdf_name(text: str, source_pdf: str = None):
         if m:
             name = " ".join(m.group(1).split()).strip()
             if len(name) >= 4:
-                return name
+                return name.upper()
 
     return None
 
@@ -394,7 +368,7 @@ def parse_zebris_pdf(uploaded_pdf):
     source_pdf = uploaded_pdf.name
 
     data = {
-        "athlete_name": None,
+        "athlete_name": extract_name_from_filename(source_pdf),
         "source_pdf": source_pdf,
         "transition_g": np.nan,
         "transition_d": np.nan,
@@ -419,32 +393,34 @@ def parse_zebris_pdf(uploaded_pdf):
         "attaque_pdf": "indéterminée",
     }
 
-    # Nom athlète : priorité au nom de fichier, car le texte extrait est parfois corrompu
-    data["athlete_name"] = extract_pdf_name("", source_pdf=source_pdf)
-
     def to_float(x):
-        return float(x.replace(",", "."))
+        return float(x.replace(",", ".").replace(" ", ""))
 
     with pdfplumber.open(io.BytesIO(raw)) as pdf:
         page_texts = []
         for page in pdf.pages:
             txt = page.extract_text() or ""
             txt = txt.replace("\xa0", " ")
+            txt = re.sub(r"\s+", " ", txt).strip()
             page_texts.append(txt)
 
-    full_text = "\n".join(page_texts)
+    full_text = " ".join(page_texts)
 
-    # fallback si le nom de fichier n'a pas suffi
+    # fallback si nom non récupéré via fichier
     if not data["athlete_name"]:
         data["athlete_name"] = extract_pdf_name(full_text, source_pdf=source_pdf)
 
-    # ---------------------------------------------------
-    # PAGE 8 : "Analyse du pieds en trois zones"
-    # On parse les paires Gauche / Droite dans l'ordre.
-    # ---------------------------------------------------
+    # --------------------------------------------------
+    # Chercher la page utile Zebris
+    # --------------------------------------------------
     zone_page_text = None
     for txt in page_texts:
-        if "Analyse du pieds en trois zones" in txt:
+        txt_lower = txt.lower()
+        if (
+            "force maximale" in txt_lower
+            and "pression maximale" in txt_lower
+            and "instant pic de force" in txt_lower
+        ):
             zone_page_text = txt
             break
 
@@ -452,22 +428,34 @@ def parse_zebris_pdf(uploaded_pdf):
         data["attaque_pdf"] = estimate_attack_from_pdf(data)
         return data
 
-    # normalisation
     zone_page_text = zone_page_text.replace("\xa0", " ")
     zone_page_text = re.sub(r"\s+", " ", zone_page_text).strip()
 
-    # Dans TON extraction pdfplumber, "Droite" est souvent corrompu en "oeitDr"
-    # et l'écart-type sort avec un tiret/en dash, ex: 1066,5–67,8
-    # On récupère seulement la valeur moyenne gauche puis droite.
+    # --------------------------------------------------
+    # Regex paires Gauche / Droite
+    # Supporte :
+    # Gauche 1066,5±67,8 Droite 1040,5±49,3
+    # ou variantes OCR/pdfplumber dégradées
+    # --------------------------------------------------
     pair_pattern = re.compile(
-        r"Gauche\s+(\d+,\d+|\d+\.\d+|\d+)\s*[–-]\s*(\d+,\d+|\d+\.\d+|\d+).*?"
-        r"(?:Droite|oeitDr)\s+(\d+,\d+|\d+\.\d+|\d+)\s*[–-]\s*(\d+,\d+|\d+\.\d+|\d+)",
+        r"Gauche\s+(\d+,\d+|\d+\.\d+|\d+)\s*(?:±|[–-])?\s*(\d+,\d+|\d+\.\d+|\d+)?"
+        r".{0,60}?"
+        r"(?:Droite|Droit|oeitDr)\s+(\d+,\d+|\d+\.\d+|\d+)\s*(?:±|[–-])?\s*(\d+,\d+|\d+\.\d+|\d+)?",
         flags=re.S
     )
 
     matches = pair_pattern.findall(zone_page_text)
 
-    # Ordre attendu sur la page 8 Zebris :
+    values = []
+    for m in matches:
+        try:
+            g_mean = to_float(m[0])
+            d_mean = to_float(m[2])
+            values.append((g_mean, d_mean))
+        except Exception:
+            pass
+
+    # Ordre attendu Zebris :
     # 0 transition s
     # 1 transition %
     # 2 fore force
@@ -479,18 +467,8 @@ def parse_zebris_pdf(uploaded_pdf):
     # 8 fore peak time
     # 9 mid peak time
     # 10 heel peak time
-    #
-    # Sur ton PDF, cet ordre est bien visible dans la page "Analyse du pieds en trois zones". :contentReference[oaicite:1]{index=1}
-
-    values = []
-    for m in matches:
-        g_mean = to_float(m[0])
-        d_mean = to_float(m[2])
-        values.append((g_mean, d_mean))
-
     if len(values) >= 11:
         data["transition_g"], data["transition_d"] = values[0]
-        # values[1] = transition en %, non utilisé ici
 
         data["fore_force_g"], data["fore_force_d"] = values[2]
         data["mid_force_g"], data["mid_force_d"] = values[3]
@@ -1291,6 +1269,7 @@ if uploaded_pdfs:
     for pdf in uploaded_pdfs:
         try:
             pdfs_data.append(parse_zebris_pdf(pdf))
+            st.write("PDF parsé :", pdfs_data[-1])
         except Exception as e:
             st.warning(f"{pdf.name} : erreur lecture PDF ({e})")