Attempt fixing the purl

src/aibom_generator/generator.py

@@ -3,7 +3,9 @@ import uuid
 import datetime
 from typing import Dict, Optional, Any, List
 
+
 from huggingface_hub import HfApi, ModelCard
+from urllib.parse import urlparse
 from .utils import calculate_completeness_score
 
 
@@ -31,6 +33,7 @@ class AIBOMGenerator:
         use_best_practices: Optional[bool] = None,  # Added parameter for industry-neutral scoring
     ) -> Dict[str, Any]:
         try:
+            model_id = self._normalise_model_id(model_id)
             use_inference = include_inference if include_inference is not None else self.use_inference
             # Use method parameter if provided, otherwise use instance variable
             use_best_practices = use_best_practices if use_best_practices is not None else self.use_best_practices
@@ -159,6 +162,23 @@ class AIBOMGenerator:
             print(f"Error fetching model info for {model_id}: {e}")
             return {}
 
+    # ---- new helper ---------------------------------------------------------
+    @staticmethod
+    def _normalise_model_id(raw_id: str) -> str:
+        """
+        Accept either  'owner/model'  or a full URL like
+        'https://huggingface.co/owner/model'.  Return 'owner/model'.
+        """
+        if raw_id.startswith(("http://", "https://")):
+            path = urlparse(raw_id).path.lstrip("/")
+            # path can contain extra segments (e.g. /commit/...), keep first two
+            parts = path.split("/")
+            if len(parts) >= 2:
+                return "/".join(parts[:2])
+            return path
+        return raw_id
+    # -------------------------------------------------------------------------
+
     def _fetch_model_card(self, model_id: str) -> Optional[ModelCard]:
         try:
             return ModelCard.load(model_id)
@@ -588,3 +608,4 @@ class AIBOMGenerator:
         }
         
         return license_urls.get(license_id, "https://spdx.org/licenses/")
+