add gameplay

Signed-off-by: Better nya <nya@nya.com>

main.py

@@ -1,16 +1,37 @@
 import sys
 import pygame
+
 pygame.init()
 #SETUP
-WIDTH, HEIGHT = 800, 800
+WIDTH, HEIGHT = 800, 600
-WINDOW = pygame.display.set_mode((WIDTH, HEIGHT))
+screen = pygame.display.set_mode((WIDTH, HEIGHT))
 pygame.display.set_caption("Traffic Controller")
 
 #CONSTANTS
 FPS = 60
 
-# Game loop
+#CLOCK
 clock = pygame.time.Clock()
+
+#COLOURS
+
+ROAD_COLOUR = (50, 50, 50)
+LINE_COLOUR = (255, 255, 0)
+CAR_COLOUR = (200, 0, 0)
+BG_COLOUR = (30, 150, 30)
+
+#CAR
+car_width = 40
+car_height = 70
+car_x = WIDTH // 2 - car_width // 2
+car_y = HEIGHT - car_height - 20
+car_speed = 4
+lane_speed = 5
+
+#LANE
+lane_y = 0
+
+#LOOP
 running = True
 while running:
     clock.tick(FPS)
@@ -20,3 +41,31 @@ while running:
             pygame.quit()
             sys.exit()
 
+    #INPUT
+    keys = pygame.key.get_pressed()
+    if keys[pygame.K_LEFT]:
+        car_x -= car_speed
+    if keys[pygame.K_RIGHT]:
+        car_x += car_speed
+
+    #LANE SPEED
+    lane_y += lane_speed
+    if lane_y > HEIGHT:
+        lane_y = 0
+
+    #BG
+    screen.fill(BG_COLOUR)
+
+    #ROAD
+    pygame.draw.rect(screen, ROAD_COLOUR, (200, 0, 400, HEIGHT))
+
+    #LANE DIVIDER
+    for i in range(-1, 10):
+        pygame.draw.rect(screen, LINE_COLOUR,
+                         (WIDTH//2 - 5, lane_y + i * 80, 10, 40))
+
+    #CAR
+    pygame.draw.rect(screen, CAR_COLOUR, (car_x, car_y, car_width, car_height))
+
+    pygame.display.flip()
+    clock.tick(FPS)