# dashboard/models.py

from accounts.models import Company
from django.db import models


class DataSource(models.Model):
    """Model for data sources (CSV files or external API data)"""

    name = models.CharField(max_length=255)
    description = models.TextField(blank=True)
    file = models.FileField(
        upload_to="data_sources/",
        blank=True,
        null=True,
        help_text="Upload a CSV file or leave empty if using an external data source",
    )
    external_source = models.ForeignKey(
        "data_integration.ExternalDataSource",
        on_delete=models.SET_NULL,
        blank=True,
        null=True,
        help_text="Link to an external data source",
    )
    uploaded_at = models.DateTimeField(auto_now_add=True)
    company = models.ForeignKey(Company, on_delete=models.CASCADE, related_name="data_sources")

    def __str__(self):
        return self.name


class ChatSession(models.Model):
    """Model to store parsed chat session data from CSV"""

    data_source = models.ForeignKey(DataSource, on_delete=models.CASCADE, related_name="chat_sessions")
    session_id = models.CharField(max_length=255)
    start_time = models.DateTimeField(null=True, blank=True)
    end_time = models.DateTimeField(null=True, blank=True)
    ip_address = models.GenericIPAddressField(null=True, blank=True)
    country = models.CharField(max_length=100, blank=True)
    language = models.CharField(max_length=50, blank=True)
    messages_sent = models.IntegerField(default=0)
    sentiment = models.CharField(max_length=50, blank=True)
    escalated = models.BooleanField(default=False)
    forwarded_hr = models.BooleanField(default=False)
    full_transcript = models.TextField(blank=True)
    avg_response_time = models.FloatField(null=True, blank=True)
    tokens = models.IntegerField(default=0)
    tokens_eur = models.FloatField(null=True, blank=True)
    category = models.CharField(max_length=100, blank=True)
    initial_msg = models.TextField(blank=True)
    user_rating = models.CharField(max_length=50, blank=True)

    def __str__(self):
        return f"Session {self.session_id}"

    class Meta:
        unique_together = ("session_id", "data_source")


class Dashboard(models.Model):
    """Model for custom dashboards that can be created by users"""

    name = models.CharField(max_length=255)
    description = models.TextField(blank=True)
    company = models.ForeignKey(Company, on_delete=models.CASCADE, related_name="dashboards")
    data_sources = models.ManyToManyField(DataSource, related_name="dashboards")
    created_at = models.DateTimeField(auto_now_add=True)
    updated_at = models.DateTimeField(auto_now=True)

    def __str__(self):
        return self.name