from django.db import models
from django.contrib.auth.models import User

class UserProfile(models.Model):
    user = models.OneToOneField(User, on_delete=models.CASCADE)
    # Ajoutez d'autres champs personnalisés ici si besoin

    def __str__(self):
        return self.user.username

class Modele(models.Model):
    nom = models.CharField(max_length=255)
    code = models.CharField(max_length=255, unique=True, default="gpt-4.1-mini")
    actif = models.BooleanField(default=True)

    def __str__(self):
        return self.nom

class UserCredit(models.Model):
    user = models.OneToOneField(User, on_delete=models.CASCADE, related_name='credit')
    credit = models.IntegerField(default=10)

    def __str__(self):
        return f"{self.user.username} - Crédit: {self.credit}"