Update nygaardcodecommentclassification/api/models.py

nygaardcodecommentclassification/api/models.py

@@ -181,21 +181,31 @@ class ModelRegistry:
         except Exception as e:
             logger.error("[%s] Error loading models: %s", lang.upper(), e)
 
-    def get_model(self, language: str, model_type: str) -> Optional[Dict[str, Any]]:
-        """Retrieve a loaded model entry by language and type.
-
-        Args:
-            language: The programming language code
-            model_type: The type of model
+    # In models.py
 
-        Returns:
-            Dict containing the model and metadata, or None if not found.
-            The dict contains:
-            - "model": The loaded ML model object
-            - "feature_type": Type of features used
-            - "embedder": Optional sentence transformer for embedding generation
-        """
-        return self._registry.get(language, {}).get(model_type)
+    def get_model(self, language: str, model_type: str) -> Optional[Dict[str, Any]]:
+        """Retrieve a loaded model entry by language and type."""
+        # 1. Prova accesso diretto (veloce)
+        if language in self._registry:
+            return self._registry[language].get(model_type)
+        
+        # 2. Se fallisce, prova case-insensitive (più lento ma robusto)
+        # Normalizza tutto in minuscolo per il confronto
+        normalized_lang = language.lower()
+        for reg_lang in self._registry.keys():
+            if reg_lang.lower() == normalized_lang:
+                return self._registry[reg_lang].get(model_type)
+        
+        # 3. DEBUG: Se non trova nulla, stampa cosa c'è nel registry
+        # Questo apparirà nei log di Hugging Face e ti dirà ESATTAMENTE quali chiavi sono caricate
+        logger.warning(
+            "Model lookup failed for lang='%s', type='%s'. Available langs in registry: %s", 
+            language, 
+            model_type, 
+            list(self._registry.keys())
+        )
+        
+        return None
 
     def clear(self) -> None:
         """Clear all models from the registry and free memory.