Update app.py

app.py

@@ -3,6 +3,7 @@ import pandas as pd
 import streamlit as st
 import pydeck as pdk
 from datetime import datetime, timedelta
+from salesforce_integration import fetch_salesforce_data  # Import the Salesforce integration
 
 # ---- Constants ----
 POLES_PER_SITE = 12
@@ -20,7 +21,7 @@ def generate_location(base_lat, base_lon):
         base_lon + random.uniform(-0.02, 0.02)
     ]
 
-def simulate_pole(pole_id, site_name):
+def simulate_pole(pole_id, site_name, salesforce_data=None):
     lat, lon = generate_location(*SITES[site_name])
     solar_kwh = round(random.uniform(3.0, 7.5), 2)
     wind_kwh = round(random.uniform(0.5, 2.0), 2)
@@ -44,6 +45,20 @@ def simulate_pole(pole_id, site_name):
     health_score = max(0, 100 - (tilt_angle + vibration * 10))
     timestamp = datetime.now() - timedelta(hours=random.randint(0, 6))
 
+    # If salesforce data exists, prioritize it over simulation
+    if salesforce_data:
+        # Merge or override simulated data with Salesforce data
+        for pole_data in salesforce_data:
+            if pole_data['Site'] == site_name and pole_data['Pole ID'] == f'{site_name[:3].upper()}-{pole_id:03}':
+                solar_kwh = pole_data.get('Solar (kWh)', solar_kwh)
+                wind_kwh = pole_data.get('Wind (kWh)', wind_kwh)
+                power_required = pole_data.get('Power Required (kWh)', power_required)
+                total_power = solar_kwh + wind_kwh
+                power_status = 'Sufficient' if total_power >= power_required else 'Insufficient'
+                health_score = round(pole_data.get('Health Score', health_score), 2)
+                alert_level = pole_data.get('Alert Level', alert_level)
+                break
+
     return {
         'Pole ID': f'{site_name[:3].upper()}-{pole_id:03}',
         'Site': site_name,
@@ -70,8 +85,11 @@ st.title("🌍 Smart Renewable Pole Monitoring - Multi-Site")
 selected_site = st.text_input("Enter site to view (Hyderabad, Gadwal, Kurnool, Ballari):", "Hyderabad")
 
 if selected_site in SITES:
+    # Fetch Salesforce data
+    salesforce_data = fetch_salesforce_data(selected_site)
+    
     with st.spinner(f"Simulating poles at {selected_site}..."):
-        poles_data = [simulate_pole(i + 1, site) for site in SITES for i in range(POLES_PER_SITE)]
+        poles_data = [simulate_pole(i + 1, selected_site, salesforce_data) for i in range(POLES_PER_SITE)]
         df = pd.DataFrame(poles_data)
         site_df = df[df['Site'] == selected_site]