dasemu/Transcriptarr
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Refactor transcription worker error handling

Browse Source 
Might fix CPU memory leak.
This commit is contained in:
McCloudS
2026-01-07 13:39:19 -05:00
parent 7373f9a87b
commit 594b78b84f
1 changed files with 10 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
subgen.py
View File

@@ -1,4 +1,4 @@
subgen_version = '2025.11.1' subgen_version = '2026.01.1'
""" """
ENVIRONMENT VARIABLES DOCUMENTATION ENVIRONMENT VARIABLES DOCUMENTATION
@@ -265,6 +265,7 @@ task_queue = DeduplicatedQueue()
def transcription_worker(): def transcription_worker():
while True: while True:
task = None
try: try:
task = task_queue.get(block=True, timeout=1) task = task_queue.get(block=True, timeout=1)
if "type" in task and task["type"] == "detect_language": if "type" in task and task["type"] == "detect_language":
@@ -274,15 +275,14 @@ def transcription_worker():
else: else:
logging.info(f"Task {task['path']} is being handled by Subgen.") logging.info(f"Task {task['path']} is being handled by Subgen.")
gen_subtitles(task['path'], task['transcribe_or_translate'], task['force_language']) gen_subtitles(task['path'], task['transcribe_or_translate'], task['force_language'])
task_queue.task_done()
# show queue
logging.debug(f"Queue status: {task_queue.qsize()} tasks remaining")
except queue.Empty: except queue.Empty:
continue # This is ok, as we have a timeout, nothing needs to be printed continue
except Exception as e: except Exception as e:
logging.error(f"Error processing task: {e}", exc_info=True) # Log the error and the traceback logging.error(f"Error processing task: {e}", exc_info=True)
else: finally:
delete_model() # Call delete_model() *only* if no exception occurred if task: # Ensure a task was actually retrieved before calling task_done
task_queue.task_done()
delete_model()
for _ in range(concurrent_transcriptions): for _ in range(concurrent_transcriptions):
threading.Thread(target=transcription_worker, daemon=True).start() threading.Thread(target=transcription_worker, daemon=True).start()