Update app.py

app.py

@@ -105,26 +105,24 @@ def create_map_figure(df, title="Carte des Foyers"):
         fig.update_layout(mapbox_style="open-street-map")
         return fig
     
-    # CORRECTION : Rendre la sélection de la colonne de taille plus robuste
-    size_col_name = None
-    if 'totalAnimalsAffected' in df.columns:
-        size_col_name = 'totalAnimalsAffected'
-    elif 'cases' in df.columns:
-        size_col_name = 'cases'
-    elif 'deaths' in df.columns:
-        size_col_name = 'deaths'
-
-    df_plot = df.copy() # Travailler sur une copie pour ne pas modifier l'original
+    df_plot = df.copy()
     
+    size_col_name = None
+    if 'totalAnimalsAffected' in df_plot.columns: size_col_name = 'totalAnimalsAffected'
+    elif 'cases' in df_plot.columns: size_col_name = 'cases'
+    elif 'deaths' in df_plot.columns: size_col_name = 'deaths'
+
+    # CORRECTION : Utiliser 'locationName' qui est le bon nom de colonne
+    hover_name_col = 'locationName' if 'locationName' in df_plot.columns else None
+
     if size_col_name:
         df_plot['size_for_plotting'] = pd.to_numeric(df_plot[size_col_name], errors='coerce').fillna(1)
         fig = px.scatter_mapbox(df_plot, lat="latitude", lon="longitude", color="diseaseName", 
                                 size='size_for_plotting', size_max=30, 
-                                hover_name="location", title=title, zoom=1, height=600)
+                                hover_name=hover_name_col, title=title, zoom=1, height=600)
     else:
-        # Si aucune colonne de taille n'est trouvée, dessiner des points de taille égale
         fig = px.scatter_mapbox(df_plot, lat="latitude", lon="longitude", color="diseaseName", 
-                                hover_name="location", title=title, zoom=1, height=600)
+                                hover_name=hover_name_col, title=title, zoom=1, height=600)
 
     fig.update_layout(mapbox_style="open-street-map", margin={"r":0,"t":40,"l":0,"b":0})
     return fig
@@ -142,7 +140,7 @@ with gr.Blocks(theme=gr.themes.Soft(), title="WAHIS Scraper") as demo:
     gr.Markdown("# 🤖 Scraper pour WAHIS (WOAH) - Version Robuste")
     run_button = gr.Button("🚀 Lancer l'extraction des données", variant="primary")
     
-    status_message = gr.Markdown("") # Pour afficher des messages d'état
+    status_message = gr.Markdown("")
 
     with gr.Accordion("Journal d'exécution", open=False):
         status_textbox = gr.Textbox(lines=15, label="📜 Logs", interactive=False)