Update app.py

app.py

@@ -13,9 +13,16 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 try:
     subprocess.run(["sox", "--version"], check=True, capture_output=True)
 except FileNotFoundError:
-    print("sox is not installed. Installing it now...")
-    subprocess.run(["apt-get", "install", "-y", "sox"], check=True)
-    print("sox has been installed.")
+    print("sox is not installed. Trying to install it now...")
+    try:
+        subprocess.run(["apt-get", "update"], check=True)
+        subprocess.run(["apt-get", "install", "-y", "sox"], check=True)
+        print("sox has been installed.")
+    except subprocess.CalledProcessError as e:
+        print(f"Error installing sox: {e}")
+        print("Please install sox manually or try adding the following repository to your sources list:")
+        print("deb http://deb.debian.org/debian stretch main contrib non-free")
+        exit(1)
 
 # Define the inference function
 @spaces.GPU
@@ -69,6 +76,13 @@ def inference(audio_file, model_name, vocals, drums, bass, other, mp3, mp3_bitra
         sox_cmd = ["sox", "-m"] + selected_stems + [output_file]
         subprocess.run(sox_cmd, check=True)
 
+    # Automatically convert to MP3 if requested
+    if mp3:
+        mp3_output_file = os.path.splitext(output_file)[0] + ".mp3"
+        ffmpeg_cmd = ["ffmpeg", "-y", "-i", output_file, "-ab", str(mp3_bitrate) + "k", mp3_output_file]
+        subprocess.run(ffmpeg_cmd, check=True)
+        output_file = mp3_output_file  # Update output_file to the MP3 file
+
     return output_file
 
 # Define the Gradio interface