diff --git a/main.py b/main.py index 1e25278..0d14831 100644 --- a/main.py +++ b/main.py @@ -1,6 +1,7 @@ import cv2 import pytesseract import os +import numpy directory = 'images' k = 1 @@ -20,9 +21,22 @@ for filename in os.listdir(directory): print(data) if os.path.isfile(f'images/{data}.jpg') == False: os.rename(f'{f}', f'images/{data}.jpg') - else: - cv2.imshow(ROI) - cv2.waitKey(50) - os.rename(f'{f}', f'images/{data}.{k}.jpg') - k = k + 1 + elif os.path.isfile(f'images/{data}.jpg') == True: + cv2.imshow(data, ROI) + cv2.waitKey(1) + manpage = input("please input the number on the page (if the title is correct, enter nothing)") + print(manpage) + #check if the new name already exists + if manpage == "\n": + rng = numpy.random.default_rng().random() + os.rename (f'images/{data}.jpg', f'images/{data}.review-{rng}.jpg') + os.rename(f'{f}', f'images/{data}.jpg') + elif manpage != "\n": + if os.path.isfile(f'images/{manpage}.jpg') == True: + rng = numpy.random.default_rng().random() + os.rename (f'images/{manpage}.jpg', f'images/{manpage}.review-{rng}.jpg') + os.rename(f'{f}', f'images/{manpage}.jpg') + elif os.path.isfile(f'images/{manpage}.jpg') == False: + os.rename(f'{f}', f'images/{manpage}.jpg') + cv2.destroyAllWindows() cv2.destroyAllWindows()