Update app.py

app.py

@@ -1,72 +1,22 @@
-import random
-import pandas as pd
-import streamlit as st
-import pydeck as pdk
-from datetime import datetime, timedelta
-import altair as alt
-
-# ---- Constants ----
-POLES_PER_SITE = 12
-POLE_SPACING_FEET = 10
-FEET_TO_LAT_DEG = 0.00003  # Rough conversion
-
-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": "Urban Grid"}
-}
-
-# ---- Custom Placement for Hyderabad ----
-def generate_hyderabad_pole_locations(base_lat, base_lon, num_poles):
+# ---- Fixed Placement for Each Site ----
+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 * FEET_TO_LAT_DEG, base_lon + i * FEET_TO_LAT_DEG]
-        for i in range(num_poles)
-    ]
-
-# ---- 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)
-    ]
-
-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)
-    power_required = round(random.uniform(4.0, 8.0), 2)
-    total_power = solar_kwh + wind_kwh
-    power_status = 'Sufficient' if total_power >= power_required else 'Insufficient'
-
-    vibration = round(random.uniform(0, 5), 2)
-    camera_status = random.choice(['Online', 'Offline'])
-
-    alert_level = 'Green'
-    if vibration > 3:
-        alert_level = 'Yellow'
-    if vibration > 4.5:
-        alert_level = 'Red'
-
-    health_score = max(0, 100 - (vibration * 10))
-    timestamp = datetime.now() - timedelta(hours=random.randint(0, 6))
-
-    return {
-        'Pole ID': f'{site_name[:3].upper()}-{pole_id:03}',
-        'Site': site_name,
-        'Zone': SITES[site_name]["zone"],
-        'Latitude': lat,
-        'Longitude': lon,
-        'Solar (kWh)': solar_kwh,
-        'Wind (kWh)': wind_kwh,
-        'Power Required (kWh)': power_required,
-        'Total Power (kWh)': total_power,
-        'Power Status': power_status,
-        'Vibration (g)': vibration,
-        'Camera Status': camera_status,
-        'Health Score': round(health_score, 2),
-        'Alert Level': alert_level,
-        'Last Checked': timestamp.strftime('%Y-%m-%d %H:%M:%S')
-    }
+        [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)
 
 # ---- Streamlit UI ----
 st.set_page_config(page_title="Smart Pole Monitoring", layout="wide")
@@ -83,12 +33,8 @@ if selected_site in SITES:
         for site_name, site_data in SITES.items():
             base_lat, base_lon = site_data["coords"]
 
-            if site_name == "Hyderabad":
-                # Linearly spaced poles
-                locations = generate_hyderabad_pole_locations(base_lat, base_lon, POLES_PER_SITE)
-            else:
-                # Randomly scattered poles
-                locations = [generate_location(base_lat, base_lon) for _ in range(POLES_PER_SITE)]
+            # 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)
@@ -173,7 +119,7 @@ if selected_site in SITES:
             zoom=12,
             pitch=50
         ),
-        layers=[
+        layers=[ 
             pdk.Layer(
                 'ScatterplotLayer',
                 data=fault_df,
@@ -183,7 +129,7 @@ if selected_site in SITES:
                 pickable=True,
             )
         ],
-        tooltip={
+        tooltip={ 
             "html": """
             <b>Pole ID:</b> {Pole ID}<br/>
             <b>Zone:</b> {Zone}<br/>