Update app.py

app.py

@@ -17,7 +17,7 @@ def fib_sequence(n):
     return fib_seq[2:]  # Exclude first two numbers for this use case
 
 # Expanded set of emojis for landscape elements
-emoji_set = ["🌲", "🌳", "🍃", "🌲", "🌿", "🍄", "🏠", "🏰", "🗼", "🛤️", "🌊", "🏞️", "🌁", "🌾", "🏜️", "🏝️", "🛖", "🛤️", "🛣️", "🏕️", "🌋", "⛰️", "🧱", "🌵", "🍁", "🌼", "🌻", "🌺", "🏡", "🗺️"]
+emoji_set = ["🌲", "🌳", "🍃", "🌲", "🌿", "🍄", "🏠", "🏰", "🗼", "🛤️", "🌊", "🏞️", "🌁", "🌾", "🏜️", "🏝️", "🛖", "🛤️", "🛣️", "🏕️", "🌋", "⛰️", "🧱", "🌵", "🍁", "🌼", "🌻", "🌺", "🏡", "🗺️", "🌅", "🌄", "🏙️", "🏔️", "🗻", "🏖️", "🏟️", "🏯", "🏗️", "🌇", "🌆", "🌃", "🏙️"]
 
 # Locations and emoji grid maps initialization
 def initialize_locations(size):
@@ -28,7 +28,9 @@ def initialize_locations(size):
     locations = {
         'forest edge': np.random.choice(emoji_set, (size, size)),
         'deep forest': np.random.choice(emoji_set, (size, size)),
-        # Additional locations can be added here
+        'mountain range': np.random.choice(emoji_set, (size, size)),
+        'coastal town': np.random.choice(emoji_set, (size, size)),
+        'desert oasis': np.random.choice(emoji_set, (size, size)),
     }
     return locations
 
@@ -47,18 +49,32 @@ def main():
     st.title("Explore the Emoji World")
 
     size = st.sidebar.slider("Grid Size", 5, 40, 10)
+
     if 'locations' not in st.session_state:
         st.session_state.locations = initialize_locations(size)
-    
+    else:
+        st.session_state.locations = initialize_locations(size)
+
     current_location = st.sidebar.selectbox("Select location", options=list(st.session_state.locations.keys()))
+
     emoji_map = st.session_state.locations[current_location]
     map_str = "\n".join(["".join(row) for row in emoji_map])
     st.text(map_str)
 
-    direction = st.sidebar.selectbox("Move direction", ["None", "North", "South", "East", "West"])
-    if direction != "None":
-        st.session_state.locations = move_emojis(st.session_state.locations, current_location, direction)
-        st.experimental_rerun()
+    col1, col2, col3 = st.columns(3)
+    with col1:
+        if st.button("West"):
+            st.session_state.locations = move_emojis(st.session_state.locations, current_location, "West")
+    with col2:
+        if st.button("North"):
+            st.session_state.locations = move_emojis(st.session_state.locations, current_location, "North")
+        if st.button("Stop"):
+            pass
+        if st.button("South"):
+            st.session_state.locations = move_emojis(st.session_state.locations, current_location, "South")
+    with col3:
+        if st.button("East"):
+            st.session_state.locations = move_emojis(st.session_state.locations, current_location, "East")
 
 if __name__ == "__main__":
     main()