Update modules/simulator.py

modules/simulator.py

@@ -2,11 +2,14 @@ import pandas as pd
 import numpy as np
 import datetime
 
-def simulate_data(n=10, faults=True):
+def simulate_data(n=48, faults=True):
     today = datetime.date.today()
+    sites = ["Hyderabad", "Gadwal", "Kurnool", "Ballari"]
     poles = [f"Pole_{i+1:03}" for i in range(n)]
     data = []
-    for pole in poles:
+
+    for i, pole in enumerate(poles):
+        site = sites[i // 12]  # 12 poles per 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)
@@ -16,26 +19,24 @@ def simulate_data(n=10, faults=True):
         vib = round(np.random.uniform(0.1, 2.5), 2)
         sufficient = "Yes" if total >= required else "No"
         anomaly = []
+
         if faults:
-            if solar < 4.0:
-                anomaly.append("Low Solar Output")
-            if wind < 0.7:
-                anomaly.append("Low Wind Output")
-            if tilt > 10:
-                anomaly.append("Pole Tilt Risk")
-            if vib > 2.0:
-                anomaly.append("Vibration Alert")
-            if cam == "Offline":
-                anomaly.append("Camera Offline")
-            if sufficient == "No":
-                anomaly.append("Power Insufficient")
+            if solar < 4.0: anomaly.append("Low Solar Output")
+            if wind < 0.7: anomaly.append("Low Wind Output")
+            if tilt > 10: anomaly.append("Pole 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) == 1:
             alert = "Yellow"
         elif len(anomaly) > 1:
             alert = "Red"
+
         data.append({
             "Pole ID": pole,
+            "Site": site,
             "Date": today,
             "Solar Gen (kWh)": solar,
             "Wind Gen (kWh)": wind,
@@ -47,4 +48,5 @@ def simulate_data(n=10, faults=True):
             "Anomalies": "; ".join(anomaly) if anomaly else "None",
             "Alert Level": alert
         })
-    return pd.DataFrame(data)
+
+    return pd.DataFrame(data)