Update app.py

app.py

@@ -3,7 +3,37 @@ import os
 import zipfile
 from PIL import Image, ImageDraw, ImageFont
 
-def draw_ruler(image, dpi=72, font_size=144):
+def draw_ruler(image, dpi=72, unit="in", color="black"):
+    """
+    Draws a ruler on the top and left edges of an image.
+
+    Args:
+        image: The PIL Image object to draw on.
+        dpi: The DPI of the image (default 300).
+        unit: The unit of measurement (default "in").
+        color: The color of the ruler (default "black").
+    """
+
+    draw = ImageDraw.Draw(image)
+    width, height = image.size
+
+    # Draw top ruler
+    for i in range(0, width, int(dpi)):
+        x = i
+        draw.line((x, 0, x, 10), fill=color)
+        if i % int(dpi) == 0:
+            draw.text((x, 12), str(i // dpi), fill=color)
+
+    # Draw left ruler
+    for i in range(0, height, int(dpi)):
+        y = i
+        draw.line((0, y, 10, y), fill=color)
+        if i % int(dpi) == 0:
+            draw.text((12, y), str(i // dpi), fill=color)
+
+    return image
+    
+def draw_ruler1(image, dpi=72, font_size=144):
     """
     Draw a virtual ruler 10 cm from the bottom of the image.
     Args: