Update app.py

app.py

@@ -3,9 +3,10 @@ import os
 from audio_separator.separator import Separator
 
 # Initialize the Separator
-    separator = Separator()
 
-def separate_audio(input_file, output_dir, model_name):
+separator = Separator()
+
+def separate_audio(input_file, output_dir, model_name, output_format):
     # Create output directory if it doesn't exist
     os.makedirs(output_dir, exist_ok=True)
 
@@ -16,7 +17,7 @@ def separate_audio(input_file, output_dir, model_name):
         output_dir,
         model_name=model_name,
         denoise=True,
-        output_format='wav',
+        output_format=output_format,
         normalization_threshold=0.9,
         mdx_segment_size=256,
         mdx_overlap=8,
@@ -51,6 +52,13 @@ with iface:
                 label="Model",
                 value="UVR-MDX-NET-Inst_HQ_3"
             )
+        with gr.Column():
+            audio_format = gr.Dropdown(
+                choices=["wav", "mp3", "flac"], 
+                label="Audio Format",
+                value="wav"
+            )
+            
             submit_btn = gr.Button("Separate Audio")
         with gr.Column():
             with gr.Row():
@@ -60,7 +68,7 @@ with iface:
 
     submit_btn.click(
         process_audio,
-        inputs=[audio_input, model_name],
+        inputs=[audio_input, model_name, audio_format],
         outputs=[vocal_output, backing_vocal_output, instrumental_output]
     )