Update modules/simulator.py

modules/simulator.py

@@ -6,30 +6,14 @@ def simulate_data(n=10, faults=True):
     today = datetime.date.today()
     poles = [f"Pole_{i+1:03}" for i in range(n)]
     data = []
-
-    # Define location coordinates for the sites
-    locations = {
-        "Hyderabad": (17.385044, 78.486671),
-        "Gadwal": (16.2333, 77.1833),
-        "Kurnool": (15.8281, 78.0469),
-        "Ballari": (15.1502, 75.9246)
-    }
-
-    # Assign poles to sites randomly
     for pole in poles:
-        site = np.random.choice(list(locations.keys()))
-        lat, lon = locations[site]
-
         solar = round(np.random.uniform(3.0, 7.5), 2)
         wind = round(np.random.uniform(0.5, 2.0), 2)
-        required = round(np.random.uniform(1.0, 1.5), 2)
-        total = solar + wind
-        cam = np.random.choice(['Online', 'Offline'], p=[0.85, 0.15]) if faults else "Online"
         tilt = round(np.random.uniform(0, 12), 1)
         vib = round(np.random.uniform(0.1, 2.5), 2)
-        sufficient = "Yes" if total >= required else "No"
+        cam = np.random.choice(['Online', 'Offline'], p=[0.85, 0.15])
+        alert = "Green"
         anomaly = []
-
         if faults:
             if solar < 4.0:
                 anomaly.append("Low Solar Output")
@@ -41,29 +25,17 @@ def simulate_data(n=10, faults=True):
                 anomaly.append("Vibration Alert")
             if cam == "Offline":
                 anomaly.append("Camera Offline")
-            if sufficient == "No":
-                anomaly.append("Power Insufficient")
-
-        alert = "Green"
-        if len(anomaly) == 1:
-            alert = "Yellow"
-        elif len(anomaly) > 1:
-            alert = "Red"
-
+            alert = "Red" if len(anomaly) > 1 else "Yellow" if anomaly else "Green"
+        
         data.append({
             "Pole ID": pole,
             "Date": today,
             "Solar Gen (kWh)": solar,
             "Wind Gen (kWh)": wind,
-            "Power Required (kWh)": required,
-            "Power Sufficient": sufficient,
-            "Camera Status": cam,
             "Tilt (°)": tilt,
             "Vibration (g)": vib,
-            "Anomalies": "; ".join(anomaly) if anomaly else "None",
-            "Alert Level": alert,
-            "Latitude": lat,
-            "Longitude": lon
+            "Camera Status": cam,
+            "Anomalies": "; ".join(anomaly),
+            "Alert Level": alert
         })
-
     return pd.DataFrame(data)