Update modules/simulator.py

modules/simulator.py

@@ -1,50 +1,47 @@
 import pandas as pd
 import numpy as np
+import datetime
 
-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)
-    }
+def simulate_data(n=10, faults=True):
+    today = datetime.date.today()
+    poles = [f"Pole_{i+1:03}" for i in range(n)]
     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"
 
+    for pole in poles:
+        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"
         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 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) == 0 else "Yellow" if len(anomaly) == 1 else "Red"
+        alert = "Green"
+        if len(anomaly) == 1: alert = "Yellow"
+        elif len(anomaly) > 1: alert = "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,
+            "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
         })
+
     return pd.DataFrame(data)