Upload app/core/database.py with huggingface_hub

app/core/database.py

@@ -6,12 +6,22 @@ from sqlalchemy.orm import sessionmaker, declarative_base
 
 from app.core.config import settings
 
-# Create async SQLAlchemy engine
-engine = create_async_engine(
-    str(settings.DATABASE_URL).replace("postgresql://", "postgresql+asyncpg://"),
-    pool_pre_ping=True,
-    echo=settings.DEBUG,
-)
+# Get the database URL from settings
+db_url = str(settings.DATABASE_URL)
+
+# For SQLite, don't add asyncpg prefix
+if db_url.startswith("sqlite"):
+    engine = create_async_engine(
+        db_url,
+        echo=settings.DEBUG,
+    )
+else:
+    # For PostgreSQL, add asyncpg prefix
+    engine = create_async_engine(
+        db_url.replace("postgresql://", "postgresql+asyncpg://"),
+        pool_pre_ping=True,
+        echo=settings.DEBUG,
+    )
 
 # Create async SessionLocal class
 AsyncSessionLocal = sessionmaker(