modules/simulator.py

import pandas as pd
import numpy as np

def simulate_pole_data(n=100):
    np.random.seed(1)
    sites = ["Hyderabad", "Ballari", "Gadwal", "Kurnool"]
    df = pd.DataFrame({
        "PoleID": [f"Pole_{i:05}" for i in range(n)],
        "Site": np.random.choice(sites, size=n),
        "Location_Latitude": np.random.uniform(16.5, 18.5, n),
        "Location_Longitude": np.random.uniform(77.0, 79.5, n),
        "SolarGen(kWh)": np.random.uniform(2.0, 6.5, n),
        "WindGen(kWh)": np.random.uniform(0.3, 2.5, n),
        "Tilt(°)": np.random.uniform(0, 20, n),
        "Vibration(g)": np.random.uniform(0.1, 3.5, n),
        "CameraStatus": np.random.choice(["Online", "Offline"], n),
        "PowerSufficient": np.random.choice(["Yes", "No"], n),
    })
    return df