plugins/ia_writer.py: fix for .PNG etc

mmagnus · Feb 18, 2022 · 3bc75c0 · 3bc75c0
1 parent 54e206c
commit 3bc75c0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/engine/plugins/ia_writer.py b/engine/plugins/ia_writer.py
@@ -54,7 +54,7 @@ def edit_syntax_from_ai_writer_to_geekbook(text, img_prefix):
         img_prefix += '/'
 
     for line in textlist:
-        if line.startswith('/') and (line.endswith('.png') or line.endswith('.jpeg') or line.endswith('.gif')):
+        if line.startswith('/') and (line.lower().endswith('.png') or line.lower().endswith('.jpg') or line.lower().endswith('.jpeg') or line.lower().endswith('.gif')):
             pfile = line.replace('/', '')
             pfile_fullpath = PATH_TO_IMG + os.sep + IMG_PREFIX + os.sep + pfile