def writeKpiToPostgres(**context):
    ti = context["ti"]

    # --- Pull KPI values from previous tasks ---
    elapsed_time = ti.xcom_pull(task_ids="getProgrammeData")  # KPI 1 (days)

    contractKPIs = ti.xcom_pull(task_ids="getContractData")   # KPI 2–4
    riskKPIs = ti.xcom_pull(task_ids="getRiskRegisterData")   # KPI 9–10

    kpi1 = float(elapsed_time or 0)
    kpi2 = float(contractKPIs.get("kpi2_ratio_pwdd") or 0)
    kpi3 = float(contractKPIs.get("kpi3_estimate_vs_forecast") or 0)
    kpi4 = float(contractKPIs.get("kpi4_pain_share") or 0)

    kpi9 = float(riskKPIs.get("kpi9_open_gt_3m") or 0)
    kpi10 = float(riskKPIs.get("kpi10_open_gt_6m") or 0)

    # --- DB connection ---
    host           = 'drowdatewarehouse.crlwwhgepgi7.ap-east-1.rds.amazonaws.com'
    dbUserName     = 'dRowAdmin'
    dbUserPassword = 'drowsuper'
    database       = 'drowDateWareHouse'
    port           = "5432"

    conn_string = (
        'postgres://'
        + dbUserName + ':' + dbUserPassword
        + '@' + host + ':' + port
        + '/' + database
    )

    db = create_engine(conn_string)

    delete_table_sql_query = """DROP TABLE IF EXISTS nd201905_kpi"""

    create_table_sql_query = """
    CREATE TABLE IF NOT EXISTS nd201905_kpi (
        id SERIAL PRIMARY KEY,
        kpi_no INT NOT NULL,
        kpi_name VARCHAR(50) NOT NULL,
        subject TEXT,
        description TEXT,
        value VARCHAR(100) NOT NULL,
        unit VARCHAR(50),
        condition VARCHAR(10),
        action TEXT,
        created_at TIMESTAMP DEFAULT NOW(),
        updated_at TIMESTAMP DEFAULT NOW()
    );
    """

    # --- Build KPI rows ---
    rows = []

    # KPI 1
    cond, action = evaluate_kpi_condition(1, kpi1)
    rows.append({
        "kpi_no": 1,
        "kpi_name": "KPI No. 1",
        "subject": "Programme",
        "description": "No. of days between the acceptance of programme and the reporting date",
        "value": str(kpi1),
        "unit": "days",
        "condition": cond,
        "action": action,
    })

    # KPI 2
    cond, action = evaluate_kpi_condition(2, kpi2)
    rows.append({
        "kpi_no": 2,
        "kpi_name": "KPI No. 2",
        "subject": "Prices",
        "description": "PWDD % for target cost contract",
        "value": f"{kpi2:.2f}",
        "unit": "%",
        "condition": cond,
        "action": action,
    })

    # KPI 3
    cond, action = evaluate_kpi_condition(3, kpi3)
    rows.append({
        "kpi_no": 3,
        "kpi_name": "KPI No. 3",
        "subject": "Prices",
        "description": "Difference of total of the Prices between PM and Contractor",
        "value": f"{kpi3:.2f}",
        "unit": "%",
        "condition": cond,
        "action": action,
    })

    # KPI 4
    cond, action = evaluate_kpi_condition(4, kpi4)
    rows.append({
        "kpi_no": 4,
        "kpi_name": "KPI No. 4",
        "subject": "Prices",
        "description": "% of Pain Share / total of the Prices (adjusted by fee percentage)",
        "value": f"{kpi4:.2f}",
        "unit": "%",
        "condition": cond,
        "action": action,
    })

    # KPI 9
    cond, action = evaluate_kpi_condition(9, kpi9)
    rows.append({
        "kpi_no": 9,
        "kpi_name": "KPI No. 9",
        "subject": "Early Warning",
        "description": "No. of Early Warnings with Elapse time > 3 months",
        "value": str(int(kpi9)),
        "unit": "no.",
        "condition": cond,
        "action": action,
    })

    # KPI 10
    cond, action = evaluate_kpi_condition(10, kpi10)
    rows.append({
        "kpi_no": 10,
        "kpi_name": "KPI No. 10",
        "subject": "Early Warning",
        "description": "No. of Early Warnings with Elapse time > 6 months",
        "value": str(int(kpi10)),
        "unit": "no.",
        "condition": cond,
        "action": action,
    })

    kpi_df = pd.DataFrame(rows)

    print(f"check dataframe: {kpi_df}")

    # --- Drop, create, insert ---
    with db.begin() as conn:
        conn.execute(delete_table_sql_query)
        conn.execute(create_table_sql_query)

        kpi_df.to_sql(
            "nd201905_kpi",
            con=conn,
            if_exists="append",
            index=False,
        )

    print("nd201905_kpi table refreshed with KPI 1,2,3,4,9,10.")