Code Structure

Interface/__init__.py

@@ -0,0 +1 @@
+from .interface import *

Interface/helpers.py

@@ -3,14 +3,18 @@ import pygame
 BLACK = 0, 0, 0
 WHITE = 255, 255, 255
 
+
 def rect(screen, color, position_x, position_y, size_x, size_y):
-  pygame.draw.rect(screen, color, (position_x,position_y,size_x,size_y))
+    pygame.draw.rect(screen, color, (position_x, position_y, size_x, size_y))
+
 
 def line(screen, color, x_1, y_1, x_2, y_2):
-  pygame.draw.line(screen, color, (x_1, y_1), (x_2, y_2))
+    pygame.draw.line(screen, color, (x_1, y_1), (x_2, y_2))
+
 
 def create_screen(size):
-  return pygame.display.set_mode(size)
+    return pygame.display.set_mode(size)
+
 
 def draw_board(screen, color, width, height, tiles_x, tiles_y):
     tiles_width = width / tiles_x
@@ -25,4 +29,3 @@ def draw_board(screen, color, width, height, tiles_x, tiles_y):
     for i in range(tiles_y-1):
         line(screen, color, 0, temp_y, width, temp_y)
         temp_y += tiles_height
-

Interface/interface.py

@@ -0,0 +1,21 @@
+import sys
+import pygame
+from .helpers import *
+
+
+def initialize_interface(tiles_x, tiles_y):
+    size = width, height = 800, 800
+    screen = create_screen(size)
+    pygame.display.set_caption('Epic AI Vacuum Cleaner')
+
+    while True:
+        for event in pygame.event.get():
+            if event.type == pygame.QUIT:
+                sys.exit()
+
+        screen.fill(BLACK)
+        # rect(screen, WHITE, 50,50,90,90)
+
+        draw_board(screen, WHITE, width, height, tiles_x, tiles_y)
+
+        pygame.display.flip()

Interface/main.py

@@ -1,23 +0,0 @@
-import sys, pygame
-from helpers import *
-pygame.init()
-
-size = width, height = 1000, 1000
-screen = create_screen(size)
-
-tiles_x, tiles_y = 10, 10
-
-
-while True:
-    for event in pygame.event.get():
-        if event.type == pygame.QUIT: sys.exit()
-
-
-    screen.fill(BLACK)
-    # rect(screen, WHITE, 50,50,90,90)
-    
-    draw_board(screen, WHITE, width, height, tiles_x, tiles_y)
-
-
-
-    pygame.display.flip()

main.py

@@ -0,0 +1,5 @@
+import pygame
+from Interface import initialize_interface
+
+pygame.init()
+initialize_interface(10, 10)