from ..extensions import db
from .user import User, Tenant


class ChatSession(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    user_id = db.Column(db.Integer, db.ForeignKey('public.user.id'), nullable=True)
    session_start = db.Column(db.DateTime, nullable=False)
    session_end = db.Column(db.DateTime, nullable=True)

    # Relations
    chat_interactions = db.relationship('Interaction', backref='chat_session', lazy=True)
    user = db.relationship('User', backref='chat_sessions', lazy=True)

    def __repr__(self):
        return f"<ChatSession {self.id} by {self.user_id}>"


class Interaction(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    chat_session_id = db.Column(db.Integer, db.ForeignKey('public.chat_session.id'), nullable=False)
    question = db.Column(db.Text, nullable=False)
    answer = db.Column(db.Text, nullable=True)
    language = db.Column(db.String(2), nullable=False)
    appreciation = db.Column(db.Integer, nullable=True, default=100)

    # Timing information
    question_at = db.Column(db.DateTime, nullable=False)
    answer_at = db.Column(db.DateTime, nullable=True)

    def __repr__(self):
        return f"<Interaction {self.id}>"


class InteractionEmbedding(db.Model):
    interaction_id = db.Column(db.Integer, db.ForeignKey('interaction.id', ondelete='CASCADE'), primary_key=True)
    embedding_id = db.Column(db.Integer, db.ForeignKey('embedding.id', ondelete='CASCADE'), primary_key=True)