Merge pull request #195 from rishab11250/feature/hf-token-column

backend/app/database.py

@@ -3,11 +3,13 @@ SQLAlchemy database setup with SQLite.
 Uses synchronous SQLAlchemy for simplicity and compatibility.
 """
 import os
-from sqlalchemy import create_engine
+import logging
+from sqlalchemy import create_engine, inspect, text
 from sqlalchemy.orm import sessionmaker, declarative_base
 from app.config import get_settings
 
 settings = get_settings()
+logger = logging.getLogger(__name__)
 
 # ── Ensure data directory exists ─────────────────────
 db_path = settings.DATABASE_URL.replace("sqlite:///", "")
@@ -34,7 +36,34 @@ def get_db():
         db.close()
 
 
+def _migrate_schema():
+    """Apply schema migrations for existing databases (SQLite-compatible).
+
+    SQLAlchemy's ``create_all`` only creates new tables and does **not**
+    add missing columns to existing tables.  This helper fills that gap
+    for non-destructive changes such as new nullable columns.
+    """
+    inspector = inspect(engine)
+    existing_columns = {c["name"] for c in inspector.get_columns("users")}
+
+    migrations = [
+        ("users", "hf_token", "ALTER TABLE users ADD COLUMN hf_token VARCHAR(255)"),
+    ]
+
+    for table, column, ddl in migrations:
+        if column not in existing_columns:
+            try:
+                with engine.begin() as conn:
+                    conn.execute(text(ddl))
+                logger.info("Migration: added column %s.%s", table, column)
+            except Exception:
+                logger.warning(
+                    "Migration skipped (may already exist): %s.%s", table, column
+                )
+
+
 def init_db():
-    """Create all tables on startup."""
+    """Create all tables on startup and apply schema migrations."""
     from app import models  # noqa: F401 — import to register models
     Base.metadata.create_all(bind=engine)
+    _migrate_schema()

backend/app/models.py

@@ -22,6 +22,7 @@ class User(Base):
     is_admin = Column(Boolean, default=False)
     created_at = Column(DateTime, default=lambda: datetime.now(timezone.utc))
     last_login = Column(DateTime, nullable=True, index=True)
+    hf_token = Column(String(255), nullable=True)
 
     # Relationships
     documents = relationship("Document", back_populates="owner", cascade="all, delete-orphan")

backend/app/schemas.py

@@ -53,11 +53,17 @@ class RefreshRequest(BaseModel):
     refresh_token: str
 
 
+class HFTokenUpdate(BaseModel):
+    """Request schema for updating the user's HuggingFace token."""
+    hf_token: str
+
+
 class UserResponse(BaseModel):
     id: str
     username: str
     email: str
     is_admin: bool
+    hf_token: Optional[str] = None
     created_at: datetime
 
     class Config: