finished revising the script.

This commit is contained in:
2026-01-11 17:37:45 -05:00
parent 203fc69393
commit 64add85920
7 changed files with 771 additions and 904 deletions
+35 -9
View File
@@ -42,22 +42,39 @@ def main():
except ImportError:
pass
# 1. Input File/Folder
# 1. Input File/Folder (Multiple)
input_paths = []
while True:
input_path = get_input("Enter the path to the video file or folder")
prompt_text = "Enter a path to a video file or folder"
if input_paths:
prompt_text += " (or press Enter to finish)"
input_path = get_input(prompt_text)
if not input_path:
if input_paths:
break
else:
print("Error: You must provide at least one path.")
continue
# Clean up input
input_path = input_path.strip("'\"")
input_path = input_path.strip("\'"")
input_path = input_path.replace(r'\ ', ' ')
# Expand user (~) and resolve absolute path
input_path = os.path.abspath(os.path.expanduser(input_path))
if os.path.exists(input_path):
break
print(f"Error: Path '{input_path}' does not exist. Please try again.\n")
input_paths.append(input_path)
print(f"Added: {input_path}")
else:
print(f"Error: Path '{input_path}' does not exist. Please try again.\n")
print(f"Selected: {input_path}\n")
print("\nSelected Inputs:")
for p in input_paths:
print(f" - {p}")
print("")
# 2. Languages
source_lang = get_input("Source Language (e.g., French, es)", default="auto")
@@ -80,6 +97,8 @@ def main():
print(" It will only run if the new subtitled video is successfully created.")
do_delete_source = get_yes_no("Delete original source files after embedding?", default="n")
do_prefer_deep = get_yes_no("Prefer DeepTranslate (Free) over Gemini API?", default="n")
hf_token = None
if do_diarize:
if not os.getenv("HF_TOKEN"):
@@ -93,7 +112,8 @@ def main():
script_dir = os.path.dirname(os.path.abspath(__file__))
main_script = os.path.join(script_dir, "ai_transcriber_v2", "main.py")
cmd = [sys.executable, main_script, input_path]
cmd = [sys.executable, main_script]
cmd.extend(input_paths)
cmd.extend(["--lang", target_lang])
cmd.extend(["--model", model_size])
@@ -115,12 +135,17 @@ def main():
if hf_token:
cmd.extend(["--hf-token", hf_token])
if do_prefer_deep:
cmd.append("--prefer-deep")
# 6. Confirmation and Execution
clear_screen()
print_header()
print("Configuration Complete!")
print("-" * 30)
print(f"Input: {input_path}")
print("Inputs:")
for p in input_paths:
print(f" - {p}")
print(f"Source Lang: {source_lang}")
print(f"Target Lang: {target_lang}")
print(f"Model: {model_size}")
@@ -128,6 +153,7 @@ def main():
print(f"Embed Subs: {do_embed}")
print(f"Delete Src: {do_delete_source}")
print(f"Diarization: {do_diarize}")
print(f"Prefer Deep: {do_prefer_deep}")
print("-" * 30)
if not get_yes_no("Run this job now?", default="y"):
@@ -150,4 +176,4 @@ def main():
print("\nJob interrupted by user.")
if __name__ == "__main__":
main()
main()