labels assignment to all products

app.py

@@ -538,32 +538,27 @@ def render_html_page(chat_text, images_to_use):
     ]
     processed_links = [link for img in images_to_use for link in IMAGES[img]["links"]]
 
-    # Counter for total labels
-    total_labels_assigned = 0
-    MAX_TOTAL_LABELS = 3  # Maximum number of labels allowed
-
     # Initialize labels for all products
     labels = [""] * len(processed_prices)
 
-    # Assign "BEST VALUE!" to the lowest price
-    if processed_prices:
-        min_price_index = processed_prices.index(min(processed_prices))
+    # Group prices in subsets of 3 and assign labels
+    for group_start in range(0, len(processed_prices), 3):
+        subset_indices = list(
+            range(group_start, min(group_start + 3, len(processed_prices)))
+        )
+        subset_prices = [processed_prices[i] for i in subset_indices]
+
+        # Find the index of the lowest price in this subset
+        min_price_index = subset_indices[subset_prices.index(min(subset_prices))]
         labels[min_price_index] = "<b>BEST VALUE</b>"
-        total_labels_assigned += 1
-
-    # Assign random labels to others
-    other_indices = [i for i in range(len(processed_prices)) if i != min_price_index]
-    random.shuffle(other_indices)
-
-    for idx in other_indices:
-        if total_labels_assigned < MAX_TOTAL_LABELS:
-            label_choice = random.choice(
-                ["<b>BEST QUALITY</b>", "<b>FAST PRODUCTION</b>"]
-            )
-            labels[idx] = label_choice
-            total_labels_assigned += 1
-        else:
-            break
+
+        # Assign remaining labels randomly
+        remaining_indices = [i for i in subset_indices if i != min_price_index]
+        random.shuffle(remaining_indices)
+        labels[remaining_indices[0]] = "<b>BEST QUALITY</b>"
+        labels[remaining_indices[1]] = (
+            "<b>FAST PRODUCTION</b>" if len(remaining_indices) > 1 else ""
+        )
 
     # Generate HTML content
     for i in range(len(processed_images)):