[NeuralChat] update path in audio utils (#593)

intel · Nov 1, 2023 · 0c57d11 · 0c57d11
1 parent 9daae2c
commit 0c57d11
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 25 deletions.
diff --git a/intel_extension_for_transformers/neural_chat/pipeline/plugins/audio/utils/cut_video.py b/intel_extension_for_transformers/neural_chat/pipeline/plugins/audio/utils/cut_video.py
@@ -31,7 +31,8 @@ def get_length(filename):
 def cut_video(args, outdir):
     path = shlex.quote(args.path)
     save_path = outdir
-    video_list = os.listdir(shlex.quote(path))
+    path, basename = os.path.split(path)
+    video_list = [basename]
     delta_X = int(shlex.quote(args.min))
 
     for file_name in video_list:

diff --git a/intel_extension_for_transformers/neural_chat/pipeline/plugins/audio/utils/video_to_wav.py b/intel_extension_for_transformers/neural_chat/pipeline/plugins/audio/utils/video_to_wav.py
@@ -23,12 +23,8 @@
 from pydub import AudioSegment
 
 def convert_video_to_wav(path, output_sample_rate, is_mono=True):
-    is_dir = os.path.isdir(path)
-    if is_dir: 
-        path_list=os.listdir(shlex.quote(path))
-    else: # input is a file
-        path, basename = os.path.split(path)
-        path_list = [basename]
+    path, basename = os.path.split(path)
+    path_list = [basename]
     print(path)
 
     output_dir = os.path.join(path, "../raw")

diff --git a/intel_extension_for_transformers/neural_chat/tests/audio/test_cut_video.py b/intel_extension_for_transformers/neural_chat/tests/audio/test_cut_video.py
@@ -31,7 +31,7 @@ def tearDown(self) -> None:
 
     def test_cut_video(self):
             parser = argparse.ArgumentParser(__doc__)
-            parser.add_argument("--path", type=str, default="/media")
+            parser.add_argument("--path", type=str, default="/media/intel.mp4")
             parser.add_argument("--min", type=str, default='1')
             parser.add_argument("--sr", type=str, default='16000')
             parser.add_argument("--out_path", type=str, default="../raw")

diff --git a/intel_extension_for_transformers/neural_chat/tests/audio/test_video_to_wav.py b/intel_extension_for_transformers/neural_chat/tests/audio/test_video_to_wav.py
@@ -48,23 +48,6 @@ def test_video_to_wav_file(self):
 
         self.assertTrue(os.path.exists('/raw/intel.wav'))
 
-    def test_video_to_wav_folder(self):
-        parser = argparse.ArgumentParser(__doc__)
-        parser.add_argument("--path", type=str, default='/media')
-        parser.add_argument("--is_mono", type=str, default='True')
-        parser.add_argument("--sr", type=str, default='16000')
-        parser.add_argument("--verbose", help="increase output verbosity", action="store_true")
-        args = parser.parse_args()
-        output_sample_rate = shlex.quote(args.sr)
-        is_exist = os.path.exists(shlex.quote(args.path))
-        if not is_exist:
-            print("path not existed!")
-        else:
-            path = shlex.quote(args.path)
-            is_mono = shlex.quote(args.is_mono)
-            convert_video_to_wav(path, output_sample_rate, is_mono) 
-
-        self.assertTrue(os.path.exists('/raw/intel.wav'))
 
 if __name__ == "__main__":
     unittest.main()