Update modules/simulator.py

modules/simulator.py

@@ -8,9 +8,19 @@ def simulate_data(n=50, faults=True):
     poles = [f"Pole_{i+1:03}" for i in range(n)]
     # Distribute poles across 4 locations
     locations = ["Hyderabad"] * 12 + ["Gadwal"] * 12 + ["Kurnool"] * 12 + ["Bangalore"] * 14
-    np.random.shuffle(locations)  # Randomize for realism
+    # Simulate coordinates and zones for each location
+    coords = {
+        "Hyderabad": [(17.385 + np.random.uniform(-0.05, 0.05), 78.486 + np.random.uniform(-0.05, 0.05), f"Zone_{np.random.choice(['North', 'South', 'Central'])}") for _ in range(12)],
+        "Gadwal": [(16.235 + np.random.uniform(-0.03, 0.03), 77.795 + np.random.uniform(-0.03, 0.03), f"Zone_{np.random.choice(['East', 'West'])}") for _ in range(12)],
+        "Kurnool": [(15.828 + np.random.uniform(-0.04, 0.04), 78.037 + np.random.uniform(-0.04, 0.04), f"Zone_{np.random.choice(['Urban', 'Rural'])}") for _ in range(12)],
+        "Bangalore": [(12.971 + np.random.uniform(-0.06, 0.06), 77.594 + np.random.uniform(-0.06, 0.06), f"Zone_{np.random.choice(['Downtown', 'Suburban', 'Industrial'])}") for _ in range(14)]
+    }
+    location_coords = []
+    for loc in locations:
+        coord = coords[loc].pop(0)
+        location_coords.append(coord)
     data = []
-    for i, (pole, location) in enumerate(zip(poles, locations)):
+    for i, (pole, location, (lat, lon, zone)) in enumerate(zip(poles, locations, location_coords)):
         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)
@@ -43,6 +53,9 @@ def simulate_data(n=50, faults=True):
             "PoleID": pole,
             "RFID": rfid,
             "Location": location,
+            "Zone": zone,
+            "Latitude": lat,
+            "Longitude": lon,
             "Date": today,
             "SolarGen(kWh)": solar,
             "WindGen(kWh)": wind,