From 2929ae14d09430eaa2e8e72e2609efc31048fc68 Mon Sep 17 00:00:00 2001
From: chappjo <62079229+chappjo@users.noreply.github.com>
Date: Mon, 14 Dec 2020 21:51:19 +1300
Subject: [PATCH 1/2] Update inference_video.py

---
 inference_video.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/inference_video.py b/inference_video.py
index f155cbb..56c2426 100644
--- a/inference_video.py
+++ b/inference_video.py
@@ -209,4 +209,5 @@ if args.png == False and fpsNotAssigned == True and not args.skip and not args.v
     try:
         transferAudio(args.video, outputVideoFileName)
     except:
+        os.rename("noAudio_"+outputVideoFileName, outputVideoFileName)
         print("Audio transfer failed. Interpolated video will have no audio")

From 17672ca0a064146af94eb667a306c21790ad3ff2 Mon Sep 17 00:00:00 2001
From: chappjo <62079229+chappjo@users.noreply.github.com>
Date: Mon, 14 Dec 2020 23:06:00 +1300
Subject: [PATCH 2/2] Add support for edge cases like 3gp file container

---
 inference_video.py | 14 ++++++++++----
 1 file changed, 10 insertions(+), 4 deletions(-)

diff --git a/inference_video.py b/inference_video.py
index 56c2426..b36ac6d 100644
--- a/inference_video.py
+++ b/inference_video.py
@@ -36,10 +36,16 @@ def transferAudio(sourceVideo, targetVideo):
         tempAudioFileName = "./temp/audio.mp3"
         os.system("ffmpeg -y -i " + sourceVideo + " -c:a mp3 -vn " + tempAudioFileName)
         os.system("ffmpeg -y -i " + "noAudio_"+targetVideo + " -i " + tempAudioFileName + " -c copy " + targetVideo)
-        print("Lossless audio transfer failed. Audio was transcoded to mp3 instead.")
+        if (os.path.getsize(targetVideo) == 0): # if mp3 not supported by selected format
+            os.rename("noAudio_"+targetVideo, targetVideo)
+            print("Audio transfer failed. Interpolated video will have no audio")
+        else:
+            print("Lossless audio transfer failed. Audio was transcoded to mp3 instead.")
     
-    # remove audio-less video
-    os.remove("noAudio_"+targetVideo)
+            # remove audio-less video
+            os.remove("noAudio_"+targetVideo)
+    else:
+        os.remove("noAudio_"+targetVideo)
 
     # remove temp directory
     shutil.rmtree("temp")
@@ -209,5 +215,5 @@ if args.png == False and fpsNotAssigned == True and not args.skip and not args.v
     try:
         transferAudio(args.video, outputVideoFileName)
     except:
-        os.rename("noAudio_"+outputVideoFileName, outputVideoFileName)
         print("Audio transfer failed. Interpolated video will have no audio")
+        os.rename("noAudio_"+outputVideoFileName, outputVideoFileName)