Code Editor for app.py

import streamlit as st
import numpy as np
import plotly.graph_objects as go

st.title("Snake Game")

# Game settings
snake_length = 3
snake = [(0, 0), (0, 1), (0, 2)]  # Start with an initial snake
food = (5, 5)

# Initialize game state
if "score" not in st.session_state:
    st.session_state.score = 0

if "snake" not in st.session_state:
    st.session_state.snake = snake

if "food" not in st.session_state:
    st.session_state.food = food

# Functions to render the game
def update_snake(direction):
    dx, dy = direction
    new_head = (st.session_state.snake[0][0] + dx, st.session_state.snake[0][1] + dy)
    st.session_state.snake.insert(0, new_head)
    if new_head == st.session_state.food:
        st.session_state.score += 1
        place_food()
    else:
        st.session_state.snake.pop()

def place_food():
    while True:
        new_food = (np.random.randint(0, 10), np.random.randint(0, 10))
        if new_food not in st.session_state.snake:
            st.session_state.food = new_food
            break

def render_game():
    fig = go.Figure()
    # Draw snake
    snake_x, snake_y = zip(*st.session_state.snake)
    fig.add_trace(go.Scatter(x=snake_x, y=snake_y, mode="markers+lines", marker=dict(size=20, color="green")))

    # Draw food
    food_x, food_y = st.session_state.food
    fig.add_trace(go.Scatter(x=[food_x], y=[food_y], mode="markers", marker=dict(size=20, color="red")))

    # Update layout
    fig.update_layout(xaxis=dict(range=[0, 9]), yaxis=dict(range=[0, 9]),
                      width=400, height=400, showlegend=False)
    st.plotly_chart(fig, use_container_width=True)

# Control direction from streamlit sidebar
st.sidebar.title("Controls")
if st.sidebar.button("Up"):
    update_snake((0, 1))
if st.sidebar.button("Down"):
    update_snake((0, -1))
if st.sidebar.button("Left"):
    update_snake((-1, 0))
if st.sidebar.button("Right"):
    update_snake((1, 0))

# Render game
render_game()
st.sidebar.write(f"Score: {st.session_state.score}")