Update app.py

app.py

@@ -308,14 +308,14 @@ import shutil
 
 import zipfile
 
-def zip_sliced_files(directory, zip_filename):
+def zip_sliced_files(directory, zip_filename, filename):
     # Create a ZipFile object
     with zipfile.ZipFile(zip_filename, 'w') as zipf:
         # Iterate over all files in the directory
         for foldername, subfolders, filenames in os.walk(directory):
             for filename in filenames:
                 # Check if the file starts with "sliced" and has a .wav extension
-                if filename.startswith("sliced") and filename.endswith(".wav"):
+                if filename.startswith(f"{filename}") and filename.endswith(".wav"):
                     # Create the complete file path
                     file_path = os.path.join(foldername, filename)
                     # Add the file to the zip file
@@ -336,12 +336,12 @@ def change_speed(audio_inp, speed=1.0):
 
 # delete files first
 
-def delete_sliced_files(directory):
+def delete_sliced_files(directory, filename):
     # Iterate over all files in the directory
     for foldername, subfolders, filenames in os.walk(directory):
         for filename in filenames:
             # Check if the file starts with "sliced"
-            if filename.startswith("sliced"):
+            if filename.startswith(f"{filename}"):
                 # Create the complete file path
                 file_path = os.path.join(foldername, filename)
                 # Delete the file
@@ -352,7 +352,7 @@ def delete_sliced_files(directory):
 def convert_from_srt(api_key, filename, audio_full, voice, multilingual):
 
     subtitle_list = read_srt(filename)
-    delete_sliced_files("./")
+    delete_sliced_files("./", "sliced")
 
     #audio_data, sr = librosa.load(audio_full, sr=44100)
         
@@ -382,7 +382,7 @@ def convert_from_srt(api_key, filename, audio_full, voice, multilingual):
                 pass
     merge_audios("output")
 
-    zip_sliced_files("./", "参考音频.zip")
+    zip_sliced_files("./", "参考音频.zip", "sliced")
     
     return "AI配音版.wav", "参考音频.zip"
     
@@ -392,14 +392,19 @@ restart_markdown = ("""
 
 import ffmpeg
 
-def denoise(video_full):
-    
-    if os.path.exists("audio_full.wav"):
-        os.remove("audio_full.wav")
+def denoise(input_files):
+    delete_sliced_files("./", "input_video")
+
+    #if os.path.exists("audio_full.wav"):
+    #    os.remove("audio_full.wav")
+    for video_file in input_files:
+        
+        ffmpeg.input(video_file).output("input_video" + video_file.name + ".wav", ac=2, ar=44100).run()
+
+    zip_sliced_files("./", "转换后的音频.zip", "input_video")
 
-    ffmpeg.input(video_full).output("audio_full.wav", ac=2, ar=44100).run()
 
-    return "audio_full.wav"
+    return "转换后的音频.zip"
 
 
 with gr.Blocks() as app: