Update app.py

app.py

@@ -1,40 +1,30 @@
 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_heatmap
+from modules.visuals import display_dashboard, display_heatmap
+from modules.ai_engine import AIEngine
 
-# Set the page configuration
-st.set_page_config(page_title="Vedavathi Smart Pole Monitoring", layout="wide")
-st.title("📡 Vedavathi Smart Pole Monitoring - Heatmap Dashboard")
+st.set_page_config(page_title="Pole Monitoring Dashboard", layout="wide")
+st.title("📡 Pole Monitoring - Real-Time Simulation")
 
-# Sidebar - Controls for simulation
 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)
 
-# Simulate data
+# Simulate pole data
 df = simulate_data(num_poles, simulate_faults)
 
-# Sidebar - Filters for the data
+# AI predictions (optional)
+ai_engine = AIEngine()
+df = ai_engine.predict_health(df)
+
 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)
-
-# Apply filters on the simulated data
-filtered_df = apply_filters(df, alert_filter, cam_filter)
+filtered_df = apply_filters(df, alert_filter)
 
-# Display Dashboard Metrics
-display_dashboard(df)
-
-# Display the Heatmap
-st.subheader("📍 Pole Locations Heatmap")
+# Dashboard and Heatmap Display
+display_dashboard(filtered_df)
 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)