import streamlit as st
from traffic_logic import get_next_green

st.set_page_config(page_title="Smart Traffic Light System", layout="wide")
st.title("🚦 Smart Traffic Light Simulation")

st.sidebar.header("Vehicle Count per Road")
north = st.sidebar.slider("North Road", 0, 50, 10)
east = st.sidebar.slider("East Road", 0, 50, 20)
south = st.sidebar.slider("South Road", 0, 50, 5)
west = st.sidebar.slider("West Road", 0, 50, 15)

vehicle_counts = {'North': north, 'East': east, 'South': south, 'West': west}
next_green = get_next_green(vehicle_counts)

st.subheader("Traffic Light Status")

cols = st.columns(4)
for i, (road, count) in enumerate(vehicle_counts.items()):
    light_color = "🟢 Green" if road == next_green else "🔴 Red"
    cols[i].metric(label=f"{road} Road", value=f"{count} vehicles", delta=light_color)