Update app.py

app.py

@@ -358,7 +358,31 @@ def extract_pdf_name(text: str, source_pdf: str = None):
                 return name.upper()
 
     return None
+def extract_speed_from_pdf_text_or_filename(full_text: str, source_pdf: str):
+    # 1) Cherche dans le texte PDF
+    patterns = [
+        r"VMA\s*(\d+(?:[.,]\d+)?)\s*kmh",
+        r"VMA\s*(\d+(?:[.,]\d+)?)\s*km/h",
+        r"Vitesse,\s*km/h\s*(\d+(?:[.,]\d+)?)",
+        r"(\d+(?:[.,]\d+)?)\s*kmh",
+        r"(\d+(?:[.,]\d+)?)\s*km/h",
+    ]
+
+    if full_text:
+        for pattern in patterns:
+            m = re.search(pattern, full_text, flags=re.I)
+            if m:
+                return float(m.group(1).replace(",", "."))
+
+    # 2) Fallback : cherche dans le nom du fichier
+    if source_pdf:
+        m = re.search(r"(\d+(?:[.,]\d+)?)\s*kmh", source_pdf, flags=re.I)
+        if not m:
+            m = re.search(r"(\d+(?:[.,]\d+)?)\s*km/h", source_pdf, flags=re.I)
+        if m:
+            return float(m.group(1).replace(",", "."))
 
+    return np.nan
 
 def parse_zebris_pdf(uploaded_pdf):
     uploaded_pdf.seek(0)
@@ -408,12 +432,7 @@ def parse_zebris_pdf(uploaded_pdf):
     full_text = " ".join(page_texts)
 
     # Extraction de l'allure du PDF (ex: "VMA 14kmh" ou "14,0 km/h")
-    speed_match = re.search(r"(\d+(?:[.,]\d+)?)\s*km\s*/?\s*h", full_text, flags=re.I)
-    if not speed_match:
-        speed_match = re.search(r"(\d+(?:[.,]\d+)?)\s*kmh", full_text, flags=re.I)
-
-    if speed_match:
-        data["speed_kmh"] = float(speed_match.group(1).replace(",", "."))
+    data["speed_kmh"] = extract_speed_from_pdf_text_or_filename(full_text, source_pdf)
 
     # fallback si nom non récupéré via fichier
     if not data["athlete_name"]:
@@ -1536,8 +1555,7 @@ with tab_pdf:
                 ],
                 "Valeur": [
                     matched_pdf["source_pdf"],
-                    f"{matched_pdf.get('speed_kmh', 'N/A')} km/h" if matched_pdf.get("speed_kmh") else "N/A",
-                    matched_pdf["attaque_pdf"],
+                    f"{matched_pdf['speed_kmh']:.1f} km/h" if pd.notna(matched_pdf.get("speed_kmh")) else "N/A",
                     f"{matched_pdf['transition_g']:.3f} s" if pd.notna(matched_pdf["transition_g"]) else "N/A",
                     f"{matched_pdf['transition_d']:.3f} s" if pd.notna(matched_pdf["transition_d"]) else "N/A",
                     f"{matched_pdf['heel_force_g']:.1f} N" if pd.notna(matched_pdf["heel_force_g"]) else "N/A",