Update app.py

app.py

@@ -1,6 +1,7 @@
 import streamlit as st
 from pytube import YouTube
 import subprocess
+import os
 
 def download_youtube_video(url):
     yt = YouTube(url)
@@ -25,7 +26,10 @@ def main():
             mp3_file = convert_to_mp3(video_filename)
             st.success(f"Conversion completed! MP3 file saved as: {mp3_file}")
             st.text('Download your file below:')
-            st.file_download(mp3_file)
+            if os.path.exists(mp3_file):
+                with open(mp3_file, 'rb') as f:
+                    data = f.read()
+                st.download_button(label='Click here to download', data=data, file_name='converted_audio.mp3', mime='audio/mp3')
 
 if __name__ == "__main__":
     main()