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
    for i in range(2, int(n**0.5) + 1):
        if n % i == 0:
            return False
    return True

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

# Expanded set of emojis for landscape elements
emoji_set = ["🌲", "🌳", "🍃", "🌲", "🌿", "🍄", "🏠", "🏰", "🗼", "🛤️", "🌊", "🏞️", "🌁", "🌾", "🏜️", "🏝️", "🛖", "🛤️", "🛣️", "🏕️", "🌋", "⛰️", "🧱", "🌵", "🍁", "🌼", "🌻", "🌺", "🏡", "🗺️", "⛺", "🌅", "🌄", "🌠", "🍂", "🍀", "🌴", "🎄", "🌵", "🌙", "⭐", "🌤️", "🌪"]

# Initialize locations with more varied and staggered emojis
def initialize_locations(size):
    np.random.seed(42)  # For reproducibility
    locations = {
        'forest edge': np.random.choice(emoji_set, (size, size)),
        'deep forest': np.random.choice(emoji_set, (size, size)),
        # Add other locations with corresponding maps as needed
    }
    return locations

# Directions for movement
directions = {"North": (-1, 0), "South": (1, 0), "West": (0, -1), "East": (0, 1)}

# Shift emojis based on direction with wrap-around
def move_emojis(locations, current_location, direction):
    dx, dy = directions[direction]
    locations[current_location] = np.roll(locations[current_location], shift=(dx, dy), axis=(0, 1))
    return locations

# WASD movement interface
def wasd_interface():
    direction = None
    with st.form("wasd_form", clear_on_submit=True):
        direction_input = st.text_input("Enter direction (WASD):", "")
        submitted = st.form_submit_button("Move")
        if submitted and direction_input.upper() in ['W', 'A', 'S', 'D']:
            direction = {'W': 'North', 'A': 'West', 'S': 'South', 'D': 'East'}[direction_input.upper()]
    return direction

# Main application
def main():
    st.title("Explore the Emoji World")

    size = st.sidebar.slider("Grid Size", 5, 40, 10)
    delay = st.sidebar.slider("Movement Delay (seconds)", 0.0, 1.0, 0.1)

    if 'locations' not in st.session_state:
        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 = wasd_interface()
    if direction:
        st.session_state.locations = move_emojis(st.session_state.locations, current_location, direction)
        time.sleep(delay)  # Delay to adjust the speed of movement
        st.experimental_rerun()

if __name__ == "__main__":
    main()