Update app.py

app.py

@@ -1,16 +1,16 @@
 import streamlit as st
 from modules.simulator import simulate_data
 from modules.filters import apply_filters
-from modules.visuals import display_dashboard, display_charts
+from modules.visuals import display_dashboard, display_charts, display_heatmap
 
 st.set_page_config(page_title="Vedavathi Smart Pole Monitoring", layout="wide")
 st.title("📡 Vedavathi Smart Pole Monitoring - PoC Simulator")
 
 st.sidebar.header("🛠️ Simulation Controls")
-num_poles = st.sidebar.slider("Number of Poles", min_value=5, max_value=50, value=10)
 simulate_faults = st.sidebar.checkbox("Simulate Random Faults", value=True)
 
-df = simulate_data(num_poles, simulate_faults)
+# Always simulate 4 sites × 12 poles = 48
+df = simulate_data(n=48, faults=simulate_faults)
 
 st.sidebar.header("📂 Filter Data")
 alert_filter = st.sidebar.multiselect("Alert Level", ["Green", "Yellow", "Red"], default=["Green", "Yellow", "Red"])
@@ -23,4 +23,6 @@ display_dashboard(df)
 st.subheader("📋 Pole Monitoring Table")
 st.dataframe(filtered_df, use_container_width=True)
 
+display_heatmap(df)
 display_charts(df)
+