Update app.py

app.py

@@ -2,14 +2,26 @@ import streamlit as st
 from modules.simulator import simulate_data
 from modules.filters import apply_filters
 from modules.visuals import display_dashboard, display_charts
+from modules.salesforce_connector import connect_salesforce, fetch_pole_data_from_salesforce
 
 st.set_page_config(page_title="Vedavathi Smart Pole Monitoring", layout="wide")
 st.title("📡 Vedavathi Smart Pole Monitoring - PoC Simulator")
 
 st.sidebar.header("🛠️ Simulation Controls")
+data_mode = st.sidebar.radio("Data Mode", ["Simulated", "Salesforce"])
 num_poles = st.sidebar.slider("Number of Poles", min_value=5, max_value=50, value=10)
 simulate_faults = st.sidebar.checkbox("Simulate Random Faults", value=True)
 
+if data_mode == "Simulated":
+    df = simulate_data(num_poles, simulate_faults)
+else:
+    try:
+        sf = connect_salesforce()
+        df = fetch_pole_data_from_salesforce(sf)
+        st.success("✅ Fetched data from Salesforce")
+    except Exception as e:
+        st.error(f"❌ Salesforce Error: {e}")
+        st.stop()
 df = simulate_data(num_poles, simulate_faults)
 
 st.sidebar.header("📂 Filter Data")