"""
Test Model Demo - Main Module
Questo è un modello di esempio per dimostrare la struttura di un modello su Hugging Face.
"""

import json
from pathlib import Path


class TestModelDemo:
    """
    Classe di esempio per un modello di test.
    """
    
    def __init__(self, config_path="config.json"):
        """
        Inizializza il modello di test.
        
        Args:
            config_path: Percorso del file di configurazione
        """
        self.config_path = config_path
        self.config = self._load_config()
        
    def _load_config(self):
        """Carica la configurazione dal file JSON."""
        try:
            with open(self.config_path, 'r', encoding='utf-8') as f:
                return json.load(f)
        except FileNotFoundError:
            return {"model_type": "test-demo", "error": "config not found"}
    
    def get_info(self):
        """Restituisce informazioni sul modello."""
        return {
            "model_type": self.config.get("model_type", "unknown"),
            "architecture": self.config.get("architecture", "unknown"),
            "version": self.config.get("version", "unknown"),
            "task": self.config.get("task", "unknown"),
            "vocab_size": self.config.get("parameters", {}).get("vocab_size", 0)
        }
    
    def process(self, text):
        """
        Metodo di esempio per processare testo.
        In un modello reale, questo farebbe elaborazione ML.
        
        Args:
            text: Testo da processare
            
        Returns:
            Risultato del processamento (dummy)
        """
        return {
            "input": text,
            "processed": text.upper(),
            "model_info": self.get_info()
        }


def main():
    """Funzione principale per testare il modello."""
    model = TestModelDemo()
    info = model.get_info()
    
    print("=== Test Model Demo ===")
    print(f"Model Type: {info['model_type']}")
    print(f"Architecture: {info['architecture']}")
    print(f"Version: {info['version']}")
    print(f"Task: {info['task']}")
    print(f"Vocab Size: {info['vocab_size']}")
    
    # Esempio di processamento
    result = model.process("Hello, this is a test!")
    print(f"\nEsempio di processamento:")
    print(f"Input: {result['input']}")
    print(f"Output: {result['processed']}")


if __name__ == "__main__":
    main()