Update app.py

app.py

@@ -1,34 +1,28 @@
 import streamlit as st
 import pandas as pd
+import numpy as np
 from modules.simulator import simulate_data
-from filters import apply_filters
-from modules.visuals import display_dashboard, display_charts, display_map_with_alerts
-from modules.ai_engine import AIEngine
+from modules.filters import apply_filters
+from modules.visuals import display_map
+from modules.alert_system import alert_system
 
-st.set_page_config(page_title="Vedavathi Smart Pole Monitoring", layout="wide")
-st.title("📡 Vedavathi Smart Pole Monitoring - Heatmap AI Simulator")
+st.set_page_config(page_title="Smart Pole Monitoring System", layout="wide")
 
-# Sidebar Controls
-st.sidebar.header("🛠️ Simulation Controls")
-num_poles = st.sidebar.slider("Number of Poles", min_value=10, max_value=500, value=50)
-simulate_faults = st.sidebar.checkbox("Simulate Random Faults", value=True)
+st.title("Smart Pole Monitoring System - Real-Time Dashboard")
 
-# Simulate Data
-df = simulate_data(num_poles, simulate_faults)
+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 Faults", value=True)
 
-# AI-based Health Prediction
-ai = AIEngine()
-df = ai.predict_health(df)
+df = simulate_data(num_poles, simulate_faults)
 
-# Sidebar Filters
-st.sidebar.header("📂 Filter Data")
+st.sidebar.header("Filters")
 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)
-df = apply_filters(df, alert_filter, cam_filter)
-
-# Dashboard UI
-display_dashboard(df)
-st.subheader("📍 Live Pole Map")
-display_map_with_alerts(df)
-st.subheader("📈 Energy + Sensor Charts")
-display_charts(df)
+
+filtered_df = apply_filters(df, alert_filter, cam_filter)
+
+st.subheader("📍 Heatmap of Poles")
+display_map(filtered_df)
+
+alert_system(filtered_df)