| 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 | |