import os
from dotenv import load_dotenv

load_dotenv(override=True)

class Constants:
    # MODEL NAMES
    MODEL = os.getenv("MODEL")
    EMBEDDING_MODEL = os.getenv("EMBEDDING_MODEL")

    # Database
    DB_NAME = os.getenv("DB_NAME")
    COLLECTION_NAME = os.getenv("COLLECTION_NAME")
    ATLAS_VECTOR_SEARCH_INDEX_NAME = os.getenv("ATLAS_VECTOR_SEARCH_INDEX_NAME")

    # Search parameters
    CUSTOMER_SEARCH_LIMIT = 5
    STAFF_SEARCH_LIMIT = 1

    # Prompts (in French)
    IMAGE_DESCRIPTION_PROMPT = """
    Décrivez cette image en détail. Concentrez-vous sur les éléments de ferronnerie décorative 
    visibles dans l'image. Mentionnez le style, les motifs, les formes, et l'utilisation possible 
    de cet article dans un jardin ou une maison.
    """

    CUSTOMER_SEARCH_SYSTEM_PROMPT = """
    Vous êtes un assistant spécialisé dans la ferronnerie décorative pour l'Atelier du Fer.
    Votre tâche est d'aider les clients à trouver des articles qui correspondent à leurs besoins.
    """

    STAFF_SEARCH_SYSTEM_PROMPT = """
    Vous êtes un assistant pour les employés de l'Atelier du Fer.
    Votre tâche est d'identifier précisément un article du catalogue à partir d'une image.
    """

    # UI Text (in French)
    UI_TITLE = "L'Atelier du Fer - Recherche Catalogue"
    UI_CUSTOMER_TAB = "Recherche Client"
    UI_STAFF_TAB = "Recherche Personnel"
    UI_CUSTOMER_DESCRIPTION = "Trouvez des articles de ferronnerie décorative pour embellir votre espace"
    UI_STAFF_DESCRIPTION = "Identifiez rapidement un article du catalogue"
    UI_TEXT_INPUT_LABEL = "Décrivez ce que vous recherchez"
    UI_IMAGE_INPUT_LABEL = "Téléchargez une image (optionnel)"
    UI_SEARCH_BUTTON = "Rechercher"
    UI_LOADING_TEXT = "Recherche en cours..."
    UI_NO_RESULTS = "Aucun résultat trouvé. Veuillez essayer avec une autre description ou image."
    UI_PRODUCT_BUTTON = "Voir le produit"
    UI_PRICE_LABEL = "Prix:"

    # Placeholder image URL for articles without images
    PLACEHOLDER_IMAGE_URL = "https://placehold.co/150?text=Pas+d'image"