Update Doors_Schedule.py

Doors_Schedule.py

@@ -282,7 +282,7 @@ def get_st_op_pattern(clmn_name):
             return name
     return None
 
-def get_similar_colors(selected_columns_new):
+'''def get_similar_colors(selected_columns_new):
   def generate_rgb():
       return (random.randint(0, 255), random.randint(0, 255), random.randint(0, 255))  # RGB tuple
 
@@ -305,6 +305,33 @@ def get_similar_colors(selected_columns_new):
         col_dict[key]['heights'].append(row['height'])
       col_dict[key]['color'] = key_colors[key]  # Assign the unique RGB color
 
+  # Convert defaultdict to a normal dictionary
+  col_dict = dict(col_dict)
+  return col_dict'''
+
+def get_similar_colors(selected_columns_new):
+  def generate_rgb():
+      return (random.randint(0, 255), random.randint(0, 255), random.randint(0, 255))  # RGB tuple
+
+  unique_keys = selected_columns_new['door_type'].unique()
+  key_colors = {key: generate_rgb() for key in unique_keys}  # Assign a unique RGB color to each key
+
+  # Create dictionary storing values, colors, and widths
+  if 'structural_opening' in selected_columns_new.columns:
+      col_dict = defaultdict(lambda: {'values': [], 'color': None, 'widths': []})
+  else:
+      col_dict = defaultdict(lambda: {'values': [], 'color': None, 'widths': [], 'heights': []})
+
+  for _, row in selected_columns_new.iterrows():
+      key = row['door_type']
+      col_dict[key]['values'].append(row['door_id'])
+      if 'structural_opening' in selected_columns_new.columns:
+        col_dict[key]['widths'].append(row['structural_opening'])  # Add structural opening
+      else:
+        col_dict[key]['widths'].append(row['width'])  # Assuming 'widht' is a typo for 'width'
+        col_dict[key]['heights'].append(row['height'])
+      col_dict[key]['color'] = key_colors[key]  # Assign the unique RGB color
+
   # Convert defaultdict to a normal dictionary
   col_dict = dict(col_dict)
   return col_dict