Update modules/simulator.py

modules/simulator.py

@@ -1,19 +1,50 @@
 import pandas as pd
 import numpy as np
 
-def simulate_pole_data(n=100):
-    np.random.seed(1)
-    sites = ["Hyderabad", "Ballari", "Gadwal", "Kurnool"]
-    df = pd.DataFrame({
-        "PoleID": [f"Pole_{i:05}" for i in range(n)],
-        "Site": np.random.choice(sites, size=n),
-        "Location_Latitude": np.random.uniform(16.5, 18.5, n),
-        "Location_Longitude": np.random.uniform(77.0, 79.5, n),
-        "SolarGen(kWh)": np.random.uniform(2.0, 6.5, n),
-        "WindGen(kWh)": np.random.uniform(0.3, 2.5, n),
-        "Tilt(°)": np.random.uniform(0, 20, n),
-        "Vibration(g)": np.random.uniform(0.1, 3.5, n),
-        "CameraStatus": np.random.choice(["Online", "Offline"], n),
-        "PowerSufficient": np.random.choice(["Yes", "No"], n),
-    })
-    return df
+def simulate_data(n=50, faults=True):
+    np.random.seed(42)
+    sites = {
+        "Hyderabad": (17.385, 78.4867),
+        "Ballari": (15.1394, 76.9214),
+        "Gadwal": (16.232, 77.795),
+        "Kurnool": (15.8281, 78.0373)
+    }
+    data = []
+    for i in range(n):
+        site = np.random.choice(list(sites.keys()))
+        lat, lon = sites[site]
+        solar = np.round(np.random.uniform(3, 7.5), 2)
+        wind = np.round(np.random.uniform(0.5, 2.0), 2)
+        tilt = np.round(np.random.uniform(0, 15), 1)
+        vib = np.round(np.random.uniform(0.1, 3.0), 2)
+        cam = np.random.choice(["Online", "Offline"], p=[0.85, 0.15])
+        power_required = np.round(np.random.uniform(1.0, 1.5), 2)
+        total = solar + wind
+        sufficient = "Yes" if total >= power_required else "No"
+
+        anomaly = []
+        if faults:
+            if solar < 4.0: anomaly.append("Low Solar")
+            if wind < 0.7: anomaly.append("Low Wind")
+            if tilt > 10: anomaly.append("Tilt Risk")
+            if vib > 2.0: anomaly.append("Vibration Alert")
+            if cam == "Offline": anomaly.append("Camera Offline")
+            if sufficient == "No": anomaly.append("Power Insufficient")
+
+        alert = "Green" if len(anomaly) == 0 else "Yellow" if len(anomaly) == 1 else "Red"
+
+        data.append({
+            "Pole ID": f"Pole_{i+1:03}",
+            "Site": site,
+            "Latitude": lat + np.random.uniform(-0.01, 0.01),
+            "Longitude": lon + np.random.uniform(-0.01, 0.01),
+            "SolarGen(kWh)": solar,
+            "WindGen(kWh)": wind,
+            "Tilt(°)": tilt,
+            "Vibration(g)": vib,
+            "Camera Status": cam,
+            "Power Sufficient": sufficient,
+            "Anomalies": "; ".join(anomaly) if anomaly else "None",
+            "Alert Level": alert
+        })
+    return pd.DataFrame(data)