traffic_logic.py

def get_next_green(vehicle_counts):
    # Returns the road with the highest vehicle count
    return max(vehicle_counts, key=vehicle_counts.get)