from datetime import datetime

from db.init import DB_TIMEZONE, get_db_connection


def save_feedback(
    conversation_id, source, relevance=None, explanation=None, score=None
):
    timestamp = datetime.now(DB_TIMEZONE)

    conn = get_db_connection()
    try:
        with conn.cursor() as cur:
            cur.execute(
                """
                INSERT INTO feedback (
                    conversation_id, source, relevance,
                    explanation, score, timestamp
                ) VALUES (
                    %s, %s, %s, %s, %s, %s
                )
                """,
                (conversation_id, source, relevance, explanation, score, timestamp),
            )
        conn.commit()
    finally:
        conn.close()