Update app_share.py

app_share.py

@@ -308,14 +308,14 @@ import shutil
 
 import zipfile
 
-def zip_sliced_files(directory, zip_filename):
+def zip_sliced_files(directory, zip_filename, chosen_name):
     # 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"{chosen_name}") and filename.endswith(".wav"):
                     # Create the complete file path
                     file_path = os.path.join(foldername, filename)
                     # Add the file to the zip file
@@ -324,24 +324,27 @@ def zip_sliced_files(directory, zip_filename):
 
 # set speed
 
-def change_speed(audio_inp, speed=1.0):
-    audio = AudioSegment.from_file(audio_inp)
+from pydub.effects import speedup
 
-    sound_with_altered_frame_rate = audio._spawn(audio.raw_data, overrides={
-         "frame_rate": int(audio.frame_rate * speed)
-    })
-    slower_audio = sound_with_altered_frame_rate.set_frame_rate(audio.frame_rate)
-    slower_audio.export("slower_speech.wav", format="wav")
-    return "slower_speech.wav"
+def change_speed(input_file, speed=1.0):
+    # Load the audio file
+    audio = AudioSegment.from_file(input_file)
 
+    # Change the speed of the audio
+    faster_audio = speedup(audio, playback_speed=speed)
+
+    # Export the modified audio to a new file
+    faster_audio.export("speed_changed_speech.wav", format="wav")
+    return "speed_changed_speech.wav"
+    
 # delete files first
 
-def delete_sliced_files(directory):
+def delete_sliced_files(directory, chosen_name):
     # 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"{chosen_name}"):
                 # Create the complete file path
                 file_path = os.path.join(foldername, filename)
                 # Delete the file
@@ -352,7 +355,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 +385,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 +395,29 @@ restart_markdown = ("""
 
 import ffmpeg
 
-def denoise(video_full):
-    
-    if os.path.exists("audio_full.wav"):
-        os.remove("audio_full.wav")
+def save_file_with_new_name(original_file_path, new_file_path):
+    shutil.copyfile(original_file_path, new_file_path)
+
+
+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:
+
+        name1 = video_file.name
+        file_name_with_extension = name1.split('/')[-1]
+        file_name1 = file_name_with_extension.split('.mp4')[0] + ".mp4"
+
+        save_file_with_new_name(video_file.name, file_name1)
+        
+        ffmpeg.input(file_name1).output("input_video" + file_name1 + ".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:
@@ -407,9 +425,9 @@ with gr.Blocks() as app:
     gr.Markdown("### <center>🌟 只需上传SRT文件和原版配音文件即可，每次一集视频AI自动配音！Developed by Kevin Wang </center>")
     with gr.Tab("📺视频转音频"):
         with gr.Row():
-            inp_video = gr.Video(label="请上传一集包含原声配音的视频", info="需要是.mp4视频文件")
+            inp_video = gr.Files(label="您可以上传多集包含原声配音的视频", file_types=['.mp4'])
             btn_convert = gr.Button("视频文件转音频", variant="primary")
-            out_audio = gr.Audio(label="视频对应的音频文件，可以下载至本地后进行降噪处理", type="filepath")
+            out_audio = gr.File(label="包含所有配音音频的zip文件")
 
         btn_convert.click(denoise, [inp_video], [out_audio])
     with gr.Tab("🎶AI配音"):
@@ -426,7 +444,7 @@ with gr.Blocks() as app:
             with gr.Column():
                 out1 = gr.Audio(label="为您生成的AI完整配音", type="filepath")
                 out2 = gr.File(label="包含所有参考音频的zip文件")
-                inp_speed = gr.Slider(label="设置AI配音的速度", minimum=0.8, maximum=1.2, value=1.0, step=0.01)
+                inp_speed = gr.Slider(label="设置AI配音的速度", minimum=1.02, maximum=1.5, value=1.02, step=0.01)
                 btn2 = gr.Button("一键改变AI配音速度")
                 out3 = gr.Audio(label="变速后的AI配音", type="filepath")