Update app.py

app.py

@@ -1,50 +1,63 @@
+import gradio as gr
 import os
 import ffmpeg
-import gradio as gr
 
-def split_audio_ffmpeg(input_audio, chunk_duration=30):
+# Function to validate and split audio
+def split_audio_ffmpeg(input_audio, output_directory, chunk_duration=30):
     """
-    Splits an audio file into smaller chunks using FFmpeg.
-    :param input_audio: The uploaded audio file
+    Split an audio file into smaller chunks using FFmpeg.
+    :param input_audio: Path to the input audio file
+    :param output_directory: Directory where output chunks will be saved
     :param chunk_duration: Duration of each chunk in seconds
     """
-    # Define input and output paths
-    input_audio_path = input_audio.name
-    output_directory = "./output_chunks"
     os.makedirs(output_directory, exist_ok=True)
 
     try:
-        # Get audio duration
-        probe = ffmpeg.probe(input_audio_path, v='error', select_streams='a', show_entries='stream=duration')
-        duration = float(probe['streams'][0]['duration'])
-        print(f"Audio duration: {duration} seconds")
+        # 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."
 
-        # Split audio into chunks
+        # 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_path, ss=i, t=chunk_duration).output(chunk_file).run(quiet=True)
-            print(f"Generated: {chunk_file}")
+            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
 
-        # Collect all chunk paths for download
-        return [os.path.join(output_directory, f) for f in os.listdir(output_directory)]
-    
     except Exception as e:
-        print(f"Error during splitting: {e}")
-        return []
+        error_message = f"Error processing file: {str(e)}"
+        print(error_message)
+        return error_message
 
-# Gradio interface
+# Gradio interface function
 def process_audio(input_audio):
-    chunks = split_audio_ffmpeg(input_audio)
-    if not chunks:
-        return "An error occurred during processing."
-    return chunks
+    output_directory = "/tmp/audio_chunks"
+    result = split_audio_ffmpeg(input_audio, output_directory, chunk_duration=30)
+
+    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."
 
-# Launch Gradio interface
+# Gradio interface setup
 interface = gr.Interface(
     fn=process_audio,
-    inputs=gr.File(label="Upload Audio File", file_types=["audio"]),
-    outputs=gr.File(label="Download Chunks", file_count="multiple"),
-    title="Audio Splitter"
+    inputs=gr.File(label="Upload Audio", file_types=["audio"]),
+    outputs=gr.Files(label="Download Audio Chunks"),
+    title="Audio Splitter",
+    description="Upload an audio file to split it into smaller chunks of 30 seconds each.",
 )
 
-interface.launch(share=True)
+# Launch the app
+if __name__ == "__main__":
+    interface.launch()