Update Counting_Columns_2_1.py

Counting_Columns_2_1.py

@@ -217,8 +217,28 @@ def mainfun(plan):
       columns_types_v = getColumnsTypesKeyValue(nearby, txtpts_ky_vlu)
       legend = generate_legend(columns_types_v)
   pdf_document = add_annotations_to_pdf(image, pdf_name, slctd_clm, columns_types_v)
-
-  return legend
+  page=pdf_document[0]
+  pix = page.get_pixmap()  # render page to an image
+  pl=Image.frombytes('RGB', [pix.width,pix.height],pix.samples)
+  img=np.array(pl)
+  annotatedimg = cv2.cvtColor(img, cv2.COLOR_RGB2BGR)
+    
+  legend = legend.fillna(' ')
+  gc,spreadsheet_service,spreadsheetId, spreadsheet_url , namepathArr=google_sheet_Legend.legendGoogleSheets(df_doors , pdf_name,pdfpath)
+  list1=pd.DataFrame(columns=['content',  'id',  'subject','color'])
+  for page in doc2:  
+      for annot in page.annots():
+          annot_color = annot.colors
+          if annot_color is not None:
+              stroke_color = annot_color.get('stroke')  # Border color
+              print('strokeee',stroke_color)
+              if stroke_color:
+                v='stroke'
+              list1.loc[len(list1)] =[annot.info['content'],annot.info['id'],annot.info['subject'],[255,0,0]]
+    
+    
+  print('list1',list1)
+  return annotatedimg, pdf_document , spreadsheet_url, list1, legend 
 
 '''def mainfun(plan):
   texts_from_pdf = get_text_from_pdf(plan)