import streamlit as st
import numpy as np

# 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

nature_emojis = ["🌲", "🌳", "🍃", "🌿", "🍄", "🌊", "🏞️", "🌁", "🌾", "🏜️", "🏝️", "🌋", "⛰️", "🌵", "🍁", "🌼", "🌻", "🌺"]
building_emojis = ["🏠", "🏰", "🗼", "🛖", "🏕️", "🧱", "🏡"]
infrastructure_emojis = ["🛤️", "🛣️", "🗺️"]
bird_emojis = ["🦜", "🦚", "🦉", "🦆", "🦅", "🦩", "🕊️", "🦃"]
animal_emojis = ["🐒", "🦍", "🐘", "🦏", "🐆", "🐅", "🐃", "🐂", "🐄", "🐪", "🐫", "🦙", "🦘", "🦥", "🐿️"]
fish_emojis = ["🐠", "🐟", "🐡", "🦈", "🐙", "🐚", "🦀", "🦞", "🦐", "🐬", "🐳", "🐋"]

# Locations and emoji grid maps initialization

def initialize_locations(size):
    """Initialize different locations with unique emoji grids."""
    np.random.seed(42) # Optional: for reproducible emoji distributions
    locations = {
        'forest edge': np.random.choice(nature_emojis + bird_emojis + animal_emojis, (size, size)),
        'deep forest': np.random.choice(nature_emojis + bird_emojis + animal_emojis, (size, size)),
        'mountain range': np.random.choice(nature_emojis + animal_emojis, (size, size)),
        'coastal town': np.random.choice(building_emojis + infrastructure_emojis + fish_emojis, (size, size)),
        'desert oasis': np.random.choice(nature_emojis + animal_emojis, (size, size)),
        'urban cityscape': np.random.choice(building_emojis + infrastructure_emojis, (size, size)),
        'tropical island': np.random.choice(nature_emojis + bird_emojis + fish_emojis, (size, size)),
        'savanna plains': np.random.choice(nature_emojis + animal_emojis + bird_emojis, (size, 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):
    """Shift emojis in the specified direction with wrap-around for the current location."""
    dx, dy = 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, 50, 20)

    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 = 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()

if __name__ == "__main__":
    main()