20
20
import yaml
21
21
import numpy as np
22
22
import warnings
23
+ import shlex
23
24
from skimage import img_as_ubyte
24
25
import safetensors
25
26
import safetensors .torch
@@ -276,7 +277,7 @@ def generate(
276
277
]
277
278
278
279
video_name = x ["video_name" ] + ".mp4"
279
- path = os .path .join (video_save_dir , "temp_" + video_name )
280
+ path = os .path .join (video_save_dir , "temp_" + shlex . quote ( video_name ) )
280
281
281
282
imageio .mimsave (path , result , fps = float (25 ))
282
283
@@ -285,7 +286,7 @@ def generate(
285
286
286
287
audio_path = x ["audio_path" ]
287
288
audio_name = os .path .splitext (os .path .split (audio_path )[- 1 ])[0 ]
288
- new_audio_path = os .path .join (video_save_dir , audio_name + ".wav" )
289
+ new_audio_path = os .path .join (video_save_dir , shlex . quote ( audio_name ) + ".wav" )
289
290
start_time = 0
290
291
# cog will not keep the .mp3 filename
291
292
sound = AudioSegment .from_file (audio_path )
@@ -323,7 +324,7 @@ def generate(
323
324
start_time = end_time
324
325
if enhancer :
325
326
video_name_enhancer = x ["video_name" ] + "_enhanced.mp4"
326
- enhanced_path = os .path .join (video_save_dir , "temp_" + video_name_enhancer )
327
+ enhanced_path = os .path .join (video_save_dir , "temp_" + shlex . quote ( video_name_enhancer ) )
327
328
av_path_enhancer = os .path .join (video_save_dir , video_name_enhancer )
328
329
return_path = av_path_enhancer
329
330
enhanced_images = face_enhancer (
@@ -333,12 +334,12 @@ def generate(
333
334
imageio .mimsave (enhanced_path , enhanced_images , fps = float (25 ))
334
335
save_video_with_watermark (enhanced_path , new_audio_path , av_path_enhancer , watermark = False )
335
336
print (f"The generated video is named { video_save_dir } /{ video_name_enhancer } " )
336
- os .remove (enhanced_path )
337
+ os .remove (shlex . quote ( enhanced_path ) )
337
338
end_time = time .time ()
338
339
print (f"[***6/6***] enhancing takes: { end_time - start_time } sec" )
339
340
else :
340
341
print (f"[***6/6***] no enhancing" )
341
- os .remove (path )
342
- os .remove (new_audio_path )
342
+ os .remove (shlex . quote ( path ) )
343
+ os .remove (shlex . quote ( new_audio_path ) )
343
344
344
345
return return_path
0 commit comments