Update app.py

app.py

@@ -1,13 +1,14 @@
 import gradio as gr
 from pydub import AudioSegment
 
-def compress_audio(input_audio_path, compression_rate=128):
+def compress_audio(input_audio_path, compression_rate=128, output_name="compressed_audio.mp3"):
     """
     Compress an audio file.
     
     Args:
         input_audio_path: Path to the input audio file.
         compression_rate: Desired bitrate for compression (in kbps).
+        output_name: Desired name for the compressed file.
         
     Returns:
         Compressed audio file path.
@@ -16,13 +17,14 @@ def compress_audio(input_audio_path, compression_rate=128):
         # Load the audio file using pydub
         audio = AudioSegment.from_file(input_audio_path)
         
-        # Temporary output path
-        output_path = "compressed_audio.mp3"
+        # Ensure output name has the .mp3 extension
+        if not output_name.endswith(".mp3"):
+            output_name += ".mp3"
         
         # Export the compressed audio
-        audio.export(output_path, format="mp3", bitrate=f"{compression_rate}k")
+        audio.export(output_name, format="mp3", bitrate=f"{compression_rate}k")
         
-        return output_path
+        return output_name
     except Exception as e:
         return str(e)
 
@@ -32,10 +34,11 @@ interface = gr.Interface(
     inputs=[
         gr.Audio(label="Upload Audio File", type="filepath"),
         gr.Slider(32, 320, step=16, label="Compression Rate (kbps)", value=128),
+        gr.Textbox(label="Output File Name", placeholder="compressed_audio.mp3"),
     ],
     outputs=gr.File(label="Download Compressed Audio"),
     title="Audio Compressor",
-    description="Upload an audio file and select the compression rate to reduce the file size.",
+    description="Upload an audio file, set the compression rate, and specify a name for the compressed file.",
     allow_flagging="never",
 )