Update app.py

app.py

@@ -59,6 +59,36 @@ next_week = [f"{(today + timedelta(days=d)).strftime('%Y-%m-%d')},{(today + time
 
 df = pd.DataFrame(data=[preds, poll_level], index=["AQI", "Air pollution level"], columns=next_week)
 
+
+
+st.write(36 * "-")
+fancy_header(f"🗺 Processing the map...")
+
+fig = Figure(width=550,height=350)
+
+my_map = folium.Map(location=[58, 20], zoom_start=3.71)
+fig.add_child(my_map)
+folium.TileLayer('Stamen Terrain').add_to(my_map)
+folium.TileLayer('Stamen Toner').add_to(my_map)
+folium.TileLayer('Stamen Water Color').add_to(my_map)
+folium.TileLayer('cartodbpositron').add_to(my_map)
+folium.TileLayer('cartodbdark_matter').add_to(my_map)
+folium.LayerControl().add_to(my_map)
+
+data_to_display = data_to_display[["city", "temp", "humidity",
+                                            "conditions", "aqi"]]
+
+cities_coords = {("Sundsvall", "Sweden"): [62.390811, 17.306927],
+                 ("Stockholm", "Sweden"): [59.334591, 18.063240],
+                 ("Malmo", "Sweden"): [55.604981, 13.003822]}
+
+if "Kyiv" in data_to_display["city"]:
+    cities_coords[("Kyiv", "Ukraine")]: [50.450001, 30.523333]
+
+data_to_display = data_to_display.set_index("city")
+
+
+
 st.write(df)
 
 st.button("Re-run")