Add Farsi Support (Fix BIDI Right-to-left Problem)

trdg/data_generator.py

@@ -26,6 +26,7 @@ def generate(
  cls,
  index: int,
  text: str,
+ language: str,
  font: str,
  out_dir: str,
  size: int,
@@ -249,6 +250,15 @@ def generate(
  # Generate name for resulting image #
  #####################################
  # We remove spaces if space_width == 0
+ if language in ("ar", "fa"):
+ from arabic_reshaper import ArabicReshaper
+ from bidi.algorithm import get_display
+
+ arabic_reshaper = ArabicReshaper()
+ text = " ".join(
+ [get_display(arabic_reshaper.reshape(w)) for w in text.split(" ")[::-1]]
+ )
+
  if space_width == 0:
  text = text.replace(" ", "")
  if name_format == 0:

trdg/run.py

@@ -416,7 +416,7 @@ def main():
  args.length, args.random, args.count, lang_dict
  )
 
- if args.language == "ar":
+ if args.language in ("ar", "fa"):
  from arabic_reshaper import ArabicReshaper
  from bidi.algorithm import get_display
 
@@ -441,6 +441,7 @@ def main():
  zip(
  [i for i in range(0, string_count)],
  strings,
+ [args.language] * string_count,
  [fonts[rnd.randrange(0, len(fonts))] for _ in range(0, string_count)],
  [args.output_dir] * string_count,
  [args.format] * string_count,
@@ -486,6 +487,13 @@ def main():
  label = strings[i]
  if args.space_width == 0:
  label = label.replace(" ", "")
+ if args.language in ("ar", "fa"):
+ label = " ".join(
+ [
+ get_display(arabic_reshaper.reshape(w))
+ for w in strings[i].split(" ")[::-1]
+ ]
+ )
  f.write("{} {}\n".format(file_name, label))