Update layout

app.py

@@ -187,6 +187,9 @@ st.write(
     unsafe_allow_html=True,
 )
 
+# Layout
+number_of_coloumns = 9
+
 # NUS Stops
 for stop_info in nus_bus_stops:
     shuttle_info = get_nus_bus_arrival(stop_info['name'])
@@ -208,10 +211,10 @@ for stop_info in nus_bus_stops:
     buses.sort(key=lambda x: x["eta"])
     
     with st.container(border=True):
-        cols = st.columns(7)
+        cols = st.columns(number_of_coloumns)
         cols[0].metric("NUS Stop", stop_info['name'])
         
-        for i, bus in enumerate(buses[:6]):
+        for i, bus in enumerate(buses[:number_of_coloumns-1]):
             cols[i+1].metric(bus['plate'], bus["shuttle_name"], str(bus["eta"]))
             
 # Public Bus
@@ -241,10 +244,10 @@ for stop_info in public_bus_stops:
     buses.sort(key=lambda x: x["eta"])
     
     with st.container(border=True):
-        cols = st.columns(7)
+        cols = st.columns(number_of_coloumns)
         cols[0].metric("Public Stop", stop_info['name'])
         
-        for i, bus in enumerate(buses[:6]):
+        for i, bus in enumerate(buses[:number_of_coloumns-1]):
             cols[i+1].metric(bus["type"], bus['service'], bus["eta"])
 
 # SMRT
@@ -261,8 +264,8 @@ for station in public_mrt_station:
     trains.sort(key=lambda x: x["eta"])
     
 with st.container(border=True):
-    cols = st.columns(7)
+    cols = st.columns(number_of_coloumns)
     cols[0].metric("MRT Station", station['name'])
     
-    for i, train in enumerate(trains[:6]):
+    for i, train in enumerate(trains[:number_of_coloumns-1]):
         cols[i+1].metric(train['direction'], 'CC', str(train['eta']))