from django.db import models

class AiModel(models.Model):
    # 1. ข้อมูลพื้นฐานและการลงทะเบียน
    name = models.CharField(max_length=255, unique=True, verbose_name="ชื่อโมเดล")
    model_version = models.CharField(max_length=50, default="v1.0.0", verbose_name="เวอร์ชัน")
    developer = models.CharField(max_length=255, blank=True, null=True, verbose_name="ผู้พัฒนา/ทีม")

    # 2. ข้อมูลการเรียกใช้ (API Management)
    base_url = models.URLField(max_length=500, verbose_name="Base URL ของ AI Service (Internal)")
    inference_path = models.CharField(max_length=255, verbose_name="Endpoint Path สำหรับ Inference")
    # ตัวอย่าง: base_url="http://ai_model_server:8001/", inference_path="inference/spleen/"

    # 3. สถานะและการควบคุม
    status_choices = [
        ('ACTIVE', 'ใช้งาน'),
        ('INACTIVE', 'ไม่ใช้งาน'),
        ('TESTING', 'กำลังทดสอบ'),
    ]
    status = models.CharField(max_length=20, choices=status_choices, default='INACTIVE', verbose_name="สถานะบริการ")

    auth_required = models.BooleanField(default=False, verbose_name="ต้องการ Internal Auth Key")

    created_at = models.DateTimeField(auto_now_add=True)
    updated_at = models.DateTimeField(auto_now=True)

    def full_inference_url(self):
        # สร้าง Full URL สำหรับ DRF ใช้เรียก (Proxy)
        return f"{self.base_url.rstrip('/')}/{self.inference_path.lstrip('/')}"

    def __str__(self):
        return f"{self.name} ({self.model_version}) - {self.status}"

    class Meta:
        verbose_name = "AI Model Registry"
        verbose_name_plural = "AI Model Registry"
        # สามารถเพิ่ม unique_together (name, model_version) ได้ในภายหลัง