Нарисовать оранжевые ромбики на желтом фоне

import pygame
 
n = int(input())
# n = 50
start_points = [(n * 1, n * 0), (n * 2, n * 1), (n * 1, n * 2), (n * 0, n * 1)]
list_points = []
 
pygame.init()
w = 300
h = 300
a = h // (n * 2)
screen = pygame.display.set_mode((300, 300))
clock = pygame.time.Clock()
yellow = pygame.Color('yellow')
orange = pygame.Color('orange')
count_y = 0
flag_y = True
 
while flag_y:
    a -= 1
    flag_x = True
    count_x = -2
    while flag_x:
        count_x += 2
        if n * (2 + count_x) <= w:
            point = [(n * (1 + count_x), n * (0 + count_y)), (n * (2 + count_x), n * (1 + count_y)),
                     (n * (1 + count_x), n * (2 + count_y)), (n * (0 + count_x), n * (1 + count_y))]
            list_points.append(point)
        else:
            flag_x = False
    if a > 0:
        count_y += 2
    else:
        flag_y = False
 
 
running = True
while running:
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            running = False
 
    screen.fill(yellow)
    for i in range(len(list_points)):
        pygame.draw.polygon(screen, orange, list_points[i])
    pygame.display.flip()
    clock.tick(60)

import pygame
 
def draw_diamond(surface, color, x, y, size):
    pygame.draw.polygon(surface, color, [[x + size / 2, y], [x + size, y + size / 2], [x + size / 2, y + size], [x, y + size / 2]])
 
try:
    n = int(input("Enter size: "))
except ValueError:
    print("Invalid input format")
    quit()
 
size = (300, 300)
screen = pygame.display.set_mode(size)
pygame.display.set_caption("Orange Diamonds")
 
running = True
while running:
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            running = False
 
    screen.fill(pygame.Color("yellow"))
    draw_diamond(screen, pygame.Color("orange"), size[0] // 2 - n // 2, size[1] // 2 - n // 2, n)
    draw_diamond(screen, pygame.Color("orange"), size[0] // 2 + n // 2, size[1] // 2 - n // 2, n)
    draw_diamond(screen, pygame.Color("orange"), size[0] // 2 - n // 2, size[1] // 2 + n // 2, n)
    draw_diamond(screen, pygame.Color("orange"), size[0] // 2 + n // 2, size[1] // 2 + n // 2, n)
    
    pygame.display.update()
 
pygame.quit()

import pygame
 
 
try:
    n = input()
    if int(n) != float(n):
        raise ValueError
 
    n = int(n)
    pygame.init()
    size = width, height = 300, 300
    screen = pygame.display.set_mode((width, height))
    screen.fill("yellow")
    s = n // 2
    color = pygame.Color("orange")
    for i in range(0, 300, s * 2):
        for j in range(0, 300, s * 2):
            if i + s * 2 > width or j + s * 2 > height:
                continue
 
            points = [(i, j + s), (i + s, j), (i + s * 2, j + s), (i + s, j + s * 2)]
            pygame.draw.polygon(screen, color, points)
 
    pygame.display.flip()
    while pygame.event.wait().type != pygame.QUIT:
        pass
 
    pygame.quit()
except ValueError:
    print("Неправильный формат ввода")