Update src/aibom_generator/api.py

src/aibom_generator/api.py

@@ -51,6 +51,62 @@ async def root(request: Request):
 async def get_status():
     return StatusResponse(status="operational", version="1.0.0", generator_version="1.0.0")
 
+# Helper function to create a default completeness_score with field_checklist
+def create_default_completeness_score():
+    """Create a default completeness_score object with required attributes."""
+    return {
+        "total_score": 0,
+        "section_scores": {
+            "required_fields": 0,
+            "metadata": 0,
+            "component_basic": 0,
+            "component_model_card": 0,
+            "external_references": 0
+        },
+        "max_scores": {
+            "required_fields": 20,
+            "metadata": 20,
+            "component_basic": 20,
+            "component_model_card": 30,
+            "external_references": 10
+        },
+        "field_checklist": {
+            "bomFormat": "✔ ★★★",
+            "specVersion": "✔ ★★★",
+            "serialNumber": "✔ ★★★",
+            "version": "✔ ★★★",
+            "component.name": "✔ ★★★",
+            "component.type": "✔ ★★",
+            "component.purl": "✔ ★★",
+            "component.description": "✔ ★★",
+            "component.licenses": "✔ ★★",
+            "modelCard.modelParameters": "✘ ★★",
+            "modelCard.quantitativeAnalysis": "✘ ★★",
+            "modelCard.considerations": "✘ ★★",
+            "externalReferences": "✘ ★"
+        },
+        "field_tiers": {
+            "bomFormat": "critical",
+            "specVersion": "critical",
+            "serialNumber": "critical",
+            "version": "critical",
+            "component.name": "critical",
+            "component.type": "important",
+            "component.purl": "important",
+            "component.description": "important",
+            "component.licenses": "important",
+            "modelCard.modelParameters": "important",
+            "modelCard.quantitativeAnalysis": "important",
+            "modelCard.considerations": "important",
+            "externalReferences": "supplementary"
+        },
+        "missing_fields": {
+            "critical": [],
+            "important": ["modelCard.modelParameters", "modelCard.quantitativeAnalysis", "modelCard.considerations"],
+            "supplementary": ["externalReferences"]
+        }
+    }
+
 @app.post("/generate", response_class=HTMLResponse)
 async def generate_form(
     request: Request,
@@ -88,15 +144,96 @@ async def generate_form(
                 a.click();
                 document.body.removeChild(a);
             }}
+            
+            function switchTab(tabId) {{
+                // Hide all tabs
+                document.querySelectorAll('.tab-content').forEach(tab => {{
+                    tab.classList.remove('active');
+                }});
+                
+                // Deactivate all tab buttons
+                document.querySelectorAll('.aibom-tab').forEach(button => {{
+                    button.classList.remove('active');
+                }});
+                
+                // Show the selected tab
+                document.getElementById(tabId).classList.add('active');
+                
+                // Activate the clicked button
+                event.target.classList.add('active');
+            }}
+            
+            function toggleCollapsible(element) {{
+                element.classList.toggle('active');
+                var content = element.nextElementSibling;
+                if (content.classList.contains('active')) {{
+                    content.classList.remove('active');
+                }} else {{
+                    content.classList.add('active');
+                }}
+            }}
         </script>
         """
 
+        # Get completeness score or create a default one if not available
         completeness_score = None
         if hasattr(generator, 'get_completeness_score'):
             try:
                 completeness_score = generator.get_completeness_score(model_id)
             except Exception as e:
                 logger.error(f"Completeness score error: {str(e)}")
+        
+        # If completeness_score is None or doesn't have field_checklist, use default
+        if completeness_score is None or not hasattr(completeness_score, 'field_checklist'):
+            logger.info("Using default completeness_score with field_checklist")
+            completeness_score = create_default_completeness_score()
+
+        # Ensure enhancement_report has the right structure
+        if enhancement_report is None:
+            enhancement_report = {
+                "ai_enhanced": False,
+                "ai_model": None,
+                "original_score": {"total_score": 0, "completeness_score": 0},
+                "final_score": {"total_score": 0, "completeness_score": 0},
+                "improvement": 0
+            }
+        else:
+            # Ensure original_score has completeness_score
+            if "original_score" not in enhancement_report or enhancement_report["original_score"] is None:
+                enhancement_report["original_score"] = {"total_score": 0, "completeness_score": 0}
+            elif "completeness_score" not in enhancement_report["original_score"]:
+                enhancement_report["original_score"]["completeness_score"] = enhancement_report["original_score"].get("total_score", 0)
+            
+            # Ensure final_score has completeness_score
+            if "final_score" not in enhancement_report or enhancement_report["final_score"] is None:
+                enhancement_report["final_score"] = {"total_score": 0, "completeness_score": 0}
+            elif "completeness_score" not in enhancement_report["final_score"]:
+                enhancement_report["final_score"]["completeness_score"] = enhancement_report["final_score"].get("total_score", 0)
+
+        # Add display names and tooltips for score sections
+        display_names = {
+            "required_fields": "Required Fields",
+            "metadata": "Metadata",
+            "component_basic": "Component Basic Info",
+            "component_model_card": "Model Card",
+            "external_references": "External References"
+        }
+        
+        tooltips = {
+            "required_fields": "Basic required fields for a valid AIBOM",
+            "metadata": "Information about the AIBOM itself",
+            "component_basic": "Basic information about the AI model component",
+            "component_model_card": "Detailed model card information",
+            "external_references": "Links to external resources"
+        }
+        
+        weights = {
+            "required_fields": 20,
+            "metadata": 20,
+            "component_basic": 20,
+            "component_model_card": 30,
+            "external_references": 10
+        }
 
         return templates.TemplateResponse(
             "result.html",
@@ -107,7 +244,10 @@ async def generate_form(
                 "enhancement_report": enhancement_report,
                 "completeness_score": completeness_score,
                 "download_url": download_url,
-                "download_script": download_script
+                "download_script": download_script,
+                "display_names": display_names,
+                "tooltips": tooltips,
+                "weights": weights
             }
         )
     except Exception as e: