import sqlite3
import json

DB = "data/ekyc.db"

def init_db():
    conn = sqlite3.connect(DB)
    c = conn.cursor()
    c.execute("""
        CREATE TABLE IF NOT EXISTS users (
            id TEXT PRIMARY KEY,
            name TEXT,
            father TEXT,
            dob TEXT,
            type TEXT,
            embedding TEXT
        )
    """)
    conn.commit()
    conn.close()

def insert_record(data):
    conn = sqlite3.connect(DB)
    c = conn.cursor()
    c.execute("""
        INSERT INTO users VALUES (?, ?, ?, ?, ?, ?)
    """, (
        data["ID"],
        data["Name"],
        data["Father's Name"],
        data["DOB"],
        data["ID Type"],
        json.dumps(data["Embedding"])
    ))
    conn.commit()
    conn.close()

def check_duplicate(user_id):
    conn = sqlite3.connect(DB)
    c = conn.cursor()
    c.execute("SELECT * FROM users WHERE id=?", (user_id,))
    exists = c.fetchone() is not None
    conn.close()
    return exists