Update src/aibom_generator/api.py

src/aibom_generator/api.py

@@ -62,7 +62,8 @@ def log_sbom_generation(model_id: str):
         # Try to load existing dataset to append
         try:
             # Use trust_remote_code=True if required by the dataset/model on HF
-            existing_ds = load_dataset(HF_REPO, token=HF_TOKEN, split=\'train\', trust_remote_code=True)
+            # Corrected: Removed unnecessary backslashes around 'train'
+            existing_ds = load_dataset(HF_REPO, token=HF_TOKEN, split='train', trust_remote_code=True)
             # Check if dataset is empty or has different columns (handle initial creation)
             if len(existing_ds) > 0 and set(existing_ds.column_names) == set(log_data.keys()):
                  ds_to_push = concatenate_datasets([existing_ds, ds_new_log])
@@ -76,12 +77,14 @@ def log_sbom_generation(model_id: str):
                  ds_to_push = concatenate_datasets([existing_ds, ds_new_log])
 
         except Exception as load_err:
-             # Handle case where dataset doesn\'t exist yet or other loading errors
+             # Handle case where dataset doesn't exist yet or other loading errors
+             # Corrected: Removed unnecessary backslash in doesn't
              logger.info(f"Could not load existing dataset {HF_REPO} (may not exist yet): {load_err}. Pushing new dataset.")
              ds_to_push = ds_new_log # ds is already prepared with the new log entry
 
         # Push the updated or new dataset
-        ds_to_push.push_to_hub(HF_REPO, token=HF_TOKEN, private=True) # Ensure it\'s private
+        # Corrected: Removed unnecessary backslash in it's
+        ds_to_push.push_to_hub(HF_REPO, token=HF_TOKEN, private=True) # Ensure it's private
         logger.info(f"Successfully logged SBOM generation for {model_id} to {HF_REPO}")
 
     except Exception as e:
@@ -93,9 +96,10 @@ def get_sbom_count() -> str:
         logger.warning("HF_TOKEN not set. Cannot retrieve SBOM count.")
         return "N/A"
     try:
-        # Load the dataset - assumes \'train\' split exists after first push
+        # Load the dataset - assumes 'train' split exists after first push
         # Use trust_remote_code=True if required by the dataset/model on HF
-        ds = load_dataset(HF_REPO, token=HF_TOKEN, split=\'train\', trust_remote_code=True)
+        # Corrected: Removed unnecessary backslashes around 'train'
+        ds = load_dataset(HF_REPO, token=HF_TOKEN, split='train', trust_remote_code=True)
         count = len(ds)
         logger.info(f"Retrieved SBOM count: {count} from {HF_REPO}")
         # Format count for display (e.g., add commas for large numbers)
@@ -374,7 +378,8 @@ async def generate_form(
         enhancement_report = generator.get_enhancement_report()
 
         # Save AIBOM to file
-        filename = f"{model_id.replace(\'/\', \'_\')}_aibom.json"
+        # Corrected: Removed unnecessary backslashes around '/' and '_'
+        filename = f"{model_id.replace('/', '_')}_aibom.json"
         filepath = os.path.join(OUTPUT_DIR, filename)
 
         with open(filepath, "w") as f:
@@ -388,12 +393,13 @@ async def generate_form(
         download_url = f"/output/{filename}"
 
         # Create download and UI interaction scripts
+        # Corrected: Removed unnecessary backslashes in script string
         download_script = f"""
         <script>
             function downloadJSON() {{
-                const a = document.createElement(\'a\');
-                a.href = \'{download_url}\';
-                a.download = \'{filename}\';
+                const a = document.createElement('a');
+                a.href = '{download_url}';
+                a.download = '{filename}';
                 document.body.appendChild(a);
                 a.click();
                 document.body.removeChild(a);
@@ -401,31 +407,31 @@ async def generate_form(
 
             function switchTab(tabId) {{
                 // Hide all tabs
-                document.querySelectorAll(\'.tab-content\').forEach(tab => {{
-                    tab.classList.remove(\'active\');
+                document.querySelectorAll('.tab-content').forEach(tab => {{
+                    tab.classList.remove('active');
                 }});
 
                 // Deactivate all tab buttons
-                document.querySelectorAll(\'.aibom-tab\').forEach(button => {{
-                    button.classList.remove(\'active\');
+                document.querySelectorAll('.aibom-tab').forEach(button => {{
+                    button.classList.remove('active');
                 }});
 
                 // Show the selected tab
-                document.getElementById(tabId).classList.add(\'active\');
+                document.getElementById(tabId).classList.add('active');
 
                 // Activate the clicked button
-                event.currentTarget.classList.add(\'active\');
+                event.currentTarget.classList.add('active');
             }}
 
             function toggleCollapsible(element) {{
-                element.classList.toggle(\'active\');
+                element.classList.toggle('active');
                 var content = element.nextElementSibling;
                 if (content.style.maxHeight) {{
                     content.style.maxHeight = null;
-                    content.classList.remove(\'active\');
+                    content.classList.remove('active');
                 }} else {{
                     content.style.maxHeight = content.scrollHeight + "px";
-                    content.classList.add(\'active\');
+                    content.classList.add('active');
                 }}
             }}
         </script>
@@ -433,15 +439,17 @@ async def generate_form(
 
         # Get completeness score or create a comprehensive one if not available
         completeness_score = None
-        if hasattr(generator, \'get_completeness_score\'):
+        # Corrected: Removed unnecessary backslash in 'get_completeness_score'
+        if hasattr(generator, 'get_completeness_score'):
             try:
                 completeness_score = generator.get_completeness_score(model_id)
                 logger.info("Successfully retrieved completeness_score from generator")
             except Exception as e:
                 logger.error(f"Completeness score error from generator: {str(e)}")
 
-        # If completeness_score is None or doesn\'t have field_checklist, use comprehensive one
-        if completeness_score is None or not isinstance(completeness_score, dict) or \'field_checklist\' not in completeness_score:
+        # If completeness_score is None or doesn't have field_checklist, use comprehensive one
+        # Corrected: Removed unnecessary backslash in doesn't and 'field_checklist'
+        if completeness_score is None or not isinstance(completeness_score, dict) or 'field_checklist' not in completeness_score:
             logger.info("Using comprehensive completeness_score with field_checklist")
             completeness_score = create_comprehensive_completeness_score(aibom)