import sqlite3

conn = sqlite3.connect('instance/swiftcart_cloud.db')
c = conn.cursor()

new_columns = [
    ("contact_person", "VARCHAR(150)"),
    ("whatsapp_number", "VARCHAR(50)"),
    ("address", "TEXT"),
    ("credit_limit", "FLOAT DEFAULT 0.0"),
    ("category", "VARCHAR(100)")
]

for col_name, col_type in new_columns:
    try:
        c.execute(f"ALTER TABLE khata_account ADD COLUMN {col_name} {col_type}")
        print(f"Added column {col_name}")
    except sqlite3.OperationalError as e:
        print(f"Column {col_name} might already exist or error: {e}")

conn.commit()
conn.close()