database.py

import sqlite3

def init_db():
    conn = sqlite3.connect("db/careercoach.db")
    c = conn.cursor()
    c.execute('''CREATE TABLE IF NOT EXISTS chats (

        id INTEGER PRIMARY KEY AUTOINCREMENT,

        question TEXT,

        answer TEXT,

        timestamp DATETIME DEFAULT CURRENT_TIMESTAMP

    )''')
    conn.commit()
    conn.close()

def save_chat(question, answer):
    conn = sqlite3.connect("db/careercoach.db")
    c = conn.cursor()
    c.execute("INSERT INTO chats (question, answer) VALUES (?, ?)", (question, answer))
    conn.commit()
    conn.close()