random 4 letter code generator

import random
import string

def generate_code():
    letters = string.ascii_uppercase
    code = ''.join(random.choice(letters) for _ in range(4))
    return code

# Generate a 4-letter code
code = generate_code()
print(code)