Update app.py

app.py

@@ -5,12 +5,12 @@ import pydeck as pdk
 from datetime import datetime, timedelta
 
 # ---- Constants ----
-TOTAL_POLES = 200  # 50 poles per site
+POLES_PER_SITE = 12
 SITES = {
     "Hyderabad": [17.385044, 78.486671],
     "Gadwal": [16.2351, 77.8052],
     "Kurnool": [15.8281, 78.0373],
-    "Bangalore": [12.9716, 77.5946]
+    "Ballari": [12.9716, 77.5946]
 }
 
 # ---- Helper Functions ----
@@ -67,7 +67,7 @@ selected_site = st.text_input("Enter site to view (Hyderabad, Gadwal, Kurnool, B
 
 if selected_site in SITES:
     with st.spinner(f"Simulating poles at {selected_site}..."):
-        poles_data = [simulate_pole(i + 1 + j * 50, site) for j, site in enumerate(SITES) for i in range(50)]
+        poles_data = [simulate_pole(i + 1, site) for site in SITES for i in range(POLES_PER_SITE)]
         df = pd.DataFrame(poles_data)
         site_df = df[df['Site'] == selected_site]
 
@@ -118,4 +118,4 @@ if selected_site in SITES:
         st.info("No red alerts at this time.")
 
 else:
-    st.warning("Invalid site. Please enter one of: Hyderabad, Gadwal, Kurnool, Bangalore")
+    st.warning("Invalid site. Please enter one of: Hyderabad, Gadwal, Kurnool, Ballari")