Update app.py

app.py

@@ -1,23 +1,88 @@
 import gradio as gr
-from audio_processor import process_audio
+import matplotlib.pyplot as plt
+import numpy as np
+import os
+import soundfile as sf
+from pydub import AudioSegment
 
-iface = gr.Interface(
-    fn=process_audio,
-    inputs=[gr.File(label="Upload an audio file"), gr.Textbox("5000", label="Interval Length (ms)")],
-    outputs=gr.HTML("Download link will appear here")
-)
+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
 
-iface.launch()
+            Need help with AI? [Join AI Hub!](https://discord.gg/aihub)
+            """
+        )
 
+        with gr.Row():
+            with gr.Column():
+                audio_input = gr.File(label="Upload an audio file")
+                interval_input = gr.Textbox(default="5000", label="Interval Length (ms)")
+                process_button = gr.Button(value='Process Audio and Get Info', variant='primary')
+            with gr.Column():
+                output_html = gr.HTML("Processed audio information will appear here")
 
-from flask import Flask, send_file
-import os
+        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(input_path, output_path, interval):
+    audio = AudioSegment.from_file(input_path)
+    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)
+
+    bit_depth = {'PCM_16': 16, 'FLOAT': 32}.get(audio_info.subtype, 0)
+
+    # Convert duration to minutes, seconds, and milliseconds
+    minutes, seconds = divmod(audio_info.duration, 60)
+    seconds, milliseconds = divmod(seconds, 1)
+    milliseconds *= 1000  # convert from seconds to milliseconds
+
+    # Convert bitrate to mb/s
+    bitrate = audio_info.samplerate * audio_info.channels * bit_depth / 8 / 1024 / 1024
+
+    # Calculate speed in kbps
+    speed_in_kbps = audio_info.samplerate * bit_depth / 1000
+
+    # Create a table with the audio file info
+    info_table = f"""
+    | Information | Value |
+    | :---: | :---: |
+    | File Name | {os.path.basename(audio_file)} |
+    | Duration | {int(minutes)} minutes - {int(seconds)} seconds - {int(milliseconds)} milliseconds |
+    | Bitrate | {speed_in_kbps} kbp/s |
+    | Audio Channels | {audio_info.channels} |
+    | Samples per second | {audio_info.samplerate} Hz |
+    | 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))
 
-app = Flask(__name__)
+    # Create a download link
+    download_link = f'<a href="/download/{os.path.basename(zip_file_path)}" download>Click here to download processed audio</a>'
 
-@app.route("/download/<filename>")
-def download(filename):
-    return send_file(os.path.join("/path/to/your/app", filename), as_attachment=True)
+    # Return the information table and the download link
+    return info_table, download_link
 
 if __name__ == "__main__":
-    app.run()
+    main()