Update app.py

app.py

@@ -13,6 +13,8 @@ from torchvision.transforms.functional import to_pil_image
 import matplotlib.pyplot as plt
 import math
 from datetime import datetime
+import re
+
 # --------- Globals --------- #
 CHARS = string.ascii_letters + string.digits + string.punctuation
 CHAR2IDX = {c: i + 1 for i, c in enumerate(CHARS)}
@@ -87,6 +89,9 @@ class OCRModel(nn.Module):
         return x
 
 
+def sanitize_filename(name):
+    return re.sub(r'[^a-zA-Z0-9_-]', '_', name)
+
 def greedy_decode(log_probs):
     # log_probs shape: (T, B, C)
     # Usually, B=1 during inference
@@ -255,7 +260,6 @@ PADDING = 8
 LABEL_DIR = "./labels"
 
 def generate_labels(font_file=None, num_labels: int = 25):
-    import time
     global font_path
 
     try:
@@ -264,7 +268,6 @@ 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 font_path is None or not os.path.exists(font_path):
             font = ImageFont.load_default()
         else:
@@ -275,27 +278,26 @@ def generate_labels(font_file=None, num_labels: int = 25):
         images = []
 
         for label in labels:
-            # Measure text size
+            # Measure text size and calculate padded image dimensions
             bbox = font.getbbox(label)
             text_w = bbox[2] - bbox[0]
             text_h = bbox[3] - bbox[1]
-
-            # Add 8px padding
             pad = 8
             img_w = text_w + pad * 2
             img_h = text_h + pad * 2
 
+            # Create image and draw text
             img = Image.new("L", (img_w, img_h), color=255)
             draw = ImageDraw.Draw(img)
             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}"
+            # Save to ./labels/sanitized_label/timestamp.png
+            safe_label = sanitize_filename(label)
+            timestamp = datetime.now().strftime("%Y%m%d%H%M%S%f")
+            label_dir = os.path.join("./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")
+            filepath = os.path.join(label_dir, f"{timestamp}.png")
             img.save(filepath)
 
             images.append(img)
@@ -309,10 +311,23 @@ def generate_labels(font_file=None, num_labels: int = 25):
         draw.text((10, 50), f"Error: {str(e)}", fill=(255, 0, 0))
         return [error_img]
 
-
+custom_css = """
+#label-gallery .gallery-item img {
+    height: 43px;           /* 32pt ≈ 43px */
+    width: auto;
+    object-fit: contain;
+    padding: 4px;
+}
+
+#label-gallery {
+    flex-grow: 1;
+    overflow-y: auto;
+    height: 100%;
+}
+"""
 
 # --------- Updated Gradio UI with new tab --------- #
-with gr.Blocks() as demo:
+with gr.Blocks(css=custom_css) as demo:
     with gr.Tab("1. Upload Font & Train"):
         font_file = gr.File(label="Upload .ttf or .otf font", file_types=[".ttf", ".otf"])
         epochs_input = gr.Slider(minimum=1, maximum=4096, value=256, step=1, label="Epochs")