config.py

"""
Configuration pour l'application d'analyse des adventices agricoles
"""
import os

# Configuration Hugging Face
HF_TOKEN = os.environ.get("HF_TOKEN")
DATASET_ID = "HackathonCRA/2024"

# Configuration des données
REQUIRED_COLUMNS = ["numparcell", "surfparc", "millesime"]
OPTIONAL_GROUP_COLS = ["nomparc", "libelleusag"]

# Configuration des risques
RISK_LEVELS = ['TRÈS FAIBLE', 'FAIBLE', 'MODÉRÉ', 'ÉLEVÉ', 'TRÈS ÉLEVÉ']
RISK_COLORS = {
    'TRÈS FAIBLE': 'green',
    'FAIBLE': 'lightgreen', 
    'MODÉRÉ': 'orange',
    'ÉLEVÉ': 'red',
    'TRÈS ÉLEVÉ': 'darkred'
}

# Configuration Gradio
GRADIO_CONFIG = {
    "server_name": "0.0.0.0",
    "server_port": 7860,
    "share": False
}

# Configuration des graphiques
PLOT_CONFIG = {
    "width": 800,
    "height": 600,
    "title_font_size": 16
}

# Messages de l'application
MESSAGES = {
    "loading": "🔄 Chargement des données depuis Hugging Face...",
    "success": "✅ Données chargées avec succès",
    "error_loading": "❌ Erreur lors du chargement du dataset",
    "no_data": "❌ Aucune donnée disponible",
    "analysis_complete": "✅ Analyse terminée avec succès"
}