Update app.py

app.py

@@ -1,19 +1,14 @@
 import gradio as gr
-import matplotlib.pyplot as plt
-import numpy as np
 import os
 import soundfile as sf
-from pydub import AudioSegment
-import tempfile
-import zipfile
-
+from audio_processor import process_audio
 
 def main():
     with gr.Blocks() as app:
         gr.Markdown(
             """
             # <div align="center"> diablofx Audio Interval Cutter (BETA) </div>
-            Want to [support](https://ko-fi.com/diablofx) me? [Ko-Fi]\n
+            Want to [support](https://ko-fi.com/diablofx) me?\n
 
             Need help with AI? [Join AI Hub!](https://discord.gg/aihub)
             """
@@ -22,33 +17,15 @@ def main():
         with gr.Row():
             with gr.Column():
                 audio_input = gr.File(label="Upload an audio file")
-                interval_input = gr.Textbox("5000", label="Interval Length (ms)")
-                process_button = gr.Button(value='Process Audio and Get Info', variant='primary')
+                interval_input = gr.Textbox("5000", label="Interval Length (in ms)")
+                process_button = gr.Button(value='Start', variant='primary')
             with gr.Column():
-                output_html = gr.HTML("Processed audio information will appear here")
+                output_html = gr.HTML("Processed data will appear here")
 
         process_button.click(fn=process_audio_and_display_info, inputs=[audio_input, interval_input], outputs=output_html)
 
         app.queue(max_size=1022).launch(share=True)
 
-def cut_audio(audio_file, output_path, interval):
-    audio = AudioSegment.from_file(audio_file)
-    duration = len(audio)
-
-    cuts = [audio[i:i + interval] for i in range(0, duration, interval)]
-
-    with tempfile.TemporaryDirectory() as temp_dir:
-        for i, cut in enumerate(cuts):
-            cut.export(os.path.join(temp_dir, f'cut_{i}.wav'), format="wav")
-
-        zip_file_path = os.path.join(output_path, "cuts.zip")
-        with zipfile.ZipFile(zip_file_path, 'w') as zipf:
-            for root, _, files in os.walk(temp_dir):
-                for file in files:
-                    zipf.write(os.path.join(root, file), os.path.basename(file))
-
-    return zip_file_path
-
 def process_audio_and_display_info(audio_file, interval):
     # Get the audio file info
     audio_info = sf.info(audio_file)
@@ -78,11 +55,8 @@ def process_audio_and_display_info(audio_file, interval):
     | Bit per second | {audio_info.samplerate * audio_info.channels * bit_depth} bit/s |
     """
 
-    # Process the audio file and get the path to the zip file
-    zip_file_path = cut_audio(audio_file, tempfile.mkdtemp(), int(interval))
-
-    # Create a download link
-    download_link = f'<a href="/download/{os.path.basename(zip_file_path)}" download>Click here to download processed audio</a>'
+    # Process the audio file and get the download link
+    download_link = process_audio(audio_file, interval)
 
     # Return the information table and the download link
     return info_table, download_link