Kabooshki
/
Textbook-Arrange
1
0
Fork 0
Code Issues Pull Requests Packages Projects 1 Releases 1 Wiki Activity

removed extensions to test compatability

This commit is contained in:
Kabooshki 2024-09-18 12:22:33 -05:00
parent e0fe811463
commit fe186d6c12
Signed by: Kabooshki
GPG Key ID: 5E7B7F2F405CD62A
2 changed files with 15 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
fixer.py
View File

@ -22,9 +22,9 @@ for files in os.listdir(directory):
cv2.imshow('Review', ROI) cv2.imshow('Review', ROI)
cv2.waitKey(1) cv2.waitKey(1)
manpage = input("please input the page number") manpage = input("please input the page number")
if os.path.isfile(f'images/{manpage}.jpg') == True: if os.path.isfile(f'images/{manpage}') == True:
rng = numpy.random.default_rng().random() rng = numpy.random.default_rng().random()
os.rename (f'images/{manpage}.jpg', f'images/{manpage}.review-{rng}.jpg') os.rename (f'images/{manpage}', f'images/{manpage}.review-{rng}')
os.rename(f'{f}', f'images/{manpage}.jpg') os.rename(f'{f}', f'images/{manpage}')
elif os.path.isfile(f'images/{manpage}.jpg') == False: elif os.path.isfile(f'images/{manpage}') == False:
os.rename(f'{f}', f'images/{manpage}.jpg') os.rename(f'{f}', f'images/{manpage}')

20
main.py
View File

@ -18,23 +18,23 @@ for filename in os.listdir(directory):
ROI = cv2.hconcat([ROI1, ROI2]) ROI = cv2.hconcat([ROI1, ROI2])
data = pytesseract.image_to_string(ROI, lang='eng',config='--psm 7 -c tessedit_char_whitelist=0123456789i' ) data = pytesseract.image_to_string(ROI, lang='eng',config='--psm 7 -c tessedit_char_whitelist=0123456789i' )
print(data) print(data)
if os.path.isfile(f'images/{data}.jpg') == False: if os.path.isfile(f'images/{data}') == False:
os.rename(f'{f}', f'images/{data}.jpg') os.rename(f'{f}', f'images/{data}')
elif os.path.isfile(f'images/{data}.jpg') == True: elif os.path.isfile(f'images/{data}') == True:
cv2.imshow(data, ROI) cv2.imshow(data, ROI)
cv2.waitKey(1) cv2.waitKey(1)
manpage = input("please input the number on the page (if the title is correct, enter nothing)") manpage = input("please input the number on the page (if the title is correct, enter nothing)")
#check if the new name already exists #check if the new name already exists
if manpage == "\n": if manpage == "\n":
rng = numpy.random.default_rng().random() rng = numpy.random.default_rng().random()
os.rename (f'images/{data}.jpg', f'images/{data}.review-{rng}.jpg') os.rename (f'images/{data}', f'images/{data}.review-{rng}')
os.rename(f'{f}', f'images/{data}.jpg') os.rename(f'{f}', f'images/{data}')
elif manpage != "\n": elif manpage != "\n":
if os.path.isfile(f'images/{manpage}.jpg') == True: if os.path.isfile(f'images/{manpage}') == True:
rng = numpy.random.default_rng().random() rng = numpy.random.default_rng().random()
os.rename (f'images/{manpage}.jpg', f'images/{manpage}.review-{rng}.jpg') os.rename (f'images/{manpage}', f'images/{manpage}.review-{rng}')
os.rename(f'{f}', f'images/{manpage}.jpg') os.rename(f'{f}', f'images/{manpage}')
elif os.path.isfile(f'images/{manpage}.jpg') == False: elif os.path.isfile(f'images/{manpage}') == False:
os.rename(f'{f}', f'images/{manpage}.jpg') os.rename(f'{f}', f'images/{manpage}')
cv2.destroyAllWindows() cv2.destroyAllWindows()
cv2.destroyAllWindows() cv2.destroyAllWindows()