Update modules/simulator.py

modules/simulator.py

@@ -4,17 +4,24 @@ import datetime
 
 def simulate_data(n=48, faults=True):
     today = datetime.date.today()
-  site_coords = {
-    "Hyderabad": (17.3850, 78.4867),
-    "Gadwal": (16.2350, 77.8000),
-    "Kurnool": (15.8281, 78.0373),
-    "Ballari": (15.1394, 76.9214)
-}
+
+    sites = ["Hyderabad", "Gadwal", "Kurnool", "Ballari"]
+
+    # Define latitude and longitude for each site
+    site_coords = {
+        "Hyderabad": (17.3850, 78.4867),
+        "Gadwal": (16.2350, 77.8000),
+        "Kurnool": (15.8281, 78.0373),
+        "Ballari": (15.1394, 76.9214)
+    }
+
     poles = [f"Pole_{i+1:03}" for i in range(n)]
     data = []
 
     for i, pole in enumerate(poles):
         site = sites[i // 12]  # 12 poles per site
+        lat, lon = site_coords[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)
@@ -23,28 +30,44 @@ def simulate_data(n=48, faults=True):
         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 = []
 
+        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"
-            
-lat, lon = site_coords[site]
-
-data.append({
-    "Pole ID": pole,
-    "Site": site,
-    "Latitude": lat + np.random.uniform(-0.005, 0.005),  # Add small variation for each pole
-    "Longitude": lon + np.random.uniform(-0.005, 0.005),
-    ...
-})
+
+        data.append({
+            "Pole ID": pole,
+            "Site": site,
+            "Latitude": lat + np.random.uniform(-0.005, 0.005),  # small offset
+            "Longitude": lon + np.random.uniform(-0.005, 0.005),
+            "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)
+