Update app.py

app.py

@@ -6,7 +6,7 @@ import ast
 
 used_fields = ["classe_consommation_energie",
                "commentaires_ameliorations_recommandations",
-                "nom_methode_dpe",
+               "nom_methode_dpe",
                "usr_diagnostiqueur_id",
                "numero_dpe",
                "secteur_activite",
@@ -16,6 +16,7 @@ used_fields = ["classe_consommation_energie",
                "tv016_departement_departement",
                "long_lat",
                "address",
+               "positif_negatif",
                "commune"]
 
 
@@ -36,9 +37,10 @@ def _max_width_():
 # force screen width
 _max_width_()
 
+
 @st.cache
 def load_data():
-    return pd.read_csv("outlier_V2.csv", sep=",")# pd.read_excel("dpe-tertiaire.xlsx")
+    return pd.read_csv("outlier_V2.csv", sep=",")
 
 
 st.header("Analyse des anomalies sur les DPE")
@@ -53,8 +55,6 @@ sub_DPE = DPE[DPE.tv016_departement_departement == departe]
 commune = st.sidebar.selectbox('Séléctionnez la commune', sub_DPE['commune'].unique())
 sub_DPE = DPE[DPE.commune == commune].dropna()
 
-st.subheader("DPEs effectués")
-#st.dataframe(sub_DPE)
 st.subheader("Selectionner des certificateurs")
 certificateur = st.multiselect("Organismes certificateurs",
                                sub_DPE['organisme_certificateur'].unique(), [])
@@ -62,30 +62,36 @@ certificateur = st.multiselect("Organismes certificateurs",
 if certificateur:
     sub_DPE = sub_DPE[sub_DPE.organisme_certificateur.isin(certificateur)]
 
-
 sub_DPE.long_lat = sub_DPE.long_lat.apply(ast.literal_eval)
 
-sub_DPE["address"] = sub_DPE.address.str.replace("nan ","")
-# set the map center
-m = folium.Map(location=sub_DPE.long_lat.values[0], zoom_start=12)
-for index, row in sub_DPE.iterrows():
-    tooltip = row['address']
-
-    html = f"<h2>{row['address']}</h2><ul><li><b>Classe d'energie attribué: </b>{row['classe_consommation_energie']}</li><br>" \
-           f"<li><b>Méthode utilisée: </b>{row['nom_methode_dpe']}</li><br>" \
-           f"<li><b>Secteur Activite: </b>{row['secteur_activite']}</li><br>" \
-           f"<li><b>Année de construction: </b>{int(row['annee_construction'])}</li><br>" \
-           f"<li><b>Commentaire: </b>{row['commentaires_ameliorations_recommandations']}</li><br>" \
-           f"<li><b>Organisme Certificateur: </b>{row['organisme_certificateur']} - {row['adresse_organisme_certificateur']}</li></ul>"
-
-    iframe = folium.IFrame(html)
-    popup = folium.Popup(iframe,
-                         min_width=500,
-                         max_width=500)
-
-    folium.Marker(
-        row['long_lat'], popup=popup, tooltip=tooltip
-    ).add_to(m)
-
-# call to render Folium map in Streamlit
-folium_static(m, width=1400)
+sub_DPE["address"] = sub_DPE.address.str.replace("nan ", "")
+if sub_DPE.shape[0]:
+    # set the map center
+    m = folium.Map(location=sub_DPE.long_lat.values[0], zoom_start=12)
+    for index, row in sub_DPE.iterrows():
+        tooltip = row['address']
+
+        if row["positif_negatif"] == "sous_estimé":
+            color = "green"
+        else:
+            color = "red"
+
+        html = f"<h2>{row['address']}</h2><ul><li><b>Classe d'energie attribué: </b>{row['classe_consommation_energie']}</li><br>" \
+               f"<li><b style = 'color:{color}' >D'apres notre algorithme le DPE est {row['positif_negatif']}</b></li><br>" \
+               f"<li><b>Méthode utilisée: </b>{row['nom_methode_dpe']}</li><br>" \
+               f"<li><b>Secteur Activite: </b>{row['secteur_activite']}</li><br>" \
+               f"<li><b>Année de construction: </b>{int(row['annee_construction'])}</li><br>" \
+               f"<li><b>Commentaire: </b>{row['commentaires_ameliorations_recommandations']}</li><br>" \
+               f"<li><b>Organisme Certificateur: </b>{row['organisme_certificateur']} - {row['adresse_organisme_certificateur']}</li></ul>"
+
+        iframe = folium.IFrame(html)
+        popup = folium.Popup(iframe,
+                             min_width=500,
+                             max_width=700)
+
+        folium.Marker(
+            row['long_lat'], popup=popup, tooltip=tooltip, icon=folium.Icon(color=color),
+        ).add_to(m)
+
+    # call to render Folium map in Streamlit
+    folium_static(m, width=1400)