fix

app.py

@@ -26,6 +26,13 @@ def download_video(url):
         print(error_code)
     return error_code, info
 
+def clean_files():
+    if os.path.isfile('audio'):
+        os.remove('audio')
+    if os.path.isfile('audio.mp3'):
+        os.remove('audio.mp3')
+
+
 def main():
     """This method has a text input field, radio button and a button for downloading the video as mp3."""
     st.title('Youtube to mp3')
@@ -34,8 +41,9 @@ def main():
 
     if st.button('Download video'):
         with st.spinner('Downloading video'):
+            clean_files()
+            
             error_code, info = download_video(url)
-            print(info.keys())
 
             st.session_state['latest_video'] = url
             st.session_state['latest_title'] = info['fulltitle']
@@ -48,13 +56,17 @@ def main():
     if os.path.isfile('audio.mp3') and st.session_state.get('latest_video'):
         video_url = st.session_state.get('latest_video', '/')
         video_title = st.session_state.get('latest_title', '/')
+
         st.write(f"Last downloaded video is: {video_title} with url {video_url}")
         st.audio('audio.mp3')
         buffer = BytesIO()
         with open('audio.mp3', 'rb') as f:
             buffer.write(f.read())
         timestamp = datetime.now().strftime('%Y-%m-%d_%H-%M-%S')
-        st.download_button(label='Download mp3', data=buffer.getvalue(), file_name=f"{video_title.replace(' ', '-')}.mp3', mime='audio/mp3")
+        st.download_button(label='Download mp3',
+                           data=buffer.getvalue(),
+                           file_name=f"{video_title.replace(' ', '-')}.mp3",
+                           mime="audio/mp3")
 
 if __name__ == '__main__':
     main()