Update models.py

models.py

@@ -1,5 +1,7 @@
-from sqlalchemy import Column, Integer, String, Boolean, ARRAY
-from sqlalchemy.ext.declarative import declarative_base
+from sqlalchemy import Column, Integer, String, DateTime, ForeignKey
+from sqlalchemy.orm import relationship
+from sqlalchemy.sql import func
+from .database import Base
 
 Base = declarative_base()
 
@@ -14,5 +16,16 @@ class User(Base):
     selected_tts_models = Column(ARRAY(String), default=['default.onnx'])  # Set the default value to 'default.onnx'
 
     # Define other user fields as needed
-
+    # Add the relationship with VerificationToken
+    verification_tokens = relationship("VerificationToken", back_populates="user")
 # Define other database models as needed
+class VerificationToken(Base):
+    __tablename__ = "verification_tokens"
+
+    id = Column(Integer, primary_key=True, index=True)
+    token = Column(String, unique=True, index=True)
+    user_id = Column(Integer, ForeignKey('users.id'), nullable=False)
+    created_at = Column(DateTime(timezone=True), server_default=func.now())
+
+    # Define a relationship with the User model
+    user = relationship("User", back_populates="verification_tokens")