added diag script
This commit is contained in:
@@ -195,23 +195,20 @@ def process_file(file_path, args, source_lang=None, loaded_model=None, service_s
|
||||
should_embed = False
|
||||
|
||||
if should_embed:
|
||||
embed_subtitles(file_path, final_srt_path)
|
||||
success_embed = embed_subtitles(file_path, final_srt_path)
|
||||
|
||||
if args.delete_source:
|
||||
if args.embed:
|
||||
base, ext = os.path.splitext(file_path)
|
||||
expected_output = f"{base}.subbed{ext}"
|
||||
|
||||
if os.path.exists(expected_output):
|
||||
try:
|
||||
os.remove(file_path)
|
||||
tracker.logger.info(f"SOURCE DELETED: {file_path}")
|
||||
except OSError as e:
|
||||
tracker.logger.error(f"Error deleting source: {e}")
|
||||
else:
|
||||
tracker.logger.error(f"SAFETY ABORT: Output '{expected_output}' not found.")
|
||||
if success_embed and args.delete_source:
|
||||
base, ext = os.path.splitext(file_path)
|
||||
expected_output = f"{base}.subbed{ext}"
|
||||
|
||||
if os.path.exists(expected_output):
|
||||
try:
|
||||
os.remove(file_path)
|
||||
tracker.logger.info(f"SOURCE DELETED: {file_path}")
|
||||
except OSError as e:
|
||||
tracker.logger.error(f"Error deleting source: {e}")
|
||||
else:
|
||||
tracker.logger.warning("SAFETY ABORT: Enable --embed to delete source.")
|
||||
tracker.logger.error(f"SAFETY ABORT: Output '{expected_output}' not found.")
|
||||
tracker.update_step(file_path, "step_embed", "done")
|
||||
|
||||
# 5. Cleanup
|
||||
|
||||
Reference in New Issue
Block a user