Update app.py

app.py

@@ -1,7 +1,9 @@
 import streamlit as st
 import numpy as np
+import time
 
 # Helper functions for number theory-based emoji placement
+
 def is_prime(n):
     if n <= 1:
         return False
@@ -14,17 +16,19 @@ def fib_sequence(n):
     fib_seq = [0, 1]
     while fib_seq[-1] + fib_seq[-2] <= n:
         fib_seq.append(fib_seq[-1] + fib_seq[-2])
-    return fib_seq[2:]  # Exclude first two numbers for this use case
+    return fib_seq[2:] # Exclude first two numbers for this use case
 
 # Expanded set of emojis for landscape elements
+
 emoji_set = ["🌲", "🌳", "🍃", "🌲", "🌿", "🍄", "🏠", "🏰", "🗼", "🛤️", "🌊", "🏞️", "🌁", "🌾", "🏜️", "🏝️", "🛖", "🛤️", "🛣️", "🏕️", "🌋", "⛰️", "🧱", "🌵", "🍁", "🌼", "🌻", "🌺", "🏡", "🗺️", "🌅", "🌄", "🏙️", "🏔️", "🗻", "🏖️", "🏟️", "🏯", "🏗️", "🌇", "🌆", "🌃", "🏙️"]
 
 # Locations and emoji grid maps initialization
+
 def initialize_locations(size):
     """Initialize different locations with unique emoji grids."""
     # Placeholder for location initialization logic
     # Randomly fill grids with emojis from emoji_set for each location
-    np.random.seed(42)  # Optional: for reproducible emoji distributions
+    np.random.seed(42) # Optional: for reproducible emoji distributions
     locations = {
         'forest edge': np.random.choice(emoji_set, (size, size)),
         'deep forest': np.random.choice(emoji_set, (size, size)),
@@ -35,53 +39,60 @@ def initialize_locations(size):
     return locations
 
 # Directions for movement
+
 directions = {"North": (-1, 0), "South": (1, 0), "West": (0, -1), "East": (0, 1)}
 
 # Movement and emoji map update functions
-def move_emojis(locations, current_location, direction):
+
+def move_emojis(locations, current_location, direction, velocity):
     """Shift emojis in the specified direction with wrap-around for the current location."""
-    dx, dy = directions[direction]
+    dx, dy = [coord * velocity for coord in directions[direction]]
     locations[current_location] = np.roll(locations[current_location], shift=(dx, dy), axis=(0, 1))
     return locations
 
 # Streamlit application setup
+
 def main():
     st.title("Explore the Emoji World")
-
     size = st.sidebar.slider("Grid Size", 5, 40, 10)
+    velocity = st.sidebar.slider("Velocity (emojis/sec)", 1, 10, 1)
 
     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()))
+    if 'direction' not in st.session_state:
+        st.session_state.direction = "None"
 
+    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)
 
     col1, col2, col3 = st.columns(3)
+
     with col1:
         if st.button("West"):
-            direction = "West"
+            st.session_state.direction = "West"
+
     with col2:
         if st.button("North"):
-            direction = "North"
+            st.session_state.direction = "North"
         if st.button("Stop"):
-            direction = "None"
+            st.session_state.direction = "None"
         if st.button("South"):
-            direction = "South"
+            st.session_state.direction = "South"
+
     with col3:
         if st.button("East"):
-            direction = "East"
-
-    if 'direction' not in locals():
-        direction = "None"
-    
-    if direction != "None":
-        st.session_state.locations = move_emojis(st.session_state.locations, current_location, direction)
-        st.experimental_rerun()
+            st.session_state.direction = "East"
+
+    if st.session_state.direction != "None":
+        st.session_state.locations = move_emojis(st.session_state.locations, current_location, st.session_state.direction, velocity)
+
+    time.sleep(1)
+    st.experimental_rerun()
 
 if __name__ == "__main__":
     main()