Pytorch V0.5

app.py

@@ -1,4 +1,3 @@
-
 import streamlit as st
 import pandas as pd
 import requests
@@ -10,43 +9,16 @@ def get_data():
     response = requests.get(url)
     if response.status_code == 200:
         data = response.json()
-        records = data.get("records", [])
-        # 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)
+        records = data["records"]
+        return [record["fields"] for record in records]
+    return []
 
-def display_organisations_engagees():
+def display_organisations_engagees(data):
     st.markdown("## OPEN DATA RSE")
     st.markdown("### Découvrez les organisations engagées RSE de la métropole de Bordeaux")
     
-    data, _ = get_data()
-    if data:
-        df = pd.DataFrame(data)
+    df = pd.DataFrame(data)
+    if not df.empty:
         df = df.rename(columns={
             "nom_courant_denomination": "Nom",
             "commune": "Commune",
@@ -55,16 +27,30 @@ def display_organisations_engagees():
             "action_rse": "Action RSE"
         })
         df = df[["Nom", "Commune", "Section NAF", "Effectif", "Action RSE"]]
-        st.dataframe(df, width=None, height=None)
+        st.dataframe(df)
+    else:
+        st.write("Aucune donnée disponible.")
+
+def display_map(data):
+    m = folium.Map(location=[44.837789, -0.57918], zoom_start=12)
+    for item in data:
+        if 'latitude' in item and 'longitude' in item:
+            folium.Marker(
+                [item['latitude'], item['longitude']],
+                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"])
 
+    data = get_data()
+
     if app_mode == "Organisations engagées":
-        display_organisations_engagees()
+        display_organisations_engagees(data)
     elif app_mode == "Localisation des Entreprises":
-        data, _ = get_data()
         display_map(data)
 
 if __name__ == "__main__":