Create traffic_logic.py

src/traffic_logic.py

@@ -0,0 +1,3 @@
+def get_next_green(vehicle_counts):
+    # Returns the road with the highest vehicle count
+    return max(vehicle_counts, key=vehicle_counts.get)