Update Doors_Schedule.py

Doors_Schedule.py

@@ -1787,38 +1787,34 @@ def mainRun(schedule, plan, searcharray):
 
 
     ##### SHOULD return pretty_xml, column_xml, merged_pdf
-
-    return pretty_xml, column_xml, merged_pdf
-    
-'''    not_found = []
+    not_found = []
+    annotatedimgs=[]
     doc2 =fitz.open('pdf',merged_pdf)
     len_doc2 = len(doc2)
-    print(f"number of pges of doc2 is {len_doc2} and its type is {type(doc2)}")
-    page=doc2[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)
-    
-    
     list1=pd.DataFrame(columns=['content',  'id',  'subject','color'])
+    print(f"number of pges of doc2 is {len_doc2} and its type is {type(doc2)}")
+    for page in len_doc2:
+      #  page=doc2[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)
+      annotatedimgs.append(annotatedimg)
+      # 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
+              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'],[x,y,z]]
+    return annotatedimgs, merged_pdf , list1, repeated_labels , not_found, pretty_xml, column_xml, merged_pdf
     
-    # 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
-                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'],[x,y,z]]
-    return annotatedimg, doc2 , list1, repeated_labels , not_found 
-'''
+#  return annotatedimg, doc2 , list1, repeated_labels , not_found