From 1679c8f1dba774a9a3c836e0292b1daa03a23a39 Mon Sep 17 00:00:00 2001
From: McCloudS <64094529+McCloudS@users.noreply.github.com>
Date: Tue, 5 Mar 2024 12:36:47 -0700
Subject: [PATCH] Update subgen.py

---
 subgen/subgen.py | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/subgen/subgen.py b/subgen/subgen.py
index 13e803b..f7346d0 100644
--- a/subgen/subgen.py
+++ b/subgen/subgen.py
@@ -22,7 +22,6 @@ import requests
 import av
 import ffmpeg
 import whisper
-from tqdm.contrib.logging import logging_redirect_tqdm
 
 def convert_to_bool(in_bool):
     if isinstance(in_bool, bool):
@@ -69,6 +68,9 @@ files_to_transcribe = []
 subextension =  f".subgen.{whisper_model.split('.')[0]}.{namesublang}.srt"
 subextensionSDH =  f".subgen.{whisper_model.split('.')[0]}.{namesublang}.sdh.srt"
 
+in_docker = os.path.exists('/.dockerenv')
+docker_status = "Docker" if in_docker else "Standalone"
+
 # Define a filter class
 class MultiplePatternsFilter(logging.Filter):
     def filter(self, record):
@@ -89,10 +91,10 @@ class MultiplePatternsFilter(logging.Filter):
 # Configure logging
 if debug:
     level = logging.DEBUG
-    logging.basicConfig(stream=sys.stderr, level=level, format="%(asctime)s %(levelname)s: %(message)s")
+    logging.basicConfig(stream=sys.stdout, level=level, format="%(asctime)s %(levelname)s: %(message)s")
 else:
     level = logging.INFO
-    logging.basicConfig(stream=sys.stderr, level=level)
+    logging.basicConfig(stream=sys.stdout, level=level)
 
 # Get the root logger
 logger = logging.getLogger()
@@ -110,6 +112,9 @@ logging.getLogger("watchfiles").setLevel(logging.WARNING)
 def progress(seek, total):
     sys.stdout.flush()
     sys.stderr.flush()
+    if(docker_status) == 'Docker':
+        percent = ((seek / total) * 100)
+        logging.info(f"Progress: {seek}/{total} ({percent:.2f}%)")
 
 TIME_OFFSET = 5