Spaces:
Running
Running
refactor: Sanitize filenames to prevent FFMPEG input errors
Browse files
app.py
CHANGED
|
@@ -49,7 +49,8 @@ def get_files_infos(files):
|
|
| 49 |
file_path = Path(file.name)
|
| 50 |
info = {}
|
| 51 |
info["size"] = os.path.getsize(file_path)
|
| 52 |
-
|
|
|
|
| 53 |
file_extension = file_path.suffix
|
| 54 |
|
| 55 |
if file_extension in (".mp4", ".avi", ".mkv", ".mov"):
|
|
@@ -203,10 +204,11 @@ def update(files, prompt, top_p=1, temperature=1):
|
|
| 203 |
if args[0] != "ffmpeg":
|
| 204 |
raise Exception("Command does not start with ffmpeg")
|
| 205 |
temp_dir = tempfile.mkdtemp()
|
| 206 |
-
# copy files to temp dir
|
| 207 |
for file in files:
|
| 208 |
file_path = Path(file.name)
|
| 209 |
-
|
|
|
|
| 210 |
|
| 211 |
# test if ffmpeg command is valid dry run
|
| 212 |
ffmpg_dry_run = subprocess.run(
|
|
|
|
| 49 |
file_path = Path(file.name)
|
| 50 |
info = {}
|
| 51 |
info["size"] = os.path.getsize(file_path)
|
| 52 |
+
# Sanitize filename by replacing spaces with underscores
|
| 53 |
+
info["name"] = file_path.name.replace(" ", "_")
|
| 54 |
file_extension = file_path.suffix
|
| 55 |
|
| 56 |
if file_extension in (".mp4", ".avi", ".mkv", ".mov"):
|
|
|
|
| 204 |
if args[0] != "ffmpeg":
|
| 205 |
raise Exception("Command does not start with ffmpeg")
|
| 206 |
temp_dir = tempfile.mkdtemp()
|
| 207 |
+
# copy files to temp dir with sanitized names
|
| 208 |
for file in files:
|
| 209 |
file_path = Path(file.name)
|
| 210 |
+
sanitized_name = file_path.name.replace(" ", "_")
|
| 211 |
+
shutil.copy(file_path, Path(temp_dir) / sanitized_name)
|
| 212 |
|
| 213 |
# test if ffmpeg command is valid dry run
|
| 214 |
ffmpg_dry_run = subprocess.run(
|