Update app.py

app.py

@@ -1,59 +1,40 @@
 import streamlit as st
+import folium
+from folium.plugins import HeatMap
 from modules.simulator import simulate_data
 from modules.filters import apply_filters
-from modules.visuals import display_dashboard, display_charts, display_map_heatmap
-from modules.scheduler import get_latest_data
-from datetime import datetime
+from modules.visuals import display_dashboard, display_charts, display_heatmap
 
-# Set page config as the first Streamlit command
+# Set the page configuration
 st.set_page_config(page_title="Vedavathi Smart Pole Monitoring", layout="wide")
+st.title("📡 Vedavathi Smart Pole Monitoring - Heatmap Dashboard")
 
-# Inject CSS for blinking red dots with enhanced animation
-st.markdown("""
-<style>
-.blinking-red {
-    animation: pulse 1s infinite;
-}
-@keyframes pulse {
-    0% { opacity: 1; transform: scale(1); }
-    50% { opacity: 0.3; transform: scale(1.3); }
-    100% { opacity: 1; transform: scale(1); }
-}
-</style>
-""", unsafe_allow_html=True)
-
-st.title("📡 Vedavathi Smart Pole Monitoring - PoC Simulator")
-
+# Sidebar - Controls for simulation
 st.sidebar.header("🛠️ Simulation Controls")
-num_poles = 50  # Fixed number of poles
-st.sidebar.write(f"Number of Poles: {num_poles}")
+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)
 
-# Get latest data from scheduler
-df, last_update = get_latest_data(num_poles, simulate_faults)
-st.write(f"Last Data Update: {last_update.strftime('%Y-%m-%d %H:%M:%S')} (Updates every 6 hours)")
+# Simulate data
+df = simulate_data(num_poles, simulate_faults)
 
+# Sidebar - Filters for the data
 st.sidebar.header("📂 Filter Data")
 alert_filter = st.sidebar.multiselect("Alert Level", ["Green", "Yellow", "Red"], default=["Green", "Yellow", "Red"])
 cam_filter = st.sidebar.selectbox("Camera Status", ["All", "Online", "Offline"], index=0)
-location_filter = st.sidebar.multiselect("Location", ["Hyderabad", "Gadwal", "Kurnool", "Ballari"], default=["Hyderabad", "Gadwal", "Kurnool", "Ballari"])
-
-filtered_df = apply_filters(df, alert_filter, cam_filter, location_filter)
-
-# Create tabs for each location
-tabs = st.tabs(["Hyderabad", "Gadwal", "Kurnool", "Ballari"])
-locations = ["Hyderabad", "Gadwal", "Kurnool", "Ballari"]
-
-for tab, location in zip(tabs, locations):
-    with tab:
-        location_df = filtered_df[filtered_df["Location"] == location]
-        if not location_df.empty:
-            display_dashboard(location_df, location)
-            st.subheader(f"📋 Pole Monitoring Table - {location}")
-            st.dataframe(location_df[["PoleID", "RFID", "Timestamp", "AlertLevel", "Anomalies", "CameraStatus", "SolarGen(kWh)", "WindGen(kWh)", "Tilt(°)", "Vibration(g)"]], use_container_width=True)
-            st.subheader(f"🗺️ Alert Map - {location}")
-            display_map_heatmap(location_df, location)
-            st.subheader(f"📊 Energy and Sensor Charts - {location}")
-            display_charts(location_df)
-        else:
-            st.warning(f"No poles found for {location} with the current filters.")
+
+# Apply filters on the simulated data
+filtered_df = apply_filters(df, alert_filter, cam_filter)
+
+# Display Dashboard Metrics
+display_dashboard(df)
+
+# Display the Heatmap
+st.subheader("📍 Pole Locations Heatmap")
+display_heatmap(filtered_df)
+
+# Display Pole Monitoring Table
+st.subheader("📋 Pole Monitoring Table")
+st.dataframe(filtered_df, use_container_width=True)
+
+# Display charts for data
+display_charts(df)