Update app.py

app.py

@@ -1,86 +1,126 @@
 import gradio as gr
-import numpy as np
-
-# Conversion pour Acontista mexicana (approximation par interpolation)
-acontista_data = {
-    1: 3,
-    2: 10,
-    3.5: 20,
-    4: 27,
-    6: 60,
-    10: 100
-}
-
-def interp_age_acontista(mois):
-    mois_list = sorted(acontista_data.keys())
-    ans_list = [acontista_data[m] for m in mois_list]
-    return np.interp(mois, mois_list, ans_list)
-
-# Conversion pour Hierodula spp.
-hierodula_stades = {
-    "L1": 0,
-    "L2": 0.5,
-    "L3": 1.25,
-    "L4": 2.125,
-    "L5": 3.125,
-    "L6": 4.625,
-    "L7": 6.825,
-    "L8": 9.325,  # Adulte
-}
-
-def hierodula_age_mois(stade, sexe):
-    base = hierodula_stades.get(stade, 0)
-    if stade == "L8":  # adulte
-        if sexe == "Mâle":
-            return base + 4.5  # vie adulte plus courte
-        else:
-            return base + 7  # femelle plus longue
-    return base
-
-def convert_age(age_input, unite, espece, sexe):
-    # Conversion selon l'espèce
-    if espece == "Acontista mexicana":
-        mois = float(age_input)
-        age_humain = interp_age_acontista(mois)
-    elif espece == "Hierodula spp.":
+
+# === Conversion pour Acontista mexicana ===
+def age_humain_acontista(mois: float) -> float:
+    """Conversion Acontista mexicana."""
+    if mois <= 1:
+        return 3 * mois
+    elif mois <= 2:
+        return 3 + (mois - 1) * 7
+    elif mois <= 3.5:
+        return 10 + (mois - 2) * 6.67
+    elif mois <= 4:
+        return 20 + (mois - 3.5) * 14
+    elif mois <= 6:
+        return 27 + (mois - 4) * 16.5
+    elif mois <= 10:
+        return 60 + (mois - 6) * 10
+    else:
+        return 100
+
+
+# === Conversion pour Hierodula spp. ===
+def age_humain_hierodula(stade: str = None, mois: float = None, sexe: str = "Femelle") -> float:
+    """Conversion Hierodula spp. selon les données de Ruben."""
+    stades_mois = {
+        "L1": 0,
+        "L2": 0.5,
+        "L3": 1.25,
+        "L4": 2.125,
+        "L5": 3.125,
+        "L6": 4.625,
+        "L7": 6.825,
+        "L8": 9.325,
+    }
+
+    # Si on a donné un stade
+    if stade:
+        mois = stades_mois.get(stade, 0)
+
+    # Durée adulte selon le sexe
+    if sexe == "Mâle":
+        max_vie = 17      # ~L8 + 7.5 mois
+    elif sexe == "Femelle":
+        max_vie = 19.5    # ~L8 + 10 mois
+    else:
+        max_vie = 18.5
+
+    # Conversion en âge humain
+    if mois <= 9.325:
+        age_humain = (mois / 9.325) * 27
+    else:
+        age_humain = 27 + ((mois - 9.325) / (max_vie - 9.325)) * (100 - 27)
+
+    return min(age_humain, 100)
+
+
+# === Gestion générale ===
+def convertir_age(age_input, unite, espece, sexe):
+    espece = espece.lower()
+
+    if espece.startswith("acontista"):
         if unite == "Stade":
-            mois = hierodula_age_mois(age_input, sexe)
-        else:
+            return "Pour Acontista, indique l'âge en mois."
+        try:
             mois = float(age_input)
-        # On fait correspondre 10 mois ≈ 100 ans (même base que Acontista)
-        age_humain = (mois / 10) * 100
+        except:
+            return "Entrée invalide pour l'âge."
+        age_humain = age_humain_acontista(mois)
+        return f"Âge humain approximatif : {age_humain:.1f} ans"
+
+    elif espece.startswith("hierodula"):
+        if unite == "Stade":
+            age_humain = age_humain_hierodula(stade=age_input, sexe=sexe)
+        else:
+            try:
+                mois = float(age_input)
+            except:
+                return "Entrée invalide pour l'âge."
+            age_humain = age_humain_hierodula(mois=mois, sexe=sexe)
+        return f"Âge humain approximatif : {age_humain:.1f} ans"
+
     else:
         return "Espèce non reconnue."
 
-    return f"Âge humain approximatif : {age_humain:.1f} ans"
 
-# Interface Gradio
+# === Interface Gradio ===
 def main():
     with gr.Blocks() as demo:
-        gr.Markdown("# 🦗 Convertisseur d’âge de mante en âge humain")
+        gr.Markdown("# 🦗 Convertisseur d’âge de mante religieuse en âge humain")
+        gr.Markdown("## Projet de Ruben — modèle expérimental basé sur *Acontista mexicana* et *Hierodula spp.*")
+
+        espece = gr.Radio(["Acontista mexicana", "Hierodula spp."], label="Espèce", value="Acontista mexicana")
+        unite = gr.Radio(["Mois", "Stade"], label="Type d’entrée", value="Mois")
 
-        espece = gr.Radio(["Acontista mexicana", "Hierodula spp."], label="Espèce")
-        unite = gr.Radio(["Stade", "Mois"], label="Type d'entrée")
         age_input = gr.Dropdown(
-            ["L1","L2","L3","L4","L5","L6","L7","L8"],
-            label="Stade (si sélectionné)",
+            ["L1", "L2", "L3", "L4", "L5", "L6", "L7", "L8"],
+            label="Stade (si applicable)",
             value="L1"
         )
         mois_input = gr.Number(label="Âge en mois (si applicable)", value=1.0)
         sexe = gr.Radio(["Femelle", "Mâle", "Indéterminé"], label="Sexe", value="Femelle")
 
-        btn = gr.Button("Convertir")
-        output = gr.Textbox(label="Résultat")
+        bouton = gr.Button("Convertir")
+        resultat = gr.Textbox(label="Résultat")
 
-        def process(espece, unite, age, mois, sexe):
-            if unite == "Stade":
-                return convert_age(age, unite, espece, sexe)
+        # Fonction liée au bouton
+        def process(espece, unite, stade, mois, sexe):
+            if unite == "Mois":
+                return convertir_age(mois, unite, espece, sexe)
             else:
-                return convert_age(mois, unite, espece, sexe)
+                return convertir_age(stade, unite, espece, sexe)
 
-        btn.click(process, [espece, unite, age_input, mois_input, sexe], output)
+        bouton.click(process, [espece, unite, age_input, mois_input, sexe], resultat)
+
+        gr.Markdown(
+            "### ℹ️ Notes :\n"
+            "- Les équivalences sont basées sur des observations empiriques.\n"
+            "- Les *Acontista mexicana* se développent plus rapidement.\n"
+            "- Les *Hierodula spp.* ont un cycle de vie plus long et progressif.\n"
+        )
 
     demo.launch()
 
+
 if __name__ == "__main__":
     main()