Spaces:
Running
Running
Update database.py
Browse files- database.py +5 -6
database.py
CHANGED
|
@@ -4,15 +4,14 @@ from sqlalchemy.orm import sessionmaker, relationship
|
|
| 4 |
from datetime import datetime
|
| 5 |
import uuid
|
| 6 |
import os
|
| 7 |
-
from pathlib import Path
|
| 8 |
-
from dotenv import load_dotenv
|
| 9 |
-
|
| 10 |
-
# Load environment variables
|
| 11 |
-
env_path = Path(__file__).parent / '.env'
|
| 12 |
-
load_dotenv(dotenv_path=env_path)
|
| 13 |
|
|
|
|
|
|
|
| 14 |
DATABASE_URL = os.getenv("DATABASE_URL")
|
| 15 |
|
|
|
|
|
|
|
|
|
|
| 16 |
# Create engine
|
| 17 |
engine = create_engine(DATABASE_URL, pool_pre_ping=True)
|
| 18 |
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
|
|
|
|
| 4 |
from datetime import datetime
|
| 5 |
import uuid
|
| 6 |
import os
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 7 |
|
| 8 |
+
# For HF Spaces: Use environment variable directly (set via Secrets)
|
| 9 |
+
# For local dev: Can still use .env file
|
| 10 |
DATABASE_URL = os.getenv("DATABASE_URL")
|
| 11 |
|
| 12 |
+
if not DATABASE_URL:
|
| 13 |
+
raise ValueError("DATABASE_URL environment variable is not set!")
|
| 14 |
+
|
| 15 |
# Create engine
|
| 16 |
engine = create_engine(DATABASE_URL, pool_pre_ping=True)
|
| 17 |
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
|