-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathargument.py
26 lines (18 loc) · 901 Bytes
/
argument.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
import argparse
import sys
parser = argparse.ArgumentParser(description="Program to OCR PNG with SRT export from Subtitle Edit")
parser.add_argument("-i", "--input-folder", type=str, help="Input folder path for PNG")
parser.add_argument("-o", "--output-sub", type=str, help="Output location file for subtitle (e.g., Temp.srt)")
parser.add_argument("-l", "--language", type=str, default=None, help="Language to help OCR process")
parser.add_argument("-k", "--key", type=str, help="API key for Google Cloud Vision API")
parser.add_argument("-s", "--sub", type=str, help="Blank SRT subtitle file")
parser.add_argument("--version", action="store_true", help="Print version and exit")
args = parser.parse_args()
if args.version:
print("Vob2Srt version 1.0.0")
sys.exit(0)
input_folder = args.input_folder
output_sub = args.output_sub
lang = args.language
key = args.key
sub_file = args.sub