Update InitialMarkups.py

InitialMarkups.py

@@ -6,10 +6,9 @@ Automatically generated by Colab.
 Original file is located at
     https://colab.research.google.com/drive/12XfVkmKmN3oVjHhLVE0_GgkftgArFEK2
 """
+baselink='https://find-initialmarkups.hf.space/view-pdf?'
 
-pip install pymupdf
 
-pip install fuzzywuzzy
 
 from io import BytesIO
 import re
@@ -600,7 +599,7 @@ def same_start_word(s1, s2):
         return words1[0].lower() == words2[0].lower()
     return False
 
-baselink='https://marthee-nbslink.hf.space/view-pdf?'
+
 def extract_section_under_header(pdf_path):
     top_margin = 70
     bottom_margin = 50
@@ -968,11 +967,9 @@ def extract_section_under_header(pdf_path):
                 page_highlights[page_num] = bbox
             highlight_boxes(docHighlights, page_highlights)
 
-    docHighlights.save("highlighted_output.pdf", garbage=4, deflate=True)
-    return json_output
-
-pdflink='https://www.dropbox.com/scl/fi/jtffcxszwpcnc6wdo61p6/WH007-JAC-RP-XX-SP-AA-8501-Redoak-Pump-House-Specification.pdf?rlkey=unq4ag9eajezv2j6y6ewkkk5u&e=29&st=wu3vsd70&dl=0'
-
-jsonOutput=extract_section_under_header(pdflink)
-print(jsonOutput)
+    # docHighlights.save("highlighted_output.pdf", garbage=4, deflate=True)
 
+    pdf_bytes = BytesIO()
+    docHighlights.save(pdf_bytes)
+    print('JSONN',json_output)
+    return pdf_bytes.getvalue(), docHighlights , json_output