Update app.py

app.py

@@ -1,65 +1,87 @@
 import streamlit as st
-import folium
-import dask.dataframe as dd
-from folium.plugins import MarkerCluster
-# from folium import folium_static
+import altair as alt
+from vega_datasets import data
 import pandas as pd
+import pydeck as pdk
 
-# Top 10 locations for viewing Northern Lights in Iceland
-northern_lights_locations = [
-    {"name": "Reykjavik", "lat": 64.1265, "lon": -21.8174},
-    {"name": "Akureyri", "lat": 65.6885, "lon": -18.1262},
-    {"name": "Vik", "lat": 63.4194, "lon": -19.0096},
-    {"name": "Hofn", "lat": 64.2539, "lon": -15.2082},
-    {"name": "Husavik", "lat": 66.0449, "lon": -17.3382},
-    {"name": "Grundarfjordur", "lat": 64.9226, "lon": -23.2543},
-    {"name": "Egilsstadir", "lat": 65.2669, "lon": -14.3948},
-    {"name": "Seydisfjordur", "lat": 65.2592, "lon": -14.0101},
-    {"name": "Isafjordur", "lat": 66.0750, "lon": -23.1266},
-    {"name": "Kirkjufell", "lat": 64.9426, "lon": -23.3061},
-]
-
-# Function to display the map
-def display_map(locations, lat, lon):
-    # Create a folium map centered in Iceland
-    iceland_map = folium.Map(location=[lat, lon], zoom_start=6)
-
-    # Create a marker cluster group
-    marker_cluster = MarkerCluster().add_to(iceland_map)
+# Define the data source for the map
+iceland_geojson = "https://raw.githubusercontent.com/deldersveld/topojson/master/countries/iceland/iceland-counties.json"
 
-    # Create a Dask dataframe from the list of locations
-    dask_df = dd.from_pandas(pd.DataFrame(locations), npartitions=4)
+# Define the mythological cities with their respective latitude and longitude
+cities = {
+    "Asgard": [64.7938, -17.3413],
+    "Helheim": [63.8278, -21.1865],
+    "Jotunheim": [64.8441, -19.1669],
+    "Midgard": [63.9863, -22.6210],
+    "Muspellheim": [65.3201, -16.4235],
+    "Nidavellir": [64.9011, -18.0580],
+    "Svartalfheim": [64.0114, -21.4504],
+    "Valhalla": [63.7267, -19.6133],
+    "Vanaheim": [64.7381, -17.4497],
+    "Yggdrasil": [64.8999, -19.0044]
+}
 
-    # Add markers for the top 10 locations for viewing Northern Lights
-    def add_markers(row):
-        folium.Marker(
-            location=[row["lat"], row["lon"]],
-            popup=f"<b>{row['name']}</b>",
-        ).add_to(marker_cluster)
+# Define the colors to use for each city marker
+colors = {
+    "Asgard": [255, 0, 0],
+    "Helheim": [255, 255, 0],
+    "Jotunheim": [0, 0, 255],
+    "Midgard": [0, 255, 0],
+    "Muspellheim": [255, 153, 0],
+    "Nidavellir": [153, 0, 255],
+    "Svartalfheim": [0, 255, 255],
+    "Valhalla": [255, 0, 255],
+    "Vanaheim": [153, 255, 0],
+    "Yggdrasil": [255, 255, 255]
+}
 
-    dask_df.map_partitions(lambda df: df.apply(add_markers, axis=1)).compute()
+# Define the Streamlit app layout
+st.set_page_config(layout="wide")
+st.title("Mythological Cities in Iceland")
+st.sidebar.title("Select a City to View the Northern Lights From")
+selected_city = st.sidebar.selectbox("", sorted(cities.keys()))
 
-    return iceland_map
+# Load the Icelandic county boundaries and prepare the data for the 3D map
+iceland_data = alt.topo_feature(iceland_geojson, "iceland-counties")
+iceland_df = pd.DataFrame(iceland_data["features"])
 
-# Streamlit app
-st.title("Northern Lights Map in Iceland")
-
-# User input for latitude and longitude
-lat = st.number_input("Enter latitude:", value=64.9631, step=0.0001)
-lon = st.number_input("Enter longitude:", value=-19.0208, step=0.0001)
+# Create the 3D map with Deck.gl and Altair
+view_state = pdk.ViewState(latitude=64.9, longitude=-18.5, zoom=5, pitch=40)
+layers = [
+    pdk.Layer(
+        "PolygonLayer",
+        data=iceland_data,
+        get_polygon="coordinates",
+        filled=True,
+        extruded=True,
+        get_elevation="properties.avg_elevation",
+        elevation_scale=1000,
+        get_fill_color=[200, 200, 200, 200]
+    ),
+    pdk.Layer(
+        "ScatterplotLayer",
+        data=pd.DataFrame({"latitude": [cities[selected_city][0]], "longitude": [cities[selected_city][1]]}),
+        get_position="[longitude, latitude]",
+        get_color=colors[selected_city],
+        get_radius=20000
+    )
+]
 
-# Display the map
-st.write("Northern Lights Viewing Locations in Iceland:")
-folium_map = display_map(northern_lights_locations, lat, lon)
-folium_static(folium_map)
+r = pdk.Deck(layers=layers, initial_view_state=view_state)
+altair_chart = alt.Chart(iceland_df).mark_geoshape(
+stroke="black",
+strokeWidth=0.5
+    ).encode(
+        color=alt.Color("properties.avg_elevation:Q", scale=alt.Scale(scheme="viridis")),
+        tooltip=[alt.Tooltip("properties.name", title="County"), alt.Tooltip("properties.avg_elevation:Q", title="Elevation (m)")]
+    ).transform_lookup(
+    lookup="id",
+    from_=alt.LookupData(iceland_data, "id", ["properties.avg_elevation", "properties.name"])
+    ).properties(
+    width=900,
+    height=600
+).interactive()
 
-# Cite references
-st.write(
-    """
-    References:
-    - [Folium documentation](https://python-visualization.github.io/folium/)
-    - [Streamlit documentation](https://docs.streamlit.io/)
-    - [Dask documentation](https://dask.org/)
-    - [Top 10 locations](https://www.guidetoiceland.is/best-of-iceland/top-10-places-to-see-the-northern-lights)
-"""
-)
+Display the 3D map and the Altair chart
+st.pydeck_chart(r)
+st.altair_chart(altair_chart)