add keep_audio flag

Playbak-Team · Jan 10, 2021 · 577a75d · 577a75d
1 parent ffb6363
commit 577a75d
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/.version b/.version
@@ -1 +1 @@
-master-v0.1-1-ga7bd83c
+master-v0.1-rc1-0-gffb6363
diff --git a/main.py b/main.py
@@ -46,6 +46,10 @@ def main():
     speed = [args.silent_speed if cs else args.speech_speed for cs in normalized]
 
     pbs_writer(pbsFile, speed)
+
+    if(not args.keep_audio):
+        audioFile.unlink(missing_ok=True)
+
     print(pbsFile)
 
 
@@ -124,5 +128,8 @@ def get_audio(videofile: Path, outfile: Path):
         default=1.5,
         type=float,
     )
+    parser.add_argument(
+        "--keep_audio", help="Keep audio files generated.", action="store_true"
+    )
     args = parser.parse_args()
     main()