Update deploying_3_3.py

deploying_3_3.py

@@ -93,8 +93,11 @@ This portion is used to convert vertices read from dxf to pixels in order to acc
 
 """PDF to image"""
 
-def pdftoimg(datadoc):
-  doc = fitz.open('pdf',datadoc)
+def pdftoimg(datadoc,pdf_content=0):
+  if pdf_content:
+     doc = fitz.open(stream=pdf_content, filetype="pdf")
+  else:
+     doc = fitz.open('pdf',datadoc)
   page=doc[0]
   pix = page.get_pixmap()  # render page to an image
   pl=Image.frombytes('RGB', [pix.width,pix.height],pix.samples)
@@ -125,9 +128,12 @@ def get_paper_size_in_inches(width, height):
             return size
     return "Unknown Size"
 
-def analyze_pdf(datadoc):
+def analyze_pdf(datadoc,pdf_content=0):
     # Open the PDF file
-    pdf_document = fitz.open('pdf',datadoc)
+    if pdf_content:
+       pdf_document = fitz.open(stream=pdf_content, filetype="pdf")
+    else:
+      pdf_document = fitz.open('pdf',datadoc)
 
     # Iterate through pages and print their sizes
     for page_number in range(len(pdf_document)):
@@ -182,9 +188,12 @@ def switch_case(argument):
 
 
 
-def RetriveRatio(datadoc,dxfpath):
-
-  width,height,paper_size = analyze_pdf (datadoc)
+def RetriveRatio(datadoc,dxfpath,pdf_content=0):
+  if pdf_content:
+     width,height,paper_size = analyze_pdf (datadoc,pdf_content)
+  else:
+     width,height,paper_size = analyze_pdf (datadoc)
+     
 
   if(width > height ):
     bigger=width
@@ -884,10 +893,12 @@ def rotate_polygon(polygon, angle, pdfrotation,width,height,center_point=(0, 0))
 #loop 3la hatched areas and count the occurences of each shape w create a table bl hagat di
 
 
-def Create_DF(dxfpath,datadoc,hatched_areas):
+def Create_DF(dxfpath,datadoc,hatched_areas,pdf_content=0):
     
- 
-  FinalRatio= RetriveRatio(datadoc,dxfpath)
+  if pdf_content:
+    FinalRatio= RetriveRatio(datadoc,dxfpath,pdf_content)
+  else:
+     FinalRatio= RetriveRatio(datadoc,dxfpath)
 
   # hatched_areas = get_hatched_areas(dxfpath,FinalRatio)
   # print('hatched_areas',hatched_areas)
@@ -952,25 +963,6 @@ def adjustannotations(OutputPdfStage1):
 
   # Iterate over pages
   for page_index, page in enumerate(writer.pages):
-      
-      # page.update({
-      #           NameObject("/UserUnit"): FloatObject(1.0),  # 1 unit = 1 real-world unit (e.g., 1 meter)
-      #           NameObject("/VP"): ArrayObject([
-      #               DictionaryObject({
-      #                   NameObject("/Type"): NameObject("/Viewport"),
-      #                   NameObject("/BBox"): ArrayObject([
-      #                   FloatObject(0), FloatObject(0), FloatObject(1000), FloatObject(1000)
-      #                   ]),  # Bounding box for the viewport
-      #                   NameObject("/Measure"): DictionaryObject({
-      #                       NameObject("/Type"): NameObject("/Measure"),
-      #                       NameObject("/Subtype"): NameObject("/RL"),
-      #                       NameObject("/X"): FloatObject(1),
-      #                       NameObject("/Y"): FloatObject(1),
-      #                       NameObject("/U"): TextStringObject("m"),  # Units (meters)
-      #                   }),
-      #               })
-      #           ])
-      #       })
       if "/Annots" in page:
           annotations = page["/Annots"]
           for annot_index, annot in enumerate(annotations):
@@ -1009,94 +1001,34 @@ def adjustannotations(OutputPdfStage1):
   writer.write(output_pdf_io)
   output_pdf_io.seek(0)
 
+  print(f"Annotations updated and saved to {output_pdf_path}")
   return output_pdf_io.read()
 
-  # writer.write(new_bytes_object)  # This writes the modified PDF data to new_bytes_object
-  # new_bytes_object.seek(0)
-  # return new_bytes_object.read()
-
-# def adjustannotations(OutputPdfStage1):
-#     """
-#     Adjusts annotations in the PDF to include measurement and scale information.
-
-#     Parameters:
-#         OutputPdfStage1 (str): Path to the input PDF file.
-
-#     Returns:
-#         bytes: The adjusted PDF data as bytes.
-#     """
-#     with open(OutputPdfStage1, "rb") as pdf_file:
-#         reader = PdfReader(pdf_file)
-#         writer = PdfWriter()
-
-#         # Append all pages from reader to writer
-#         writer.append_pages_from_reader(reader)
-
-#         # Iterate over pages and add measurement details
-#         for page_index, page in enumerate(writer.pages):
-#             # Add scale settings at the page level
-#             # page.update({
-#             #     NameObject("/UserUnit"): FloatObject(1.0),  # 1 unit = 1 real-world unit
-#             #     NameObject("/VP"): ArrayObject([
-#             #         DictionaryObject({
-#             #             NameObject("/Type"): NameObject("/Viewport"),
-#             #             NameObject("/BBox"): ArrayObject([
-#             #                 FloatObject(0), FloatObject(0), FloatObject(100), FloatObject(100)
-#             #             ]),  # Bounding box for the viewport
-#             #             NameObject("/Measure"): DictionaryObject({
-#             #                 NameObject("/Type"): NameObject("/Measure"),
-#             #                 NameObject("/Subtype"): NameObject("/RL"),
-#             #                 NameObject("/X"): FloatObject(1),  # Horizontal scale
-#             #                 NameObject("/Y"): FloatObject(1),  # Vertical scale
-#             #                 NameObject("/U"): TextStringObject("m"),  # Units (meters)
-#             #             }),
-#             #         })
-#             #     ])
-#             # })
-
-#             # Process annotations
-#             if "/Annots" in page:
-#                 annotations = page["/Annots"]
-#                 for annot in annotations:
-#                     obj = annot.get_object()
-
-#                     # Adjust polygon annotations with area measurements
-#                     if obj.get("/Subtype") == "/Polygon" and "/Contents" in obj and "sq m" in obj["/Contents"]:
-#                         obj.update({
-#                             NameObject("/Measure"): DictionaryObject({
-#                                 NameObject("/Type"): NameObject("/Measure"),
-#                                 NameObject("/Area"): DictionaryObject({
-#                                     NameObject("/G"): FloatObject(1),
-#                                     NameObject("/U"): TextStringObject("sq m"),  # Area unit
-#                                 }),
-#                                 NameObject("/X"): FloatObject(1),  # Horizontal scale
-#                                 NameObject("/Y"): FloatObject(1),  # Vertical scale
-#                                 NameObject("/U"): TextStringObject("m"),  # Units (meters)
-#                             }),
-#                             NameObject("/IT"): NameObject("/Area_Annotation"),
-#                             NameObject("/Subj"): TextStringObject("Area Measurement"),
-#                         })
-
-#         print(obj)
-#         output_pdf_io = BytesIO()
-#         writer.write(output_pdf_io)
-#         output_pdf_io.seek(0)  # Ensure buffer is at the start
-#         return output_pdf_io
+  
 
     
-def mainFunctionDrawImgPdf(datadoc,dxfpath, dxfratio,pdfpath=0,pdfname=0):
+def mainFunctionDrawImgPdf(datadoc,dxfpath, dxfratio,pdfpath=0,pdfname=0,pdf_content=0):
   OutputPdfStage1='BB Trial.pdf'
-  FinalRatio= RetriveRatio(datadoc,dxfpath)
+  if pdf_content:
+     FinalRatio= RetriveRatio(datadoc,dxfpath,pdf_content)
+  else:
+      FinalRatio= RetriveRatio(datadoc,dxfpath)
+    
 
   # hatched_areas = get_hatched_areas(dxfpath,FinalRatio)
   # hatched_areas=remove_duplicate_shapes(new_hatched_areas)
-
-  img=pdftoimg(datadoc)
+  if pdf_content:
+    img=pdftoimg(datadoc,pdf_content)
+  else:
+     img=pdftoimg(datadoc)
   flipped_horizontal=flip(img)
   allcnts = []
   imgg = flipped_horizontal
   # imgtransparent1=imgg.copy()
-  doc = fitz.open('pdf',datadoc)
+  if pdf_content:
+    doc = fitz.open(stream=pdf_content, filetype="pdf")
+  else:
+    doc = fitz.open('pdf',datadoc)
   page2 = doc[0]
   rotationOld=page2.rotation
   derotationMatrix=page2.derotation_matrix
@@ -1118,7 +1050,10 @@ def mainFunctionDrawImgPdf(datadoc,dxfpath, dxfratio,pdfpath=0,pdfname=0):
   allshapes=[]
   # Iterate through each polygon in metric units
   NewColors = []
-  SimilarAreaDictionary=Create_DF(dxfpath,datadoc,hatched_areas)
+  if pdf_content:
+    SimilarAreaDictionary=Create_DF(dxfpath,datadoc,hatched_areas,pdf_content)
+  else:
+     SimilarAreaDictionary=Create_DF(dxfpath,datadoc,hatched_areas)
   i=0
   flagcolor = 0
   ColorCheck=[]
@@ -1328,8 +1263,10 @@ def mainFunctionDrawImgPdf(datadoc,dxfpath, dxfratio,pdfpath=0,pdfname=0):
   # doc2 = fitz.open(stream=OutputPdfStage2, filetype="pdf")
       
   doc2 =fitz.open('pdf',OutputPdfStage2)
-    
-  gc,spreadsheet_service,spreadsheetId, spreadsheet_url , namepathArr=google_sheet_Legend.legendGoogleSheets(SimilarAreaDictionary , pdfname,pdfpath)
+  if pdf_content:
+     gc,spreadsheet_service,spreadsheetId, spreadsheet_url , namepathArr=google_sheet_Legend.legendGoogleSheets(SimilarAreaDictionary , pdfname,pdfpath,pdf_content)  
+  else:
+    gc,spreadsheet_service,spreadsheetId, spreadsheet_url , namepathArr=google_sheet_Legend.legendGoogleSheets(SimilarAreaDictionary , pdfname,pdfpath)
   # dbxTeam=tsadropboxretrieval.ADR_Access_DropboxTeam('user')
   # md, res =dbxTeam.files_download(path= pdfpath+pdfname)
   # data = res.content