database.py

from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker, Session
from config import get_settings

# Load settings from .env file
settings = get_settings()
DATABASE_URL = settings.database_url
print(f"Database URL: {DATABASE_URL}")  # Debug print

# Force SQLite for now
DATABASE_URL = "sqlite:///./disaster.db"
print(f"Using SQLite: {DATABASE_URL}")

# Configure engine based on database type
if DATABASE_URL.startswith("sqlite"):
    engine = create_engine(
        DATABASE_URL,
        connect_args={"check_same_thread": False}  # SQLite specific
    )
else:
    # PostgreSQL settings
    engine = create_engine(
        DATABASE_URL,
        pool_pre_ping=True,
        pool_size=10,
        max_overflow=20
    )

SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)

def get_db():
    """Dependency for FastAPI routes"""
    db = SessionLocal()
    try:
        yield db
    finally:
        db.close()

def init_db():
    """Initialize database tables"""
    from models.base import Base
    
    # Create all tables
    Base.metadata.create_all(bind=engine)