jeevzz commited on
Commit
32f27f1
·
verified ·
1 Parent(s): a740383

Update database.py

Browse files
Files changed (1) hide show
  1. 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)