update markers (#2)

- update markers (57ede3a1aaecafd15a195d7c50cc5039e94d482e)


Co-authored-by: Loubna Ben Allal <loubnabnl@users.noreply.huggingface.co>

app.py

@@ -43,32 +43,6 @@ df = parse_gg_sheet(
 )
 
 
-def display_tab(input):
-    m = folium.Map(
-        location=[31.628674, -7.992047],
-        zoom_start=8.5,
-        min_zoom=8.5,
-        max_lat=35.628674,
-        min_lat=29.628674,
-        max_lon=-4.992047,
-        min_lon=-10.992047,
-        max_bounds=True,
-    )
-    filtered_df = df[df[' What are your needs? / ما هي احتياجاتك؟ (أضفها إذا لم يتم ذكرها)'] == input]
-
-    for index, row in filtered_df.iterrows():
-        if row["latlng"] is None:
-            continue
-        folium.Marker(
-            location=row["latlng"],
-            tooltip=row['  لأي قيادة / دوار تنتمون ؟'] if not pd.isna(row['  لأي قيادة / دوار تنتمون ؟']) else None,
-            popup=row["رقم الهاتف (اختياري)"] if not pd.isna(row["رقم الهاتف (اختياري)"]) else None,
-        ).add_to(m)
-    st_folium(m, use_container_width=True)
-
-
-# request_type = st.selectbox('Choose request type/اختر نوع الطلب', ("إغاثة", "مساعدة طبية", "مأوى", "طعام وماء", "مخاطر (تسرب الغاز، تلف في الخدمات العامة...)"))
-
 headers_mapping = {
     "إغاثة": "Rescue/إغاثة",
     "مساعدة طبية": "Medical Assistance/مساعدة طبية",
@@ -76,6 +50,13 @@ headers_mapping = {
     "طعام وماء": "Food & Water/طعام وماء",
     "مخاطر (تسرب الغاز، تلف في الخدمات العامة...)": "Danger/مخاطر",
 }
+colors_mapping = {
+    "إغاثة": "red",
+    "مساعدة طبية": "orange",
+    "مأوى": "yellow",
+    "طعام وماء": "blue",
+    "مخاطر (تسرب الغاز، تلف في الخدمات العامة...)": "grey",
+}
 options = ["إغاثة", "مساعدة طبية", "مأوى", "طعام وماء", "مخاطر (تسرب الغاز، تلف في الخدمات العامة...)"]
 selected_options = []
 
@@ -89,7 +70,7 @@ for i, option in enumerate(options):
     if checked:
         selected_options.append(option)
 
-
+df['id'] = df.index
 filtered_df = df[df[' What are your needs? / ما هي احتياجاتك؟ (أضفها إذا لم يتم ذكرها)'].isin(selected_options)]
 selected_headers = [headers_mapping[request] for request in selected_options]
 st.subheader(", ".join(selected_headers))
@@ -105,13 +86,18 @@ m = folium.Map(
     max_bounds=True,
 )
 
+
 for index, row in filtered_df.iterrows():
+    request_type = row[' What are your needs? / ما هي احتياجاتك؟ (أضفها إذا لم يتم ذكرها)']
+    phone = row["رقم الهاتف (اختياري)"] if not pd.isna(row["رقم الهاتف (اختياري)"]) else ""
+    display_text = f"{request_type} {phone}"
     if row["latlng"] is None:
         continue
     folium.Marker(
         location=row["latlng"],
         tooltip=row['  لأي قيادة / دوار تنتمون ؟'] if not pd.isna(row['  لأي قيادة / دوار تنتمون ؟']) else None,
-        popup=row["رقم الهاتف (اختياري)"] if not pd.isna(row["رقم الهاتف (اختياري)"]) else None,
+        popup=display_text,
+        icon=folium.Icon(colors_mapping[request_type])
     ).add_to(m)
 st_data = st_folium(m, use_container_width=True)