from django.db import models

class AIInquiry(models.Model):
    STATUS_CHOICES = [
        ('new', 'New'),
        ('in_progress', 'In Progress'),
        ('contacted', 'Contacted'),
        ('closed', 'Closed'),
    ]

    name = models.CharField(max_length=255, verbose_name="Full Name")
    email = models.EmailField(max_length=254, verbose_name="Business Email")
    company = models.CharField(max_length=255, verbose_name="Company / Agency")
    role = models.CharField(max_length=255, blank=True, null=True, verbose_name="Role / Title")
    phone = models.CharField(max_length=50, blank=True, null=True, verbose_name="Phone Number")
    interest = models.CharField(max_length=100, verbose_name="Interest Area")
    environment = models.CharField(max_length=100, blank=True, null=True, verbose_name="Primary Environment")
    timeline = models.CharField(max_length=100, blank=True, null=True, verbose_name="Timeline")
    use_case = models.TextField(verbose_name="Use Case Details")
    status = models.CharField(max_length=20, choices=STATUS_CHOICES, default='new', verbose_name="Status")
    created_at = models.DateTimeField(auto_now_add=True, verbose_name="Submitted At")

    class Meta:
        verbose_name = "AI Inquiry"
        verbose_name_plural = "AI Inquiries"
        ordering = ['-created_at']

    def __str__(self):
        return f"{self.name} - {self.company} ({self.created_at.strftime('%Y-%m-%d')})"


class ContactSubmission(models.Model):
    STATUS_CHOICES = [
        ('new', 'New'),
        ('in_progress', 'In Progress'),
        ('contacted', 'Contacted'),
        ('closed', 'Closed'),
    ]

    CATEGORY_CHOICES = [
        ('project', 'Project Discussion'),
        ('teaming', 'Teaming / Public Sector'),
        ('capability', 'Capability Statement'),
        ('talent', 'Talent Request'),
        ('careers', 'Careers / General'),
    ]

    category = models.CharField(max_length=50, choices=CATEGORY_CHOICES, default='project', verbose_name="Category")
    name = models.CharField(max_length=255, verbose_name="Full Name")
    email = models.EmailField(max_length=254, verbose_name="Email Address")
    company = models.CharField(max_length=255, blank=True, null=True, verbose_name="Company / Organization")
    phone = models.CharField(max_length=50, blank=True, null=True, verbose_name="Phone Number")
    role = models.CharField(max_length=255, blank=True, null=True, verbose_name="Role / Title")
    project_type = models.CharField(max_length=255, blank=True, null=True, verbose_name="Project Type / Details")
    timeline = models.CharField(max_length=100, blank=True, null=True, verbose_name="Timeline")
    budget = models.CharField(max_length=100, blank=True, null=True, verbose_name="Budget Range")
    message = models.TextField(blank=True, null=True, verbose_name="Message / Inquiry Details")
    status = models.CharField(max_length=20, choices=STATUS_CHOICES, default='new', verbose_name="Status")
    created_at = models.DateTimeField(auto_now_add=True, verbose_name="Submitted At")

    class Meta:
        verbose_name = "Contact Submission"
        verbose_name_plural = "Contact Submissions"
        ordering = ['-created_at']

    def __str__(self):
        return f"[{self.get_category_display()}] {self.name} - {self.company or self.email} ({self.created_at.strftime('%Y-%m-%d')})"