Update app.py

app.py

@@ -1,8 +1,9 @@
 import gradio as gr
 import os
 import ffmpeg
+import shutil
 
-# Function to validate and split audio
+# Function to split the audio file using FFmpeg
 def split_audio_ffmpeg(input_audio, output_directory, chunk_duration=30):
     """
     Split an audio file into smaller chunks using FFmpeg.
@@ -13,51 +14,48 @@ def split_audio_ffmpeg(input_audio, output_directory, chunk_duration=30):
     os.makedirs(output_directory, exist_ok=True)
 
     try:
-        # Validate the file duration
-        probe = ffmpeg.probe(input_audio.name, select_streams="a", show_entries="stream=duration")
-        duration = float(probe["streams"][0]["duration"])
-        max_duration = 3600  # 1 hour limit for uploaded audio files
-        
-        if duration > max_duration:
-            return f"Error: File duration exceeds {max_duration / 60} minutes. Please upload a shorter file."
+        # Get the total duration of the audio file using ffmpeg.probe()
+        probe = ffmpeg.probe(input_audio.name, v='error', select_streams='a', show_entries='stream=duration')
+        duration = float(probe['streams'][0]['duration'])
+        print(f"Audio duration: {duration} seconds")
 
         # Split the file into chunks
         chunks = []
         for i in range(0, int(duration), chunk_duration):
             chunk_file = os.path.join(output_directory, f"chunk_{i // chunk_duration + 1}.mp3")
-            ffmpeg.input(input_audio.name, ss=i, t=chunk_duration).output(chunk_file).run(
-                quiet=True, overwrite_output=True
-            )
+            ffmpeg.input(input_audio.name, ss=i, t=chunk_duration).output(chunk_file).run(quiet=True, overwrite_output=True)
             chunks.append(chunk_file)
-        
-        return chunks
+            print(f"Generated: {chunk_file}")
+
+        # Create a zip archive of the chunks
+        zip_filename = '/tmp/audio_chunks.zip'
+        shutil.make_archive(zip_filename.replace('.zip', ''), 'zip', output_directory)
+        return zip_filename
 
     except Exception as e:
-        error_message = f"Error processing file: {str(e)}"
-        print(error_message)
-        return error_message
+        print(f"Error during split: {e}")
+        return f"Error during file processing: {e}"
 
 # Gradio interface function
 def process_audio(input_audio):
-    output_directory = "/tmp/audio_chunks"
-    result = split_audio_ffmpeg(input_audio, output_directory, chunk_duration=30)
+    output_directory = "/tmp/output_chunks"
+    zip_file = split_audio_ffmpeg(input_audio, output_directory, chunk_duration=30)
+
+    # Check if zip creation was successful
+    if "Error" in zip_file:
+        return zip_file
 
-    if isinstance(result, str) and result.startswith("Error"):
-        return result
-    elif isinstance(result, list):
-        return [gr.File.update(value=chunk) for chunk in result]
-    else:
-        return "An unexpected error occurred. Please check the logs."
+    return gr.File(zip_file)
 
 # Gradio interface setup
 interface = gr.Interface(
     fn=process_audio,
     inputs=gr.File(label="Upload Audio", file_types=["audio"]),
-    outputs=gr.Files(label="Download Audio Chunks"),
+    outputs=gr.File(label="Download Chunks as Zip"),
     title="Audio Splitter",
-    description="Upload an audio file to split it into smaller chunks of 30 seconds each.",
+    description="Upload an audio file to split it into smaller chunks of 30 seconds each and download as a zip.",
 )
 
-# Launch the app
+# Launch the Gradio interface
 if __name__ == "__main__":
     interface.launch()