generators/usa.py

from faker import Faker
import pandas as pd
import random

fake = Faker("en_US")

def generate(fields, count):
    data = []
    for _ in range(count):
        row = {}
        if "name" in fields:
            row["Name"] = fake.name()
        if "email" in fields:
            row["Email"] = fake.email()
        if "phone" in fields:
            row["Phone"] = fake.phone_number()
        if "address" in fields:
            row["Address"] = fake.address().replace("\n", ", ")
        if "id" in fields:
            # Generate fake SSN
            row["SSN"] = fake.ssn()
        data.append(row)
    return pd.DataFrame(data)