Update app.py

app.py

@@ -1,38 +1,45 @@
-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")
-alert_filter = st.sidebar.multiselect("Alert Level", ["Green", "Yellow", "Red"], default=["Green", "Yellow", "Red"])
-cam_filter = st.sidebar.selectbox("Camera Status", ["All", "Online", "Offline"], index=0)
-
-filtered_df = apply_filters(df, alert_filter, cam_filter)
-
-display_dashboard(df)
-
-st.subheader("📋 Pole Monitoring Table")
-st.dataframe(filtered_df, use_container_width=True)
-
-display_charts(df)
+from flask import Flask, jsonify
+import requests
+import urllib.parse
+
+app = Flask(__name__)
+
+@app.route('/get-salesforce-data')
+def get_salesforce_data():
+    access_token = 'YOUR_ACCESS_TOKEN'
+    instance_url = 'https://vedavathi3-dev-ed.my.salesforce.com'
+
+    query = """
+    SELECT 
+        Name, 
+        Solar_Generation__c, 
+        Wind_Generation__c, 
+        Power_Required__c, 
+        Camera_Status__c, 
+        Tilt__c, 
+        Vibration__c, 
+        Alert_Level__c, 
+        Power_Sufficient__c
+    FROM 
+        Pole_Monitoring__c
+    LIMIT 50
+    """
+
+    encoded_query = urllib.parse.quote(query)
+    url = f"{instance_url}/services/data/v59.0/query/?q={encoded_query}"
+
+    headers = {
+        'Authorization': f'Bearer {access_token}',
+        'Content-Type': 'application/json'
+    }
+
+    response = requests.get(url, headers=headers)
+
+    if response.status_code == 200:
+        data = response.json()
+        return jsonify(data['records'])
+    else:
+        return jsonify({'error': response.text}), response.status_code
+
+if __name__ == '__main__':
+    app.run(debug=True)