Update app.py

app.py

@@ -62,26 +62,30 @@ def get_pose(image):
 
 
 def generate_an_image_from_text(text, text_size_, width, lenght):
-  # Create a blank image
-  image = Image.new('RGB', (width, lenght), color = (255, 255, 255))
-  # Create a drawing object
-  draw = ImageDraw.Draw(image)
-  # font def 
-  font_dir = '/usr/share/fonts/truetype/liberation/'
-  # Get a list of all the font files in the directory
-  font_files = [os.path.join(font_dir, f) for f in os.listdir(font_dir) if os.path.isfile(os.path.join(font_dir, f))]
-  # Select a random font
-  font_path = random.choice(font_files)
-  #print(font_path)
-  font = ImageFont.truetype(font_path, text_size_)
-  # Get the text size
-  text_size = draw.textsize(text, font)
-  # Calculate the x and y positions for the text
-  x = (image.width - text_size[0]) / 2
-  y = (image.height - text_size[1]) / 2
-  # Draw the text on the image
-  draw.text((x, y), text, fill=(0, 0, 0), font=font)
-  return image
+    # Create a blank image
+    image = Image.new('RGB', (width, lenght), color = (255, 255, 255))
+    # Create a drawing object
+    draw = ImageDraw.Draw(image)
+    # font def 
+    font_dir = ''
+    # Get a list of all the font files in the directory
+    font_files = glob.glob(os.path.join(dir_path, 'fonts', '*.ttf'))
+    # Get a list of font paths
+    font_paths = []
+    for font_file in font_files:
+        font_paths.append(font_file)
+    # Select a random font
+    font_path = random.choice(font_paths)
+    #print(font_path)
+    font = ImageFont.truetype(font_path, text_size_)
+    # Get the text size
+    text_size = draw.textsize(text, font)
+    # Calculate the x and y positions for the text
+    x = (image.width - text_size[0]) / 2
+    y = (image.height - text_size[1]) / 2
+    # Draw the text on the image
+    draw.text((x, y), text, fill=(0, 0, 0), font=font)
+    return image
 
 def to_Canny(image):
   # Let's load the popular vermeer image