Refactoring: adapted for changes in cuda, updated README.md, improved logger and added sentence level transcript generation.

Signed-off-by: Orihara Izaya <xloki.stormbringery@gmail.com>

main.py

@@ -1,19 +1,49 @@
+import json
+import logging
 import os
+from argparse import ArgumentParser, Namespace
+from pathlib import Path
 
+from src.logger import setup_logger
 from src.transcript_generator import TranscriptGenerator
 
 
 def main():
-    # input could be a YouTube link or a local file
-    input_sources = ["https://www.youtube.com/watch?v=BVVo1BaL6JA"]
+    parser: ArgumentParser = ArgumentParser(
+        description="Transcript Generator for Videos"
+    )
+
+    parser.add_argument("--job", type=str, help="The name of the job", required=True)
+
+    arguments: Namespace = parser.parse_args()
+
+    with open(f"data/jobs/{arguments.job}", "r") as f:
+        job_config = json.load(f)
+
+    log_directory: Path = Path("data/logs")
+    os.makedirs(log_directory, exist_ok=True)
+    log_file_path: Path = log_directory / f"{arguments.job}.log"
+    setup_logger(__name__, log_file=log_file_path)
+    logger: logging.Logger = logging.getLogger(__name__)
+
+    logger.info("Starting the transcript generation.")
 
     hf_token = os.environ.get("HF_TOKEN")
     if not hf_token:
         hf_token = input("Please enter your HF_TOKEN: ")
     os.environ["HF_TOKEN"] = hf_token
 
-    transcript_generator = TranscriptGenerator()
-    transcript_generator.generate(input_sources)
+    transcript_generator: TranscriptGenerator = TranscriptGenerator(
+        job_name=job_config["name"],
+        language_code=job_config.get("language_code", "en"),
+        get_txt_version=job_config["transcript_version"]["get_txt_version"],
+        get_json_version=job_config["transcript_version"]["get_json_version"],
+        get_sentence_level_json=job_config["transcript_version"][
+            "get_sentence_level_json_version"
+        ],
+    )
+
+    transcript_generator.generate(job_config["source"])
 
 
 if __name__ == "__main__":