Update app.py

app.py

@@ -4,46 +4,37 @@ import streamlit as st
 import pydeck as pdk
 from datetime import datetime, timedelta
 import altair as alt
+import time
+
+# ---- Auto-refresh trigger every 1 hour ----
+st.experimental_set_query_params(updated=int(time.time() // 3600))
 
 # ---- Constants ----
 POLES_PER_SITE = 12
+POLE_GRID_ROWS = 3
+POLE_GRID_COLS = 4
 POLE_SPACING_FEET = 10
-FEET_TO_LAT_DEG = 0.00003  # Rough conversion
+FEET_TO_LAT_DEG = 0.00003  # Approximate conversion
+FEET_TO_LON_DEG = 0.00003  # Can be tuned per site longitude
 
 SITES = {
     "Hyderabad": {"coords": [17.385044, 78.486671], "zone": "Dairy Farm Zone"},
     "Gadwal": {"coords": [16.2351, 77.8052], "zone": "City Center"},
     "Kurnool": {"coords": [15.8281, 78.0373], "zone": "West Industrial"},
-    "Ballari": {"coords": [12.9716, 77.5946], "zone": "Solar Power station"}
+    "Ballari": {"coords": [12.9716, 77.5946], "zone": "Urban Grid"}
 }
 
-# ---- Fixed Placement for Each Site ----
+# ---- Fixed Pole Grid Placement ----
 def generate_fixed_pole_locations(base_lat, base_lon, num_poles):
-    # Define a fixed area for pole placement (e.g., 0.01 degrees in latitude and longitude)
-    area_width = 0.01  # 0.01 degree latitude distance (approx. 1.1 km)
-    area_height = 0.01  # 0.01 degree longitude distance (approx. 1.1 km)
-    
-    # Calculate number of rows and columns for the grid
-    rows = 3  # Number of rows in the grid
-    cols = 4  # Number of columns in the grid
-    
-    # Calculate the spacing in both directions
-    lat_spacing = area_height / rows
-    lon_spacing = area_width / cols
-
-    # Generate the fixed grid of pole locations
-    return [
-        [base_lat + i * lat_spacing, base_lon + j * lon_spacing]
-        for i in range(rows) for j in range(cols)
-    ][:num_poles]  # Only take num_poles (12 in this case)
-
-# ---- Helper Functions ----
-def generate_location(base_lat, base_lon):
-    return [
-        base_lat + random.uniform(-0.02, 0.02),
-        base_lon + random.uniform(-0.02, 0.02)
-    ]
-
+    locations = []
+    for i in range(POLE_GRID_ROWS):
+        for j in range(POLE_GRID_COLS):
+            lat = base_lat + i * POLE_SPACING_FEET * FEET_TO_LAT_DEG
+            lon = base_lon + j * POLE_SPACING_FEET * FEET_TO_LON_DEG
+            locations.append([lat, lon])
+    return locations[:num_poles]
+
+# ---- Simulate a Single Pole ----
 def simulate_pole(pole_id, site_name, lat, lon):
     solar_kwh = round(random.uniform(3.0, 7.5), 2)
     wind_kwh = round(random.uniform(0.5, 2.0), 2)
@@ -61,7 +52,7 @@ def simulate_pole(pole_id, site_name, lat, lon):
         alert_level = 'Red'
 
     health_score = max(0, 100 - (vibration * 10))
-    timestamp = datetime.now() - timedelta(hours=random.randint(0, 6))
+    timestamp = datetime.now() - timedelta(minutes=random.randint(0, 59))
 
     return {
         'Pole ID': f'{site_name[:3].upper()}-{pole_id:03}',
@@ -81,134 +72,134 @@ def simulate_pole(pole_id, site_name, lat, lon):
         'Last Checked': timestamp.strftime('%Y-%m-%d %H:%M:%S')
     }
 
+# ---- Cache Data for 1 Hour ----
+@st.cache_data(ttl=3600)
+def get_simulated_data():
+    all_data = []
+    for site_name, site_data in SITES.items():
+        base_lat, base_lon = site_data["coords"]
+        locations = generate_fixed_pole_locations(base_lat, base_lon, POLES_PER_SITE)
+
+        for i, (lat, lon) in enumerate(locations):
+            pole_data = simulate_pole(i + 1, site_name, lat, lon)
+            all_data.append(pole_data)
+    return pd.DataFrame(all_data)
+
 # ---- Streamlit UI ----
 st.set_page_config(page_title="Smart Pole Monitoring", layout="wide")
 st.title("🌍 Smart Renewable Pole Monitoring - Multi-Site")
 
 selected_site = st.selectbox("Select a site to view:", options=list(SITES.keys()), index=0)
 
-if selected_site in SITES:
-    st.markdown(f"**Zone:** {SITES[selected_site]['zone']}")
-
-    with st.spinner(f"Simulating poles at {selected_site}..."):
-        all_data = []
-
-        for site_name, site_data in SITES.items():
-            base_lat, base_lon = site_data["coords"]
-
-            # Fixed placement for all poles
-            locations = generate_fixed_pole_locations(base_lat, base_lon, POLES_PER_SITE)
-
-            for i, (lat, lon) in enumerate(locations):
-                pole_data = simulate_pole(i + 1, site_name, lat, lon)
-                all_data.append(pole_data)
-
-        df = pd.DataFrame(all_data)
-        site_df = df[df['Site'] == selected_site]
-
-    # Summary
-    col1, col2, col3 = st.columns(3)
-    col1.metric("Total Poles", site_df.shape[0])
-    col2.metric("Red Alerts", site_df[site_df['Alert Level'] == 'Red'].shape[0])
-    col3.metric("Power Insufficiencies", site_df[site_df['Power Status'] == 'Insufficient'].shape[0])
-
-    # Table
-    st.subheader(f"📋 Pole Data Table for {selected_site}")
-    with st.expander("Filter Options"):
-        alert_filter = st.multiselect("Alert Level", options=site_df['Alert Level'].unique(), default=site_df['Alert Level'].unique())
-        camera_filter = st.multiselect("Camera Status", options=site_df['Camera Status'].unique(), default=site_df['Camera Status'].unique())
-
-    filtered_df = site_df[
-        (site_df['Alert Level'].isin(alert_filter)) & 
-        (site_df['Camera Status'].isin(camera_filter))
-    ]
-    st.dataframe(filtered_df, use_container_width=True)
-
-    # ---- Energy Chart ----
-    st.subheader("🔋 Energy Generation per Pole")
-
-    energy_long_df = site_df[['Pole ID', 'Solar (kWh)', 'Wind (kWh)']].melt(
-        id_vars='Pole ID',
-        value_vars=['Solar (kWh)', 'Wind (kWh)'],
-        var_name='Energy Source',
-        value_name='kWh'
+st.caption(f"🔄 Data last refreshed: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}")
+
+with st.spinner(f"Loading data for {selected_site}..."):
+    df = get_simulated_data()
+    site_df = df[df['Site'] == selected_site]
+
+# ---- Summary Metrics ----
+col1, col2, col3 = st.columns(3)
+col1.metric("Total Poles", site_df.shape[0])
+col2.metric("Red Alerts", site_df[site_df['Alert Level'] == 'Red'].shape[0])
+col3.metric("Power Insufficiencies", site_df[site_df['Power Status'] == 'Insufficient'].shape[0])
+
+# ---- Table & Filters ----
+st.subheader(f"📋 Pole Data Table for {selected_site}")
+with st.expander("Filter Options"):
+    alert_filter = st.multiselect("Alert Level", options=site_df['Alert Level'].unique(), default=site_df['Alert Level'].unique())
+    camera_filter = st.multiselect("Camera Status", options=site_df['Camera Status'].unique(), default=site_df['Camera Status'].unique())
+
+filtered_df = site_df[
+    (site_df['Alert Level'].isin(alert_filter)) & 
+    (site_df['Camera Status'].isin(camera_filter))
+]
+st.dataframe(filtered_df, use_container_width=True)
+
+# ---- Energy Chart ----
+st.subheader("🔋 Energy Generation per Pole")
+
+energy_long_df = site_df[['Pole ID', 'Solar (kWh)', 'Wind (kWh)']].melt(
+    id_vars='Pole ID',
+    value_vars=['Solar (kWh)', 'Wind (kWh)'],
+    var_name='Energy Source',
+    value_name='kWh'
+)
+
+bar_chart = alt.Chart(energy_long_df).mark_bar().encode(
+    x=alt.X('Pole ID:N', sort=None, title='Pole ID'),
+    y=alt.Y('kWh:Q'),
+    color='Energy Source:N',
+    tooltip=['Pole ID', 'Energy Source', 'kWh']
+).properties(
+    width=800,
+    height=400
+).configure_axisX(labelAngle=45)
+
+st.altair_chart(bar_chart, use_container_width=True)
+
+# ---- Fault Filter ----
+st.subheader("⚠️ Map Filter: Select Fault Type(s)")
+
+fault_options = ['High Vibration (>3g)', 'Camera Offline', 'Power Insufficient']
+selected_faults = st.multiselect("Show poles with these fault conditions:", options=fault_options, default=fault_options)
+
+def fault_condition(row):
+    return (
+        ('High Vibration (>3g)' in selected_faults and row['Vibration (g)'] > 3) or
+        ('Camera Offline' in selected_faults and row['Camera Status'] == 'Offline') or
+        ('Power Insufficient' in selected_faults and row['Power Status'] == 'Insufficient')
     )
 
-    bar_chart = alt.Chart(energy_long_df).mark_bar().encode(
-        x=alt.X('Pole ID:N', sort=None, title='Pole ID'),
-        y=alt.Y('kWh:Q'),
-        color='Energy Source:N',
-        tooltip=['Pole ID', 'Energy Source', 'kWh']
-    ).properties(
-        width=800,
-        height=400
-    ).configure_axisX(labelAngle=45)
-
-    st.altair_chart(bar_chart, use_container_width=True)
-
-    # ---- Fault Type Filter ----
-    st.subheader("⚠️ Map Filter: Select Fault Type(s)")
-
-    fault_options = ['High Vibration (>3g)', 'Camera Offline', 'Power Insufficient']
-    selected_faults = st.multiselect("Show poles with these fault conditions:", options=fault_options, default=fault_options)
-
-    def fault_condition(row):
-        return (
-            ('High Vibration (>3g)' in selected_faults and row['Vibration (g)'] > 3) or
-            ('Camera Offline' in selected_faults and row['Camera Status'] == 'Offline') or
-            ('Power Insufficient' in selected_faults and row['Power Status'] == 'Insufficient')
+fault_df = site_df[site_df.apply(fault_condition, axis=1)] if selected_faults else site_df
+
+# ---- Color Mapping ----
+def get_color(alert):
+    if alert == 'Green':
+        return [0, 255, 0, 160]
+    elif alert == 'Yellow':
+        return [255, 255, 0, 160]
+    elif alert == 'Red':
+        return [255, 0, 0, 160]
+    return [128, 128, 128, 160]
+
+fault_df['color'] = fault_df['Alert Level'].apply(get_color)
+
+# ---- Map Visualization ----
+st.subheader("📍 Pole Locations with Selected Faults")
+st.pydeck_chart(pdk.Deck(
+    initial_view_state=pdk.ViewState(
+        latitude=SITES[selected_site]["coords"][0],
+        longitude=SITES[selected_site]["coords"][1],
+        zoom=12,
+        pitch=50
+    ),
+    layers=[
+        pdk.Layer(
+            'ScatterplotLayer',
+            data=fault_df,
+            get_position='[Longitude, Latitude]',
+            get_color='color',
+            get_radius=100,
+            pickable=True,
         )
-
-    fault_df = site_df[site_df.apply(fault_condition, axis=1)] if selected_faults else site_df
-
-    # ---- Map Color Logic ----
-    def get_color(alert):
-        if alert == 'Green':
-            return [0, 255, 0, 160]
-        elif alert == 'Yellow':
-            return [255, 255, 0, 160]
-        elif alert == 'Red':
-            return [255, 0, 0, 160]
-        return [128, 128, 128, 160]
-
-    fault_df['color'] = fault_df['Alert Level'].apply(get_color)
-
-    # ---- Map ----
-    st.subheader("📍 Pole Locations with Selected Faults")
-    st.pydeck_chart(pdk.Deck(
-        initial_view_state=pdk.ViewState(
-            latitude=SITES[selected_site]["coords"][0],
-            longitude=SITES[selected_site]["coords"][1],
-            zoom=12,
-            pitch=50
-        ),
-        layers=[ 
-            pdk.Layer(
-                'ScatterplotLayer',
-                data=fault_df,
-                get_position='[Longitude, Latitude]',
-                get_color='color',
-                get_radius=100,
-                pickable=True,
-            )
-        ],
-        tooltip={ 
-            "html": """
-            <b>Pole ID:</b> {Pole ID}<br/>
-            <b>Zone:</b> {Zone}<br/>
-            <b>Alert Level:</b> {Alert Level}<br/>
-            <b>Health Score:</b> {Health Score}<br/>
-            <b>Power Status:</b> {Power Status}<br/>
-            <b>Vibration (g):</b> {Vibration (g)}<br/>
-            <b>Camera:</b> {Camera Status}<br/>
-            <b>Solar (kWh):</b> {Solar (kWh)}<br/>
-            <b>Wind (kWh):</b> {Wind (kWh)}<br/>
-            <b>Last Checked:</b> {Last Checked}
-            """,
-            "style": {
-                "backgroundColor": "steelblue",
-                "color": "white",
-                "fontSize": "12px"
-            }
+    ],
+    tooltip={
+        "html": """
+        <b>Pole ID:</b> {Pole ID}<br/>
+        <b>Zone:</b> {Zone}<br/>
+        <b>Alert Level:</b> {Alert Level}<br/>
+        <b>Health Score:</b> {Health Score}<br/>
+        <b>Power Status:</b> {Power Status}<br/>
+        <b>Vibration (g):</b> {Vibration (g)}<br/>
+        <b>Camera:</b> {Camera Status}<br/>
+        <b>Solar (kWh):</b> {Solar (kWh)}<br/>
+        <b>Wind (kWh):</b> {Wind (kWh)}<br/>
+        <b>Last Checked:</b> {Last Checked}
+        """,
+        "style": {
+            "backgroundColor": "steelblue",
+            "color": "white",
+            "fontSize": "12px"
         }
-    ))
+    }
+))