Update app.py

app.py

@@ -1,32 +1,36 @@
-from pydub import AudioSegment
 import gradio as gr
+import numpy as np
 import os
+import soundfile as sf
+from pydub import AudioSegment
 import math
+import shutil
 
-class SplitWavAudioMubin:
-    def __init__(self, folder, filename):
-        self.folder = folder
-        self.filename = filename
-        self.filepath = os.path.join(folder, filename)
-        self.audio = AudioSegment.from_wav(self.filepath)
-
+class SplitWavAudio:
+    def __init__(self, audio_file):
+        self.audio_file = audio_file
+        self.audio_data, self.sample_rate = sf.read(audio_file)
+        if len(self.audio_data.shape) > 1:
+            self.audio_data = np.mean(self.audio_data, axis=1)
+        self.audio_info = sf.info(audio_file)
+        self.bit_depth = {'PCM_16': 16, 'FLOAT': 32}.get(self.audio_info.subtype, 0)
+    
     def get_duration(self):
-        return self.audio.duration_seconds
-
-    def single_split(self, from_min, to_min, split_filename):
-        t1 = from_min * 60 * 1000
-        t2 = to_min * 60 * 1000
-        split_audio = self.audio[t1:t2]
-        split_audio.export(os.path.join(self.folder, split_filename), format="wav")
-
-    def multiple_split(self, min_per_split):
-        total_mins = math.ceil(self.get_duration() / 60)
-        for i in range(0, total_mins, min_per_split):
-            split_fn = f"{i}_{self.filename}"
-            self.single_split(i, i + min_per_split, split_fn)
-            print(f"{i} Done")
-            if i == total_mins - min_per_split:
-                print("All split successfully")
+        return self.audio_info.duration
+    
+    def single_split(self, from_sec, to_sec, split_filename):
+        t1 = int(from_sec * self.sample_rate)
+        t2 = int(to_sec * self.sample_rate)
+        split_audio = self.audio_data[t1:t2]
+        sf.write(split_filename, split_audio, self.sample_rate, subtype=self.audio_info.subtype)
+        
+    def multiple_split(self, sec_per_split):
+        total_sec = self.get_duration()
+        for i in range(0, int(total_sec), sec_per_split):
+            split_filename = f"{i}_{os.path.basename(self.audio_file)}"
+            self.single_split(i, i + sec_per_split, split_filename)
+            print(f"{i} seconds Done")
+        print('All split successfully')
 
 def main():
     # Gradio Interface
@@ -40,46 +44,51 @@ def main():
         with gr.Row():
             with gr.Column():
                 audio_input = gr.Audio(type='filepath')
-                get_info_butt = gr.Button(value='Get Audio File Info', variant='primary')
-                interval_input = gr.Number(value=5, label="Enter Interval (seconds):")
-                cut_audio_butt = gr.Button(value='Cut and Compile Audio', variant='success', visible=False)
+                create_info_butt = gr.Button(value='Get Audio File Info', variant='primary')
+                interval_input = gr.Number(min=1, placeholder='Enter Interval (seconds)', visible=False)
+                cut_audio_butt = gr.Button(value='Cut Audio', variant='success', visible=False)
             with gr.Column():
                 output_markdown = gr.Markdown(value="", visible=True)
     
-        get_info_butt.click(fn=get_audio_file_info, inputs=[audio_input, interval_input, cut_audio_butt, output_markdown])
-        cut_audio_butt.click(fn=cut_and_compile_audio, inputs=[audio_input, interval_input], outputs=[output_markdown, cut_audio_butt])
-        
+        create_info_butt.click(fn=get_audio_file_info, inputs=[audio_input, interval_input, cut_audio_butt],
+                               outputs=[output_markdown])
+        cut_audio_butt.click(fn=cut_audio, inputs=[audio_input, interval_input])
+
         app.queue(max_size=1022).launch(share=True)
 
-def get_audio_file_info(audio_file, interval, cut_audio_butt, output_markdown):
-    # Read the audio data from the file
-    audio_splitter = SplitWavAudioMubin('/mnt/data/audio_clips', 'audio_file.wav')
-    audio_duration = audio_splitter.get_duration()
+def get_audio_file_info(audio_file, interval_input, cut_audio_butt):
+    audio_info = SplitWavAudio(audio_file)
+    bit_depth = audio_info.bit_depth
+    minutes, seconds = divmod(audio_info.get_duration(), 60)
+    speed_in_kbps = audio_info.sample_rate * bit_depth / 1000
 
-    # Display audio file information
-    output_markdown.value = f"""
-    **Audio File Information:**
-    - File Name: {os.path.basename(audio_file)}
-    - Duration: {int(audio_duration / 60)} minutes {int(audio_duration % 60)} seconds
-    - Interval: {interval} seconds
+    info_table = f"""
+    | Information | Value |
+    | :---: | :---: |
+    | File Name | {os.path.basename(audio_file)} |
+    | Duration | {int(minutes)} minutes - {int(seconds)} seconds |
+    | Bitrate | {speed_in_kbps} kbp/s |
+    | Audio Channels | {audio_info.audio_info.channels} |
+    | Samples per second | {audio_info.sample_rate} Hz |
     """
 
-    # Enable the button for cutting audio
+    interval_input.visible = True
     cut_audio_butt.visible = True
 
-def cut_and_compile_audio(audio_file, interval):
-    # Read the audio data from the file
-    audio_splitter = SplitWavAudioMubin('/mnt/data/audio_clips', 'audio_file.wav')
-    audio_splitter.multiple_split(min_per_split=interval)
-
-    # Provide a download link for a zip file containing all clips
-    zip_file_path = '/mnt/data/audio_clips.zip'
-    os.system(f"zip -j {zip_file_path} {audio_splitter.folder}/*")
+    return info_table
 
-    # Provide a download link for the zip file
-    download_link = f"[Download All Clips](sandbox:{zip_file_path})"
+def cut_audio(audio_file, interval_input):
+    audio_info = SplitWavAudio(audio_file)
+    sec_per_split = interval_input
+    temp_folder = 'temp_audio_splits'
+    os.makedirs(temp_folder, exist_ok=True)
 
-    return f"Audio cut and compiled successfully. {download_link}"
+    try:
+        audio_info.multiple_split(sec_per_split)
+        shutil.make_archive('audio_splits', 'zip', temp_folder)
+        print('Audio splits compiled successfully!')
+    finally:
+        shutil.rmtree(temp_folder)
 
 # Create the Gradio interface
 main()