utils/db.py

import sqlite3
import pandas as pd

def connect():
    return sqlite3.connect("https://fabiosantos-teama.hf.space/db/aba_data.db")

def query_sessions(user_id, start=None, end=None):
    conn = connect()
    if start and end:
        query = """
        SELECT * FROM sessions WHERE user_id = ? AND date BETWEEN ? AND ? ORDER BY date
        """
        df = pd.read_sql_query(query, conn, params=(user_id, start, end))
    else:
        query = """
        SELECT * FROM sessions WHERE user_id = ? ORDER BY date
        """
        df = pd.read_sql_query(query, conn, params=(user_id,))
    conn.close()
    return df