src/data/database.py

import os
import psycopg2
from dotenv import load_dotenv

load_dotenv() # Load environment variables from .env

DATABASE_URL = os.getenv("DATABASE_URL")

def get_db_connection():
    if not DATABASE_URL:
        raise ValueError("DATABASE_URL environment variable not set.")
    conn = psycopg2.connect(DATABASE_URL)
    return conn