Update Doors_Schedule.py

Doors_Schedule.py

@@ -495,14 +495,15 @@ def get_selected_columns_all(dfs, user_patterns):
         #print(len(clm_idx))
         #details in another table
         print(column_index_list)
-        if len(dfs[i]) <10:
-          selected_columns_new = details_in_another_table(clmn_name, column_index_list, dfs[i], dfs)
+        #if len(dfs[i]) <10:
           #break
           #other_matches = details_in_another_table_mod(clmn_name, clmn_idx, dfs[i], dfs)
         #details in the same table
-        if len(dfs[i]) >10:
-          selected_columns_new = generate_current_table_without_cropping(column_index_list,dfs[i])
-          selected_columns_new.columns = clmn_name  # must match number of columns
+        #if len(dfs[i]) >10:
+        selected_columns_new = details_in_another_table(clmn_name, column_index_list, dfs[i], dfs)
+        selected_columns_new2 = generate_current_table_without_cropping(column_index_list,dfs[i])
+        selected_columns_new =  pd.concat([selected_columns_new, selected_columns_new2], ignore_index=True)
+        selected_columns_new.columns = clmn_name  # must match number of columns
 
           #break
 
@@ -1832,7 +1833,11 @@ def mainRun(schedule, plan, searcharray, sch_csv_pdf):
                                          drop_empty_locations=True,
                                          explode_locations=True)
             df_points.columns = df_points.columns.str.strip().str.replace(r"\s+", "_", regex=True)
-
+            print(f"col_dict: {col_dict}")
+            print(f"selected_columns_combined: {selected_columns_combined}")
+            print(f"df: {df_points}")
+            if df_points.empty:
+              continue 
             # handling no door type in the new dictionary logic
             if 'color' not in df_points:
               df_points['color'] = (0, 0, 255)
@@ -1871,6 +1876,8 @@ def mainRun(schedule, plan, searcharray, sch_csv_pdf):
             #widht_count, height_count = generate_separate_dimensions(widths)
             #bax = create_bb_bax_secondary(new_data3, widht_count, height_count, secondary_tobeprinted, CountStyles, user_input, page_number, page)
             #bax_annotations_all_inputs.append(bax)
+            print(f"colors: {df_points['color']}")
+            print(f"df: {df_points}")
             bax, customDta = create_bb_bax_new(df_points, CountStyles, page_number, page)
             bax_annotations_all_inputs.append(bax)      
                                       
@@ -1915,6 +1922,7 @@ def mainRun(schedule, plan, searcharray, sch_csv_pdf):
     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 doc2:
+      print("now inside page in doc2")
       #  page=doc2[0]
       pix = page.get_pixmap()  # render page to an image
       pl=Image.frombytes('RGB', [pix.width,pix.height],pix.samples)
@@ -1922,11 +1930,16 @@ def mainRun(schedule, plan, searcharray, sch_csv_pdf):
       annotatedimg = cv2.cvtColor(img, cv2.COLOR_RGB2BGR)
       annotatedimgs.append(annotatedimg)
       # Iterate through annotations on the page
+      annotations_page = page.annots()
+      print(f"annotations: {annotations_page}")
+      '''
       for annot in page.annots():
           # Get the color of the annotation
+          print("ann: {annot}")
           annot_color = annot.colors
           if annot_color is not None:
               # annot_color is a dictionary with 'stroke' and 'fill' keys
+              print(annot_color)
               stroke_color = annot_color.get('stroke')  # Border color
               fill_color = annot_color.get('fill')      # Fill color
               if fill_color:
@@ -1935,6 +1948,11 @@ def mainRun(schedule, plan, searcharray, sch_csv_pdf):
               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)
+              print(f"x: {x}")
+              print(f"y: {y}")
+              print(f"z: {z}")
               list1.loc[len(list1)] =[annot.info['content'],annot.info['id'],annot.info['subject'],[x,y,z]]
+      print(f"list1 : {list1}")
+      '''
     return annotatedimgs, doc2 , list1, repeated_labels , not_found, pretty_xml, column_xml