refactor

reflector-local/1-transcript-generator.py

@@ -1,11 +1,13 @@
 import argparse
 import os
+
 import moviepy.editor
-from loguru import logger
 import whisper
+from loguru import logger
 
 WHISPER_MODEL_SIZE = "base"
 
+
 def init_argparse() -> argparse.ArgumentParser:
     parser = argparse.ArgumentParser(
         usage="%(prog)s <LOCATION> <OUTPUT>",
@@ -15,6 +17,7 @@ def init_argparse() -> argparse.ArgumentParser:
     parser.add_argument("output", help="Output file path")
     return parser
 
+
 def main():
     import sys
     sys.setrecursionlimit(10000)
@@ -26,10 +29,11 @@ def main():
     logger.info(f"Processing file: {media_file}")
 
     # Check if the media file is a valid audio or video file
-    if os.path.isfile(media_file) and not media_file.endswith(('.mp3', '.wav', '.ogg', '.flac', '.mp4', '.avi', '.flv')):
+    if os.path.isfile(media_file) and not media_file.endswith(
+            ('.mp3', '.wav', '.ogg', '.flac', '.mp4', '.avi', '.flv')):
         logger.error(f"Invalid file format: {media_file}")
         return
-    
+
     # If the media file we just retrieved is an audio file then skip extraction step
     audio_filename = media_file
     logger.info(f"Found audio-only file, skipping audio extraction")
@@ -53,5 +57,6 @@ def main():
     transcript_file.write(whisper_result["text"])
     transcript_file.close()
 
+
 if __name__ == "__main__":
     main()