Add Snake Game implementation with Gradio UI

app.py

@@ -0,0 +1,152 @@
+import gradio as gr
+import numpy as np
+import pygame
+import random
+from PIL import Image
+import io
+
+# Initialize pygame
+pygame.init()
+
+# Game settings
+WIDTH, HEIGHT = 400, 400
+CELL_SIZE = 20
+GRID_WIDTH, GRID_HEIGHT = WIDTH // CELL_SIZE, HEIGHT // CELL_SIZE
+
+# Colors
+BLACK = (0, 0, 0)
+GREEN = (0, 255, 0)
+RED = (255, 0, 0)
+WHITE = (255, 255, 255)
+
+class SnakeGame:
+    def __init__(self):
+        self.reset_game()
+    
+    def reset_game(self):
+        self.snake = [(GRID_WIDTH // 2, GRID_HEIGHT // 2)]
+        self.direction = (1, 0)
+        self.food = self.generate_food()
+        self.score = 0
+        self.game_over = False
+    
+    def generate_food(self):
+        while True:
+            food = (random.randint(0, GRID_WIDTH - 1), random.randint(0, GRID_HEIGHT - 1))
+            if food not in self.snake:
+                return food
+    
+    def move(self, new_direction):
+        if self.game_over:
+            return
+        
+        # Prevent moving in the opposite direction
+        if (new_direction[0] * -1, new_direction[1] * -1) != self.direction:
+            self.direction = new_direction
+        
+        # Move snake
+        head = self.snake[0]
+        new_head = (head[0] + self.direction[0], head[1] + self.direction[1])
+        
+        # Check boundaries
+        if (new_head[0] < 0 or new_head[0] >= GRID_WIDTH or 
+            new_head[1] < 0 or new_head[1] >= GRID_HEIGHT or 
+            new_head in self.snake):
+            self.game_over = True
+            return
+        
+        self.snake.insert(0, new_head)
+        
+        # Check if food eaten
+        if new_head == self.food:
+            self.score += 1
+            self.food = self.generate_food()
+        else:
+            self.snake.pop()
+    
+    def get_game_image(self):
+        # Create surface
+        surface = pygame.Surface((WIDTH, HEIGHT))
+        surface.fill(BLACK)
+        
+        # Draw snake
+        for segment in self.snake:
+            rect = pygame.Rect(segment[0] * CELL_SIZE, segment[1] * CELL_SIZE, 
+                             CELL_SIZE, CELL_SIZE)
+            pygame.draw.rect(surface, GREEN, rect)
+        
+        # Draw food
+        food_rect = pygame.Rect(self.food[0] * CELL_SIZE, self.food[1] * CELL_SIZE,
+                               CELL_SIZE, CELL_SIZE)
+        pygame.draw.rect(surface, RED, food_rect)
+        
+        # Convert to PIL Image
+        w, h = surface.get_size()
+        raw = pygame.image.tostring(surface, 'RGB')
+        pil_image = Image.frombytes('RGB', (w, h), raw)
+        
+        return pil_image
+    
+    def auto_move(self):
+        """Move snake automatically in current direction"""
+        if not self.game_over:
+            self.move(self.direction)
+        return self.get_game_image(), f"Score: {self.score}" + (" - GAME OVER" if self.game_over else "")
+
+# Initialize game
+game = SnakeGame()
+
+def move_up():
+    game.move((0, -1))
+    return game.get_game_image(), f"Score: {game.score}" + (" - GAME OVER" if game.game_over else "")
+
+def move_down():
+    game.move((0, 1))
+    return game.get_game_image(), f"Score: {game.score}" + (" - GAME OVER" if game.game_over else "")
+
+def move_left():
+    game.move((-1, 0))
+    return game.get_game_image(), f"Score: {game.score}" + (" - GAME OVER" if game.game_over else "")
+
+def move_right():
+    game.move((1, 0))
+    return game.get_game_image(), f"Score: {game.score}" + (" - GAME OVER" if game.game_over else "")
+
+def restart_game():
+    game.reset_game()
+    return game.get_game_image(), f"Score: {game.score}"
+
+def auto_play():
+    return game.auto_move()
+
+# Create Gradio interface
+with gr.Blocks(title="Snake Game") as demo:
+    gr.Markdown("# 🐍 Snake Game")
+    gr.Markdown("Use the buttons below to control the snake. Eat the red food to grow and increase your score!")
+    
+    with gr.Row():
+        game_display = gr.Image(value=game.get_game_image(), width=400, height=400)
+        
+    score_display = gr.Textbox(value=f"Score: {game.score}", label="Score", interactive=False)
+    
+    with gr.Row():
+        gr.Button("⬆️", size="sm").click(move_up, outputs=[game_display, score_display])
+    
+    with gr.Row():
+        gr.Button("⬅️", size="sm").click(move_left, outputs=[game_display, score_display])
+        gr.Button("⬇️", size="sm").click(move_down, outputs=[game_display, score_display])
+        gr.Button("➡️", size="sm").click(move_right, outputs=[game_display, score_display])
+    
+    with gr.Row():
+        gr.Button("🔄 Restart", variant="secondary").click(restart_game, outputs=[game_display, score_display])
+        gr.Button("⏯️ Auto Play", variant="primary").click(auto_play, outputs=[game_display, score_display])
+    
+    gr.Markdown("**Instructions:**")
+    gr.Markdown("- Use arrow buttons to move the snake")
+    gr.Markdown("- Eat red food to grow and score points")
+    gr.Markdown("- Don't hit walls or yourself!")
+    gr.Markdown("- Use 'Auto Play' to watch the snake move automatically")
+    gr.Markdown("- Click 'Restart' to start a new game")
+
+if __name__ == "__main__":
+    demo.launch()