Update modules/simulator.py

modules/simulator.py

@@ -6,7 +6,7 @@ def simulate_data(n=10, faults=True):
     today = datetime.date.today()
     poles = [f"Pole_{i+1:03}" for i in range(n)]
 
-    # Define site coordinates
+    # Coordinates for sites
     sites = {
         "Hyderabad": (17.3850, 78.4867),
         "Gadwal": (16.2356, 77.7956),
@@ -39,11 +39,15 @@ def simulate_data(n=10, faults=True):
             if cam == "Offline": anomaly.append("Camera Offline")
             if sufficient == "No": anomaly.append("Power Insufficient")
 
-        alert = "Green"
-        if len(anomaly) == 1:
+        if len(anomaly) == 0:
+            alert = "Green"
+            alert_weight = 1
+        elif len(anomaly) == 1:
             alert = "Yellow"
-        elif len(anomaly) > 1:
+            alert_weight = 2
+        else:
             alert = "Red"
+            alert_weight = 3
 
         data.append({
             "Pole ID": pole,
@@ -59,8 +63,10 @@ def simulate_data(n=10, faults=True):
             "Tilt (°)": tilt,
             "Vibration (g)": vib,
             "Anomalies": "; ".join(anomaly) if anomaly else "None",
-            "Alert Level": alert
+            "Alert Level": alert,
+            "Alert Weight": alert_weight
         })
 
     return pd.DataFrame(data)
 
+