Add in font downloader

Author: VikParuchuri

surya/postprocessing/text.py

@@ -1,3 +1,6 @@
+import os
+
+import requests
 from PIL import Image, ImageDraw, ImageFont
 from surya.settings import settings
 
@@ -21,6 +24,14 @@ def draw_text_on_image(bboxes, texts, image_size=(1024, 1024), font_path=setting
 
         # Shrink the text to fit in the bbox if needed
         box_font_size = font_size
+
+        # Download font if it doesn't exist
+        if not os.path.exists(font_path):
+            with requests.get(settings.FONT_DL_PATH, stream=True) as r, open(font_path, 'wb') as f:
+                r.raise_for_status()
+                for chunk in r.iter_content(chunk_size=8192):
+                    f.write(chunk)
+
         font = ImageFont.truetype(font_path, box_font_size)
         text_width, text_height = get_text_size(text, font)
         while (text_width > bbox_width or text_height > bbox_height) and box_font_size > 6:

surya/settings.py

@@ -17,6 +17,7 @@ class Settings(BaseSettings):
     RESULT_DIR: str = "results"
     BASE_DIR: str = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
     FONT_DIR: str = os.path.join(BASE_DIR, "static", "fonts")
+    FONT_DL_PATH: str = "https://github.com/satbyy/go-noto-universal/releases/download/v7.0/GoNotoKurrent-Regular.ttf"
 
     @computed_field
     def TORCH_DEVICE_MODEL(self) -> str: