Add village status selection and description to the markers, change marker color if village was visited but is in critical condition

app.py

@@ -7,7 +7,6 @@ import pandas as pd
 import streamlit as st
 from huggingface_hub import HfApi
 from streamlit_folium import st_folium
-from src.utils import add_latlng_col
 
 from src.text_content import (
     COLOR_MAPPING,
@@ -21,18 +20,21 @@ from src.text_content import (
     REVIEW_TEXT,
     SLOGAN,
 )
-from src.utils import init_map, parse_gg_sheet
+from src.utils import add_latlng_col, init_map, parse_gg_sheet
 
 TOKEN = os.environ.get("HF_TOKEN", None)
 REQUESTS_URL = "https://docs.google.com/spreadsheets/d/1gYoBBiBo1L18IVakHkf3t1fOGvHWb23loadyFZUeHJs/edit#gid=966953708"
-INTERVENTIONS_URL = (
-    "https://docs.google.com/spreadsheets/d/1eXOTqunOWWP8FRdENPs4cU9ulISm4XZWYJJNR1-SrwY/edit#gid=2089222765"
-)
+INTERVENTIONS_URL = "https://docs.google.com/spreadsheets/d/1eXOTqunOWWP8FRdENPs4cU9ulISm4XZWYJJNR1-SrwY/edit#gid=2089222765"
 api = HfApi(TOKEN)
 
 
 # Initialize Streamlit Config
-st.set_page_config(layout="wide", initial_sidebar_state="collapsed", page_icon="🤝", page_title="Nt3awnou نتعاونو")
+st.set_page_config(
+    layout="wide",
+    initial_sidebar_state="collapsed",
+    page_icon="🤝",
+    page_title="Nt3awnou نتعاونو",
+)
 
 # Initialize States
 if "sleep_time" not in st.session_state:
@@ -42,7 +44,9 @@ if "auto_refresh" not in st.session_state:
 
 auto_refresh = st.sidebar.checkbox("Auto Refresh?", st.session_state.auto_refresh)
 if auto_refresh:
-    number = st.sidebar.number_input("Refresh rate in seconds", value=st.session_state.sleep_time)
+    number = st.sidebar.number_input(
+        "Refresh rate in seconds", value=st.session_state.sleep_time
+    )
     st.session_state.sleep_time = number
 
 
@@ -50,21 +54,37 @@ if auto_refresh:
 def display_interventions(interventions_df, m):
     """Display NGO interventions on the map"""
     for index, row in interventions_df.iterrows():
+        village_status = row[interventions_df.columns[7]]
         status = (
             "Done ✅"
-            if row[interventions_df.columns[5]] != "Intervention prévue dans le futur / Planned future intervention"
+            if row[interventions_df.columns[5]]
+            != "Intervention prévue dans le futur / Planned future intervention"
+            and village_status
+            != "Critique, Besoin d'aide en urgence / Critical, in urgent need of help"
             else "Planned ⌛"
         )
-        color_mk = (
-            "green"
-            if row[interventions_df.columns[5]] != "Intervention prévue dans le futur / Planned future intervention"
-            else "pink"
-        )
+        if (
+            row[interventions_df.columns[5]]
+            != "Intervention prévue dans le futur / Planned future intervention"
+            and village_status
+            != "Critique, Besoin d'aide en urgence / Critical, in urgent need of help"
+        ):
+            color_mk = "green"
+        elif (
+            row[interventions_df.columns[5]]
+            != "Intervention prévue dans le futur / Planned future intervention"
+            and village_status
+            == "Critique, Besoin d'aide en urgence / Critical, in urgent need of help"
+        ):
+            color_mk = "darkgreen"
+        else:
+            color_mk = "pink"
+
         intervention_type = row[interventions_df.columns[6]].split("/")[0].strip()
         org = row[interventions_df.columns[1]]
         city = row[interventions_df.columns[9]]
         date = row[interventions_df.columns[4]]
-        intervention_info = f"<b>Status:</b> {status}<br><b>Org:</b> {org}<br><b>Intervention:</b> {intervention_type}<br><b>📅 Date:</b> {date}"
+        intervention_info = f"<b>Intervention Status:</b> {status}<br><b>Village Status:</b> {village_status.split('/')[0]}<br><b>Org:</b> {org}<br><b>Intervention:</b> {intervention_type}<br><b>📅 Date:</b> {date}"
         if row["latlng"] is None:
             continue
         folium.Marker(
@@ -94,7 +114,9 @@ def show_requests(filtered_df, m):
             if not pd.isna(row["  لأي  جماعة / قيادة / دوار تنتمون ؟"])
             else None,
             popup=folium.Popup(display_text, max_width=300),
-            icon=folium.Icon(color=COLOR_MAPPING.get(request_type, "blue"), icon=icon_name),
+            icon=folium.Icon(
+                color=COLOR_MAPPING.get(request_type, "blue"), icon=icon_name
+            ),
         ).add_to(m)
 
 
@@ -146,10 +168,14 @@ def display_dataframe(df, drop_cols, data_url, search_id=True, status=False):
     if status:
         target = "Pouvez-vous nous préciser si vous êtes déjà intervenus ou si vous prévoyez de le faire | Tell us if you already made the intervention, or if you're planning to do it"
         if selected_status == "Done / تم":
-            display_df = display_df[display_df[target] == "Intervention déjà passée / Past intevention"]
+            display_df = display_df[
+                display_df[target] == "Intervention déjà passée / Past intevention"
+            ]
 
         elif selected_status == "Planned / مخطط لها":
-            display_df = display_df[display_df[target] != "Intervention déjà passée / Past intevention"]
+            display_df = display_df[
+                display_df[target] != "Intervention déjà passée / Past intevention"
+            ]
 
     st.dataframe(display_df, height=500)
     st.markdown(
@@ -175,7 +201,9 @@ def id_review_submission():
     st.subheader("🔍 Review of requests")
     st.markdown(REVIEW_TEXT)
 
-    id_to_review = st.number_input("Enter id / أدخل الرقم", min_value=0, max_value=len(df), value=0, step=1)
+    id_to_review = st.number_input(
+        "Enter id / أدخل الرقم", min_value=0, max_value=len(df), value=0, step=1
+    )
     reason_for_review = st.text_area("Explain why / أدخل سبب المراجعة")
     if st.button("Submit / أرسل"):
         if reason_for_review == "":
@@ -190,7 +218,9 @@ def id_review_submission():
                 repo_id="nt3awnou/review_requests",
                 repo_type="dataset",
             )
-            st.success("Submitted at https://huggingface.co/datasets/nt3awnou/review_requests/ تم الإرسال")
+            st.success(
+                "Submitted at https://huggingface.co/datasets/nt3awnou/review_requests/ تم الإرسال"
+            )
 
 
 # Logo and Title
@@ -215,14 +245,10 @@ options = [
 ]
 selected_options = []
 
-# with tab_en:
-#     st.markdown("👉 **Choose request type**")
-# with tab_ar:
-#     st.markdown("👉 **اختر نوع الطلب**")
-# with tab_fr:
-#     st.markdown("👉 **Choisissez le type de demande**")
 
-st.markdown("👉 **Choose request type | Choissisez le type de demande | اختر نوع الطلب**")
+st.markdown(
+    "👉 **Choose request type | Choissisez le type de demande | اختر نوع الطلب**"
+)
 col1, col2, col3, col4, col5 = st.columns([2, 3, 2, 3, 4])
 cols = [col1, col2, col3, col4, col5]
 
@@ -241,8 +267,32 @@ show_interventions = st.checkbox(
     value=True,
 )
 
+# Categories of villages
+
+st.markdown(
+    "<br>👉 **State of villages visited by NGOs| Etat de villages visités par les ONGs | اختر نوع القرى التي زارتها الجمعيات**",
+    unsafe_allow_html=True,
+)
+
+village_types = ["Critique", "Partiellement satisfait", "Entièrement satisfait "]
+selected_village_types = st.multiselect(
+    " ",
+    village_types,
+    default=village_types,
+)
+status_mapping = {
+    "Critique": "Critique, Besoin d'aide en urgence / Critical, in urgent need of help",
+    "Partiellement satisfait": "Partiellement satisfait / Partially Served",
+    "Entièrement satisfait ": "Entièrement satisfait / Fully served",
+}
+selected_statuses = [status_mapping[status] for status in selected_village_types]
+
 if show_interventions:
-    # print(interventions_df.columns)
+    interventions_df = interventions_df.loc[
+        interventions_df[
+            "Etat de la région actuel | Current situation of the area "
+        ].isin(selected_statuses)
+    ]
     display_interventions(interventions_df, m)
 
 # Show requests
@@ -269,6 +319,7 @@ drop_cols = [
     "GeoCode",
     "GeoAddress",
     "Status",
+    "id",
 ]
 display_dataframe(filtered_df, drop_cols, REQUESTS_URL, search_id=True)
 
@@ -277,7 +328,7 @@ st.divider()
 st.subheader("📝 **Table of interventions / جدول التدخلات**")
 display_dataframe(
     interventions_df,
-    [], # We show NGOs contact information
+    [],  # We show NGOs contact information
     INTERVENTIONS_URL,
     search_id=False,
     status=True,