main_ml.py

import tensorflow as tf
import arcade
import numpy as np
import pandas as pd
from snake import Snake
from apple import Apple

class Game(arcade.Window) :
    def __init__(self):
        super().__init__(width=512 , height=512 , title= "Super Snake 🐍🍎")
        arcade.set_background_color(arcade.color.KHAKI)  
        self.snake = Snake(self)
        self.food=Apple(self)
        self.model = tf.keras.models.load_model('weights/snake_game_model.h5')
   

    def on_draw(self):
        arcade.start_render()
        self.snake.draw()
        self.food.draw()
        #arcade.finish_render()

    def on_update(self, delta_time):
        self.snake.move()
        data = {"wu":None , 
                "wr":None ,
                "wd":None ,
                "wl":None ,
                "au":None ,
                "ar":None ,
                "ad":None ,
                "al":None ,
                "bu":None ,
                "br":None ,
                "bd":None ,
                "bl":None ,
                "dx" : None , 
                "dy" : None }
        
        dx = self.snake.center_x - self.food.center_x
        dy = self.snake.center_y - self.food.center_y


        # Data collection by calculating the distance from the apple to the head of the snake
        if self.snake.center_x == self.food.center_x and self.snake.center_y < self.food.center_y :
            data["au"] = 1
            data["ad"] = 0
            data["al"] = 0
            data["ar"] = 0 

        elif self.snake.center_x == self.food.center_x and self.snake.center_y > self.food.center_y :
            data["au"] = 0
            data["ad"] = 1
            data["al"] = 0
            data["ar"] = 0

        elif self.snake.center_x > self.food.center_x and self.snake.center_y == self.food.center_y :
            data["au"] = 0
            data["ad"] = 0
            data["al"] = 1
            data["ar"] = 0

        elif self.snake.center_x < self.food.center_x and self.snake.center_y == self.food.center_y :
            data["au"] = 0
            data["ad"] = 0
            data["al"] = 0
            data["ar"] = 1
        
        # Data collection by calculating the distance from the wall to the head of the snake
        data["wu"] = game.height - self.snake.center_y
        data["wd"] = self.snake.center_y
        data["wl"] = self.snake.center_x
        data["wr"] = game.width - self.snake.center_x
        data['dx'] = dx
        data['dy'] = dy

     

    

        
        # Data collection by calculating the distance from the snake's head to its body
        for part in self.snake.body :
             if self.snake.center_x == part['x'] and self.snake.center_y < part['y'] :
                 data['bu'] = 1
                 data['bd'] = 0
                 data['bl'] = 0
                 data['br'] = 0
             elif self.snake.center_x == part['x'] and self.snake.center_y > part['y'] :
                 data['bu'] = 0
                 data['bd'] = 1
                 data['bl'] = 0
                 data['br'] = 0
             elif self.snake.center_x < part['x'] and self.snake.center_y == part['y'] :
                 data['bu'] = 0
                 data['bd'] = 0
                 data['bl'] = 0
                 data['br'] = 1
             elif self.snake.center_x > part['x'] and self.snake.center_y == part['y'] :
                 data['bu'] = 0
                 data['bd'] = 0
                 data['bl'] = 1
                 data['br'] = 0
             elif  self.snake.center_y < part['y'] :
                 data['bu'] = 1
                 data['bd'] = 0
                 data['bl'] = 0
                 data['br'] = 0
             elif self.snake.center_y > part['y'] :
                 data['bu'] = 0
                 data['bd'] = 1
                 data['bl'] = 0
                 data['br'] = 0
             elif self.snake.center_x < part['x'] :
                 data['bu'] = 0
                 data['bd'] = 0
                 data['bl'] = 0
                 data['br'] = 1
             elif self.snake.center_x > part['x']  :
                 data['bu'] = 0
                 data['bd'] = 0
                 data['bl'] = 1
                 data['br'] = 0


    

        data = pd.DataFrame(data,index=[1])
        data.fillna(0,inplace=True)
        data = data.values

        

        output = self.model.predict(data)
        direction = output.argmax()
        print("direction" , direction)
        if direction == 0 :
            self.snake.change_x = 0
            self.snake.change_y = 1
        elif direction == 1 :
            self.snake.change_x = 1
            self.snake.change_y = 0
        elif direction == 2 :
            self.snake.change_x = 0
            self.snake.change_y = -1
        elif direction == 3 :
            self.snake.change_x = -1
            self.snake.change_y = 0



        self.snake.on_update(delta_time)
        self.food.on_update()
        
        if arcade.check_for_collision(self.snake , self.food):
            self.snake.eat(self.food)
            self.food = Apple(self)

if __name__ == "__main__" :
    game = Game()
    arcade.run()