From 815dcd33f43b23737ce2c4c18e87d68f45e624a8 Mon Sep 17 00:00:00 2001
From: Wang Guan <momocraft@gmail.com>
Date: Sun, 17 Nov 2024 23:06:42 +0900
Subject: [PATCH 1/3] workaround empty textline

---
 manga_translator/utils/textblock.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/manga_translator/utils/textblock.py b/manga_translator/utils/textblock.py
index de3c8a4c..8c2f390f 100644
--- a/manga_translator/utils/textblock.py
+++ b/manga_translator/utils/textblock.py
@@ -83,7 +83,7 @@ def __init__(self, lines: List[Tuple[int, int, int, int]],
         if self.text and len(texts) > 1:
             for txt in texts[1:]:
                 first_cjk = '\u3000' <= self.text[-1] <= '\u9fff'
-                second_cjk = '\u3000' <= txt[0] <= '\u9fff'
+                second_cjk = txt and ('\u3000' <= txt[0] <= '\u9fff')
                 if first_cjk or second_cjk :
                     self.text += txt
                 else :

From d8458f9b75f1133f679157b3352dfd8b0bf9a479 Mon Sep 17 00:00:00 2001
From: Wang Guan <momocraft@gmail.com>
Date: Sun, 17 Nov 2024 23:16:30 +0900
Subject: [PATCH 2/3] docker_prepare: allow set of models/continue_on_error

---
 .dockerignore     |  1 +
 Dockerfile        |  2 +-
 docker_prepare.py | 65 +++++++++++++++++++++++++++++++++--------------
 3 files changed, 48 insertions(+), 20 deletions(-)

diff --git a/.dockerignore b/.dockerignore
index 4bfc2510..ce674a1b 100644
--- a/.dockerignore
+++ b/.dockerignore
@@ -8,3 +8,4 @@ ocrs
 models/*
 test/testdata/bboxes
 /venv
+.git
diff --git a/Dockerfile b/Dockerfile
index 4249d46b..f94d11ac 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -22,7 +22,7 @@ RUN apt-get remove -y g++ && \
 COPY . /app
 
 # Prepare models
-RUN python -u docker_prepare.py
+RUN python -u docker_prepare.py --continue-on-error
 
 RUN rm -rf /tmp
 
diff --git a/docker_prepare.py b/docker_prepare.py
index 3a6e79cc..5acbabf0 100644
--- a/docker_prepare.py
+++ b/docker_prepare.py
@@ -1,28 +1,55 @@
 import asyncio
-
+from argparse import ArgumentParser
 from manga_translator.utils import ModelWrapper
 from manga_translator.detection import DETECTORS
 from manga_translator.ocr import OCRS
 from manga_translator.inpainting import INPAINTERS
 
+
+arg_parser = ArgumentParser()
+arg_parser.add_argument("--models", default="")
+arg_parser.add_argument("--continue-on-error", action="store_true")
+
+
+cli_args = arg_parser.parse_args()
+
+
 async def download(dict):
-  for key, value in dict.items():
-    if issubclass(value, ModelWrapper):
-      print(' -- Downloading', key)
-      try:
-        inst = value()
-        await inst.download()
-      except Exception as e:
-        print('Failed to download', key, value)
-        print(e)
+    """ """
+    for key, value in dict.items():
+        if issubclass(value, ModelWrapper):
+            print(" -- Downloading", key)
+            try:
+                inst = value()
+                await inst.download()
+            except Exception as e:
+                print("Failed to download", key, value)
+                print(e)
+                if not cli_args.continue_on_error:
+                    raise
+
 
 async def main():
-  await download(DETECTORS)
-  await download(OCRS)
-  await download({
-    k: v for k, v in INPAINTERS.items() 
-      if k not in ['sd']
-  })
-
-if __name__ == '__main__':
-  asyncio.run(main())
+    models: set[str] = set(filter(None, cli_args.models.split(",")))
+
+    await download(
+        {
+            k: v
+            for k, v in DETECTORS.items()
+            if (not models) or (f"detector.{k}" in models)
+        }
+    )
+    await download(
+        {k: v for k, v in OCRS.items() if (not models) or (f"ocr.{k}" in models)}
+    )
+    await download(
+        {
+            k: v
+            for k, v in INPAINTERS.items()
+            if (not models) or (f"inpaint.{k}" in models) and (k not in ["sd"])
+        }
+    )
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

From 16d86a5a55e7c342ff199604e54933700ec036fb Mon Sep 17 00:00:00 2001
From: Wang Guan <momocraft@gmail.com>
Date: Sun, 17 Nov 2024 23:25:05 +0900
Subject: [PATCH 3/3] type annotations

---
 manga_translator/utils/generic.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/manga_translator/utils/generic.py b/manga_translator/utils/generic.py
index 67163b69..92c55c6a 100644
--- a/manga_translator/utils/generic.py
+++ b/manga_translator/utils/generic.py
@@ -1,5 +1,5 @@
 import os
-from typing import List, Callable, Tuple
+from typing import List, Callable, Tuple, Optional
 import numpy as np
 import cv2
 import functools
@@ -65,7 +65,7 @@ def _get_args(self):
 
 # TODO: Add TranslationContext for type linting
 
-def atoi(text):
+def atoi(text: str) -> int | str:
     return int(text) if text.isdigit() else text
 
 def natural_sort(l: List[str]):
@@ -246,7 +246,7 @@ def __call__(self, val = None):
         else:
             return 0
 
-def load_image(img: Image.Image):
+def load_image(img: Image.Image) -> Tuple[np.ndarray, Optional[Image.Image]]:
     if img.mode == 'RGBA':
         # from https://stackoverflow.com/questions/9166400/convert-rgba-png-to-rgb-with-pil
         img.load()  # needed for split()