display.py

import tkinter as tk
import numpy as np

branches = np.array([[True, False], [False, False], [False, False]])

def create_rectangles():
    rect_height = 100  # Height is 2x width
    padding = 10  # Space between rectangles
    for row in range(3):
        for col in range(2):
            x1 = col * (50 + padding) + padding
            y1 = row * (rect_height + padding) + padding
            x2 = x1 + 50
            y2 = y1 + rect_height

            if (branches[row][col] == True):
                canvas.create_rectangle(x1, y1, x2, y2, fill="green", outline="black")
            else:
                canvas.create_rectangle(x1, y1, x2, y2, fill="red", outline="black")


# Create main window
root = tk.Tk()
root.title("Tkinter Rectangle Grid")

# Canvas size
canvas_width = 2 * (50 + 10) + 10
canvas_height = 3 * (100 + 10) + 10
canvas = tk.Canvas(root, width=canvas_width, height=canvas_height, bg="white")
canvas.pack()

# Draw rectangles
create_rectangles()

# Run the Tkinter event loop
root.mainloop()