refactor translate 1 and multi languages

main.py

@@ -1,6 +1,6 @@
 from fasthtml.common import *
 from fasthtml.xtend import CheckboxX
-from translator import languages as VALID_LANGUAGES, process_sbv_file
+from translator import languages as VALID_LANGUAGES, translate_to_languages
 import tempfile
 from starlette.requests import Request
 from typing import List
@@ -72,7 +72,7 @@ def translate_and_process_sbv(content: bytes, filename: str, languages: List[str
         output_dir = os.path.join(temp_dir, "translations")
         os.makedirs(output_dir, exist_ok=True)
 
-        process_sbv_file(input_file, output_dir, selected_languages)
+        translate_to_languages(input_file, output_dir, selected_languages)
 
         return create_translation_files(temp_dir, output_dir, filename, selected_languages)
     except Exception as e:

translator.py

@@ -90,39 +90,53 @@ def reconstruct_sbv_content(blocks):
 def get_output_file_path(output_dir, base_filename, lang):
     return os.path.join(output_dir, f"{base_filename}_{lang}.sbv")
 
-def process_sbv_file(input_file, output_dir, selected_languages):
+def translate_to_language(input_file, output_dir, language):
     """
-    Process an SBV subtitle file and translate it into multiple languages.
-
-    This function reads an input SBV file, parses its content, translates it into
-    the specified languages, and saves the translated versions as separate files.
+    Translate an SBV subtitle file to a single language.
 
     Args:
         input_file (str): Path to the input SBV file.
-        output_dir (str): Path to the directory where translated files will be saved.
-        selected_languages (list): List of target languages for translation.
+        output_dir (str): Path to the directory where translated file will be saved.
+        language (str): Target language for translation.
 
     Returns:
-        None
-
-    Side effects:
-        - Creates the output directory if it doesn't exist.
-        - Writes translated SBV files to the output directory.
-        - Prints progress information to the console.
+        str: Path to the translated output file.
     """
     lines = read_sbv_file(input_file)
     base_filename = get_base_filename(input_file)
     create_output_directory(output_dir)
     blocks = parse_sbv_file(lines)
 
+    print(f"\nStarting translation for language: {language}")
+    translated_blocks = translate_blocks(blocks, language)
+    output_content = reconstruct_sbv_content(translated_blocks)
+    output_file = get_output_file_path(output_dir, base_filename, language)
+    write_sbv_file(output_file, output_content)
+    print(f"Translated {language} file saved: {output_file}")
+    return output_file
+
+def translate_to_languages(input_file, output_dir, selected_languages):
+    """
+    Translate an SBV subtitle file to multiple languages.
+
+    Args:
+        input_file (str): Path to the input SBV file.
+        output_dir (str): Path to the directory where translated files will be saved.
+        selected_languages (list): List of target languages for translation.
+
+    Returns:
+        list: Paths to the translated output files.
+    """
+    create_output_directory(output_dir)
+    translated_files = []
+
     for lang in tqdm(selected_languages, desc="Translating languages"):
-        print(f"\nStarting translation for language: {lang}")
-        translated_blocks = translate_blocks(blocks, lang)
-        output_content = reconstruct_sbv_content(translated_blocks)
-        output_file = get_output_file_path(output_dir, base_filename, lang)
-        write_sbv_file(output_file, output_content)
-        print(f"Translated {lang} file saved: {output_file}")
-        
+        output_file = translate_to_language(input_file, output_dir, lang)
+        translated_files.append(output_file)
+
+    return translated_files
+
+# Update the main block to use the new functions
 if __name__ == "__main__":
     parser = argparse.ArgumentParser(description="Translate SBV subtitle files to multiple languages.")
     parser.add_argument("input_file", help="Path to the input .sbv file")
@@ -130,4 +144,5 @@ if __name__ == "__main__":
     parser.add_argument("--languages", nargs="+", choices=languages, default=languages, help="Languages to translate to")
     args = parser.parse_args()
 
-    process_sbv_file(args.input_file, args.output_dir, args.languages)
+    translated_files = translate_to_languages(args.input_file, args.output_dir, args.languages)
+    print(f"\nTranslation completed. {len(translated_files)} files created.")