Update doors_fasterrcnn.py

doors_fasterrcnn.py

@@ -223,7 +223,7 @@ def pxl2meter(width_pixels, ratio):
 def width_as_char(real_width):
   char_width = []
   for width in real_width:
-    char_width.append(f"{width}m")
+    char_width.append(f"{width}")
   return char_width
 
 def adjustannotations(OutputPdfStage1):
@@ -343,72 +343,6 @@ def add_annotations_to_pdf(image, pdf_name, lines, sanda, char_width, line_midpo
         annot.set_border(width=0.2, dashes=(1, 2))  # Optional border styling
         annot.set_colors(stroke=(1, 0, 0), fill=None)  # Set the stroke color to red
         annot.update()
-        
-    #for i in range(len(doubleD_bbox)):
-        # rect_coords should be a tuple: (x1, y1, x2, y2)
-        #tpl1 = (doubleD_bbox[i][0], doubleD_bbox[i][1])
-        #tpl2 = (doubleD_bbox[i][2], doubleD_bbox[i][3])
-        #tpl_derotated = fitz.Point(*tpl1) * derotationMatrix
-        #tpl2_derotated = fitz.Point(*tpl2) * derotationMatrix
-        #print("Derotated tuples:")
-        #print(tpl_derotated)
-        #print("Derotated tuples 2:")
-        #print(tpl2_derotated)
-        #coordinates_derotated = [tpl_derotated[0],tpl_derotated[1],tpl2_derotated[0],tpl2_derotated[1]]
-        #x1, y1 = tpl_derotated
-        #x2, y2 = tpl2_derotated
-        #x1 = int(x1)
-        #y1 = int(y1)
-        #x2 = int(x2)
-        #y2 = int(y2)
-        #x1, x2 = min(x1, x2), max(x1, x2)
-        #y1, y2 = min(y1, y2), max(y1, y2)
-
-        
-        #print("x1:")
-        #print(x1)
-        #print("y1:")
-        #print(y1)
-        #print("x2:")
-        #print(x2)
-        #print("y2:")
-        #print(y2)
-        
-        #rect = fitz.Rect(x1, y1, x2, y2)
-        #print("RECT:")
-        #print(rect)
-        #rect_annot = page.add_rect_annot(rect)
-        #rect_annot.set_border(width=2)  # Adjust border width
-        #rect_annot.set_colors(stroke=(1, 0, 0))  # Red rectangle
-        #rect_annot.update()
-
-        # annot = page.add_line_annot(lines[i][0], lines[i][1])
-        # # annot = page.add_line_annot(sanda[i][0], lines[i][0])
-        # # annot = page.add_line_annot(sanda[i][1], lines[i][1])
-        # annot.set_border(width=2, dashes=None)  # Optional border styling
-        # annot.set_colors(stroke=(1, 0, 0))  # Set the line color to red
-        # annot.update()
-       
-      #annot = page.add_line_annot(fitz.Point(*lines[i][0]) * derotationMatrix, fitz.Point(*lines[i][1]) * derotationMatrix)
-      #annot.set_border(width=2, dashes=None)  # Optional border styling
-      #annot.set_colors(stroke=(1, 0, 0))  # Set the line color to red
-      #annot.set_info(content=str(1.0)+' m',subject='Perimeter Measurement', title="ADR Team")
-      #annot.update()
-
-      #annot = page.add_line_annot(fitz.Point(*sanda[i][0]) * derotationMatrix, fitz.Point(*lines[i][0]) * derotationMatrix)
-      #annot.set_border(width=2, dashes=None)  # Optional border styling
-      #annot.set_colors(stroke=(1, 0, 0))  # Set the line color to red
-      #annot.set_info(content=str(1.0)+' m',subject='Perimeter Measurement', title="ADR Team")
-      #annot.update()
-
-      #annot = page.add_line_annot(fitz.Point(*sanda[i][1]) * derotationMatrix, fitz.Point(*lines[i][1]) * derotationMatrix)
-      #annot.set_border(width=2, dashes=None)  # Optional border styling
-      #annot.set_colors(stroke=(1, 0, 0))  # Set the line color to red
-      #annot.set_info(content=str(1.0)+' m',subject='Perimeter Measurement', title="ADR Team")
-      #annot.update()
-    # output_pdf_path = pdf_name+"_annotated.pdf"
-    # Save the PDF with annotations
-
     page.set_rotation(rotationOld)
     return pdf_document
     
@@ -463,12 +397,29 @@ def main_run(img_pillow,pdf_fullpath, weights_path, pdf_name,pdfpath,ratio): ###
 
     df_doors = df_doors.fillna(' ')
     gc,spreadsheet_service,spreadsheetId, spreadsheet_url , namepathArr=google_sheet_Legend.legendGoogleSheets(df_doors , pdf_name,pdfpath)
-    list1=pd.DataFrame(columns=['content',  'id',  'subject'])
-    for page in pdf_document:
-        for annot in page.annots():
-            list1.loc[len(list1)] =annot.info
-
-
+    list1=pd.DataFrame(columns=['content',  'id',  'subject','color'])
+
+    # for page in doc:
+    for page in doc2:  
+      # Iterate through annotations on the page
+      for annot in page.annots():
+          # Get the color of the annotation
+          annot_color = annot.colors
+          if annot_color is not None:
+              # annot_color is a dictionary with 'stroke' and 'fill' keys
+              stroke_color = annot_color.get('stroke')  # Border color
+              print('strokeee',stroke_color)
+              # fill_color = annot_color.get('fill')      # Fill color
+              # if fill_color:
+              #   v='fill'
+                # print('fill')
+              if stroke_color:
+                v='stroke'
+              # x,y,z=int(annot_color.get(v)[0]*255),int(annot_color.get(v)[1]*255),int(annot_color.get(v)[2]*255)
+              list1.loc[len(list1)] =[annot.info['content'],annot.info['id'],annot.info['subject'],[255,0,0]]
+
+
+    print('list1',list1)
     # modify this return
     # OutputPdfStage2=OutputPdfStage2.read()
     # doc2 =fitz.open('pdf',OutputPdfStage2)