category = input("Введите категорию водительских прав (A, B, C, D): ")
def transport(category):
vehicles = ""
if category == "A":
vehicles = "мотоциклы"
elif category == "B":
vehicles = "легковые автомобили и мотоциклы"
elif category == "C":
vehicles = "грузовики и автобусы"
elif category == "D":
vehicles = "автобусы"
else:
vehicles = "Введена некорректная категория"
return vehicles
result = transport(category)
print(result)

Пример работы программы:

Введите категорию водительских прав (A, B, C, D): B
легковые автомобили и мотоциклы