This commit is contained in:
Kabooshki 2024-09-17 13:03:34 -05:00
parent cc23e1df33
commit 3a558db188
Signed by: Kabooshki
GPG Key ID: 5E7B7F2F405CD62A
1 changed files with 7 additions and 19 deletions

26
main.py
View File

@ -1,9 +1,9 @@
import cv2 import cv2
import pytesseract import pytesseract
import os import os
import numpy
directory = 'images' directory = 'images'
k = 1
# iterate over files in # iterate over files in
# that directory # that directory
for filename in os.listdir(directory): for filename in os.listdir(directory):
@ -20,21 +20,9 @@ for filename in os.listdir(directory):
print(data) print(data)
if os.path.isfile(f'images/{data}.jpg') == False: if os.path.isfile(f'images/{data}.jpg') == False:
os.rename(f'{f}', f'images/{data}.jpg') os.rename(f'{f}', f'images/{data}.jpg')
elif os.path.isfile(f'images/{data}.jpg') == True: else:
cv2.imshow(data, ROI) cv2.imshow(ROI)
cv2.waitKey(1) cv2.waitKey(50)
manpage = input("please input the number on the page (if the title is correct, enter nothing)") os.rename(f'{f}', f'images/{data}.{k}.jpg')
print(manpage) k = k + 1
if manpage == "\n": cv2.destroyAllWindows()
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()