diff --git a/Model/ImageWriter.py b/Model/ImageWriter.py index 9852523..0805605 100644 --- a/Model/ImageWriter.py +++ b/Model/ImageWriter.py @@ -1,4 +1,5 @@ from iptcinfo3 import IPTCInfo +import os class ImageWriter: def writeTagsFromPredictionsInImages(self, predictions): @@ -14,5 +15,10 @@ def writeTagsFromPredictionsInImages(self, predictions): for category in prediction.m_categories: if category not in iptcInfo['keywords']: iptcInfo['keywords'].append(category) - - iptcInfo.save() \ No newline at end of file + try: + iptcInfo.save() + tempFilename = filename + "~" + if os.path.exists(tempFilename): + os.remove(tempFilename) + except Exception as e: + print('Error in file "'+filename+'" : \n\t', e) \ No newline at end of file