database.py

import sqlite3
import pandas as pd

class DatabaseHandler:
    def __init__(self, db_name="notes.db"):
        self.conn = sqlite3.connect(db_name, check_same_thread=False)
        self.create_table()

    def create_table(self):
        query = """
        CREATE TABLE IF NOT EXISTS notes (
            id INTEGER PRIMARY KEY AUTOINCREMENT,
            content TEXT,
            tag TEXT,
            priority TEXT,
            timestamp DATETIME DEFAULT CURRENT_TIMESTAMP
        )
        """
        self.conn.execute(query)
        self.conn.commit()

    def add_note(self, content, tag, priority):
        query = "INSERT INTO notes (content, tag, priority) VALUES (?, ?, ?)"
        self.conn.execute(query, (content, tag, priority))
        self.conn.commit()

    def fetch_all(self):
        return pd.read_sql_query("SELECT * FROM notes ORDER BY timestamp DESC", self.conn)