Pytorch V0.4

app.py

@@ -11,10 +11,35 @@ def get_data():
     if response.status_code == 200:
         data = response.json()
         records = data.get("records", [])
-        return [record["fields"] for record in records], data.get("nhits", 0)
+        # Ensure that 'point_geo' is extracted correctly as a dictionary with 'lat' and 'lon'
+        cleaned_data = []
+        for record in records:
+            item = record["fields"]
+            point_geo = item.get("point_geo", {})
+            if isinstance(point_geo, dict):
+                lat = point_geo.get("lat")
+                lon = point_geo.get("lon")
+                if lat and lon:
+                    item['latitude'] = lat
+                    item['longitude'] = lon
+                    cleaned_data.append(item)
+        return cleaned_data, data.get("nhits", 0)
     else:
         return [], 0
 
+def display_map(data):
+    m = folium.Map(location=[44.837789, -0.57918], zoom_start=12)
+    for item in data:
+        lat = item.get('latitude')
+        lon = item.get('longitude')
+        if lat and lon:
+            folium.Marker(
+                [lat, lon],
+                icon=folium.Icon(color="green", icon="leaf"),
+                popup=item.get('Nom', 'Sans nom'),
+            ).add_to(m)
+    folium_static(m)
+
 def display_organisations_engagees():
     st.markdown("## OPEN DATA RSE")
     st.markdown("### Découvrez les organisations engagées RSE de la métropole de Bordeaux")
@@ -32,18 +57,6 @@ def display_organisations_engagees():
         df = df[["Nom", "Commune", "Section NAF", "Effectif", "Action RSE"]]
         st.dataframe(df, width=None, height=None)
 
-def display_map(data):
-    m = folium.Map(location=[44.837789, -0.57918], zoom_start=12)
-    for item in data:
-        lat_lon = item.get('point_geo', None)
-        if lat_lon:
-            folium.Marker(
-                [lat_lon['lat'], lat_lon['lon']],
-                icon=folium.Icon(color="green", icon="leaf"),
-                popup=item.get('Nom', 'Sans nom'),
-            ).add_to(m)
-    folium_static(m)
-
 def main():
     st.sidebar.title("Navigation")
     app_mode = st.sidebar.radio("Choisissez l'onglet", ["Organisations engagées", "Localisation des Entreprises"])