gradio

MMAudio/demo.py

@@ -178,7 +178,8 @@ def main():
         audio = audios.float().cpu()[0]
         if video_path is not None:
             ####save_path = output_dir / f'{video_path.stem}.flac'
-            save_path = str(output_dir) + "/" + str(video_path).replace("/", "__").strip(".") + ".flac"
+            ####save_path = str(output_dir) + "/" + str(video_path).replace("/", "__").strip(".") + ".flac"
+            save_path = str(output_dir) + "/__" + os.path.basename(video_path).strip(".") + ".flac"
         else:
             safe_filename = prompt.replace(' ', '_').replace('/', '_').replace('.', '')
             save_path = output_dir / f'{safe_filename}.flac'
@@ -209,7 +210,8 @@ def main():
         log.info(f'Audio saved to {save_path}')
         if video_path is not None and not skip_video_composite:
             ####video_save_path = output_dir / f'{video_path.stem}.mp4'
-            video_save_path = str(output_dir) + "/" + str(video_path).replace("/", "__").strip(".") + ".mp4"
+            ####video_save_path = str(output_dir) + "/" + str(video_path).replace("/", "__").strip(".") + ".mp4"
+            video_save_path = str(output_dir) + "/__" + os.path.basename(video_path).strip(".") + ".mp4"
             make_video(video_info, video_save_path, audio, sampling_rate=seq_cfg.sampling_rate)
             log.info(f'Video saved to {video_save_path}')
 

app.py

@@ -17,6 +17,7 @@ import torchaudio
 import tempfile
 
 import requests
+import shutil
 
 log = logging.getLogger()
 
@@ -26,8 +27,8 @@ def video_to_audio(video: gr.Video, prompt: str):
 
     video_path = tempfile.NamedTemporaryFile(delete=False, suffix='.mp4').name
     
-    output_dir = video_path.rsplit("/", 1)[0]
-    video_save_path = str(output_dir) + "/" + str(video_path).replace("/", "__").strip(".") + ".mp4"
+    output_dir = os.path.dirname(video_path)
+    video_save_path = os.path.join(str(output_dir), "__" + os.path.basename(video_path).strip(".") + ".mp4")
     
     print("paths", video, video_path, output_dir, video_save_path)
 
@@ -36,9 +37,12 @@ def video_to_audio(video: gr.Video, prompt: str):
         with open(video_path, "wb") as fw:
             fw.write(data)
     else:
-        os.system("cp %s %s" % (video, video_path))
+        shutil.copy(video, video_path)
     
-    os.system("cd ./MMAudio; python ./demo.py --variant small_44k --output %s --video %s --prompt %s --calc_energy 1" % (output_dir, video_path, prompt))
+    if prompt == "":
+        os.system("cd ./MMAudio; python ./demo.py --variant small_44k --output %s --video %s --calc_energy 1" % (output_dir, video_path))
+    else:
+        os.system("cd ./MMAudio; python ./demo.py --variant small_44k --output %s --video %s --prompt %s --calc_energy 1" % (output_dir, video_path, prompt))
     
     return video_save_path