from django.db import models


class Conversation(models.Model):
    user_id = models.IntegerField(db_index=True)  # ارجاع سبک به accounts.User
    lesson_id = models.IntegerField(null=True, blank=True)  # ارجاع سبک به content.Lesson
    chapter_id = models.IntegerField(null=True, blank=True)  # شناسه‌ی فصل RAG (خارج از content/)
    grade_id = models.IntegerField(null=True, blank=True)  # ارجاع سبک به content.Grade
    created_at = models.DateTimeField(auto_now_add=True)

    def __str__(self):
        return f"conv-{self.pk}-user{self.user_id}"


class Message(models.Model):
    ROLE_CHOICES = [('user', 'کاربر'), ('assistant', 'دستیار')]

    conversation = models.ForeignKey(Conversation, on_delete=models.CASCADE, related_name='messages')
    role = models.CharField(max_length=10, choices=ROLE_CHOICES)
    text = models.TextField()
    question_id = models.IntegerField(null=True, blank=True)  # ارجاع سبک به content.Question
    created_at = models.DateTimeField(auto_now_add=True)

    class Meta:
        ordering = ['created_at']

    def __str__(self):
        return f"msg-{self.pk}-{self.role}"


class MessageFeedback(models.Model):
    message = models.OneToOneField(Message, on_delete=models.CASCADE, related_name='feedback')
    is_helpful = models.BooleanField()

    def __str__(self):
        return f"feedback-msg{self.message_id}"