Update InitialMarkups.py

InitialMarkups.py

@@ -1040,29 +1040,35 @@ def extract_section_under_header(pdf_path):
     json_output=changepdflinks(json_output,pdflink)
     return pdf_bytes.getvalue(), docHighlights , json_output
 
-def changepdflinks(data_list_JSON,pdflink):
+
+def changepdflinks(data_list_JSON, pdflink):
+    # If the input is a JSON string, convert it to a Python list
+    if isinstance(data_list_JSON, str):
+        data_list_JSON = json.loads(data_list_JSON)
+
     # Loop through all entries and update their NBSLink
     for entry in data_list_JSON:
-        old_url = entry["NBSLink"]
+        old_url = entry.get("NBSLink", "")
 
-        # Parse the URL and query params
+        # Parse URL and query params
         parsed = urllib.parse.urlparse(old_url)
         query = urllib.parse.parse_qs(parsed.query)
 
-        # Replace only the 'pdfLink' parameter
+        # Replace only the 'pdfLink' parameter if present
         if "pdfLink" in query:
             query["pdfLink"] = [pdflink]
 
-            # Rebuild the query string
+            # Rebuild query string
             new_query = urllib.parse.urlencode(query, doseq=True)
 
-            # Rebuild the final URL with the same base, page, and zoom fragment
+            # Rebuild full URL with the same fragment (page/zoom)
             new_url = f"{parsed.scheme}://{parsed.netloc}{parsed.path}?{new_query}"
             if parsed.fragment:
                 new_url += f"#{parsed.fragment}"
 
             # Update the entry
             entry["NBSLink"] = new_url
+
     return data_list_JSON