INIT

runner.py

@@ -0,0 +1,60 @@
+import whisper
+import os
+import sys
+import logging
+import datetime as dt
+from datetime import datetime
+import traceback
+from pydub import AudioSegment
+
+env_var_language_code = os.environ['LANGUAGE_CODE']
+env_var_whisper_model = os.environ['WHISPER_MODEL']
+
+# Setup Logging
+logging.basicConfig(
+    level=logging.DEBUG,
+    # level=logging.INFO,
+    format="Start: " + str(dt.datetime.now()).replace(" ", "_") + " | %(asctime)s [%(levelname)s] %(message)s",
+    handlers=[
+        logging.FileHandler("/var/log/" + str(datetime.today().strftime('%Y-%m-%d')) + "_-_cron.log"),
+        logging.StreamHandler(sys.stdout)
+    ]
+)
+
+
+def get_audio_duration(file_path):
+    audio = AudioSegment.from_file(file_path)
+    duration_seconds = len(audio) / 1000
+    return duration_seconds
+
+
+try:
+    for root, dirs, files in os.walk('/app/input_files'):
+        for file in files:
+            try:
+                file_path = os.path.join(root, file)
+                logging.debug("#" * 32)
+                logging.debug(file_path)
+
+                duration = get_audio_duration(file_path)
+                logging.debug("Duration: " + str(duration) + " Seconds")
+
+                model = whisper.load_model(env_var_whisper_model)
+                if env_var_language_code == "multi":
+                    result = model.transcribe(file_path)
+                else:
+                    result = model.transcribe(file_path, language=env_var_language_code, initial_prompt="")
+                logging.debug("result: " + str(result))
+                result_text = result["text"]
+                logging.debug("result: " + result_text)
+
+                transcript_file = '/app/transcripts/' + file.split(".")[0] + '_transcript_' + env_var_language_code + '_.txt'
+                logging.debug("result: " + str(transcript_file))
+                with open(transcript_file, 'w') as f:
+                    f.write(result_text)
+            except Exception as e:
+                logging.debug("There was an error: " + str(e))
+                logging.debug("Stacktrace: " + str(traceback.format_exc()))
+except Exception as e:
+    logging.debug("There was an error: " + str(e))
+    logging.debug("Stacktrace: " + str(traceback.format_exc()))