Update app.py

app.py

@@ -255,8 +255,8 @@ PADDING = 8
 LABEL_DIR = "./labels"
 
 def generate_labels(font_file=None, num_labels: int = 25):
+    import time
     global font_path
-    os.makedirs(LABEL_DIR, exist_ok=True)
 
     try:
         if font_file:
@@ -264,36 +264,40 @@ def generate_labels(font_file=None, num_labels: int = 25):
             with open(font_file.name, "rb") as uploaded:
                 with open(font_path, "wb") as f:
                     f.write(uploaded.read())
-        if not font_file or not os.path.exists(font_path):
+
+        if font_path is None or not os.path.exists(font_path):
             font = ImageFont.load_default()
         else:
-            font = ImageFont.truetype(font_path, FONT_SIZE)
+            font = ImageFont.truetype(font_path, 32)
 
+        os.makedirs("./labels", exist_ok=True)
+        labels = ["".join(np.random.choice(list(CHARS), np.random.randint(4, 7))) for _ in range(num_labels)]
         images = []
-        labels = [
-            "".join(np.random.choice(list(CHARS), np.random.randint(4, 7)))
-            for _ in range(num_labels)
-        ]
 
         for label in labels:
-            # Measure actual text bounding box
-            dummy_img = Image.new("L", (1, 1))
-            draw = ImageDraw.Draw(dummy_img)
-            bbox = draw.textbbox((0, 0), label, font=font)
-
-            text_width = bbox[2] - bbox[0]
-            text_height = bbox[3] - bbox[1]
+            # Measure text size
+            bbox = font.getbbox(label)
+            text_w = bbox[2] - bbox[0]
+            text_h = bbox[3] - bbox[1]
 
-            # Compute final image size with padding
-            image_width = text_width + 2 * PADDING
-            image_height = text_height + 2 * PADDING
+            # Add 8px padding
+            pad = 8
+            img_w = text_w + pad * 2
+            img_h = text_h + pad * 2
 
-            img = Image.new("L", (image_width, image_height), color=255)
+            img = Image.new("L", (img_w, img_h), color=255)
             draw = ImageDraw.Draw(img)
-            draw.text((PADDING, PADDING), label, font=font, fill=0)
+            draw.text((pad, pad), label, font=font, fill=0)
+
+            # Save image to label-specific folder
+            safe_label = ''.join(c if c.isalnum() else '_' for c in label)
+            label_dir = f"./labels/{safe_label}"
+            os.makedirs(label_dir, exist_ok=True)
+
+            timestamp = time.strftime("%Y%m%d%H%M%S%f")
+            filepath = os.path.join(label_dir, f"_{timestamp}.png")
+            img.save(filepath)
 
-            filename = f"{LABEL_DIR}/{label}_{datetime.now().strftime('%Y%m%d%H%M%S%f')}.png"
-            img.save(filename)
             images.append(img)
 
         return images