From 42c591f758ac478dd7a6b903c11586f0ba913391 Mon Sep 17 00:00:00 2001
From: Harisreedhar <46858047+harisreedhar@users.noreply.github.com>
Date: Tue, 23 Jan 2024 13:01:49 +0530
Subject: [PATCH] include audio file functions (#336)

---
 facefusion/filesystem.py | 24 ++++++++++++++++++++++++
 1 file changed, 24 insertions(+)

diff --git a/facefusion/filesystem.py b/facefusion/filesystem.py
index 87923d2c..7d59cd8c 100644
--- a/facefusion/filesystem.py
+++ b/facefusion/filesystem.py
@@ -80,6 +80,12 @@ def is_video(video_path : str) -> bool:
 	return False
 
 
+def is_audio(audio_path : str) -> bool:
+	if is_file(audio_path):
+		return filetype.helpers.is_audio(audio_path)
+	return False
+
+
 def resolve_relative_path(path : str) -> str:
 	return os.path.abspath(os.path.join(os.path.dirname(__file__), path))
 
@@ -89,3 +95,21 @@ def list_directory(directory_path : str) -> Optional[List[str]]:
 		files = os.listdir(directory_path)
 		return [ Path(file).stem for file in files if not Path(file).stem.startswith(('.', '__')) ]
 	return None
+
+
+def filter_image_paths(paths : List[str]) -> List[Optional[str]]:
+	image_paths = []
+	if paths:
+		for path in paths:
+			if is_image(path):
+				image_paths.append(path)
+	return image_paths
+
+
+def filter_audio_paths(paths : List[str]) -> List[Optional[str]]:
+	audio_paths = []
+	if paths:
+		for path in paths:
+			if is_audio(path):
+				audio_paths.append(path)
+	return audio_paths