Update app.py

app.py

@@ -12,6 +12,10 @@ def ensure_directory_exists(directory):
     if not os.path.exists(directory):
         os.makedirs(directory)
 
+# Function to check if the file exists
+def file_exists(file_path):
+    return os.path.isfile(file_path)
+
 # Function to process the audio
 def process_audio(file_path):
     # Ensure to navigate to the home directory first
@@ -75,21 +79,24 @@ if uploaded_file is not None:
 
             # Display and download each generated audio
             for file in output_files:
-                file_name = os.path.basename(file)
-                if 'vocals.wav' in file_name:
-                    st.write('Vocal')
-                elif 'bass.wav' in file_name:
-                    st.write('Bass')
-                elif 'drums.wav' in file_name:
-                    st.write('Drums')
-                elif 'accompaniment.wav' in file_name:
-                    st.write('Music')
-                
-                st.audio(file)
-                with open(file, 'rb') as f:
-                    st.download_button(
-                        label=f'Download {file_name}',
-                        data=f,
-                        file_name=file_name,
-                        mime='audio/wav'
-                    )
+                if file_exists(file):
+                    file_name = os.path.basename(file)
+                    if 'vocals.wav' in file_name:
+                        st.write('Vocal')
+                    elif 'bass.wav' in file_name:
+                        st.write('Bass')
+                    elif 'drums.wav' in file_name:
+                        st.write('Drums')
+                    elif 'accompaniment.wav' in file_name:
+                        st.write('Music')
+                    
+                    st.audio(file)
+                    with open(file, 'rb') as f:
+                        st.download_button(
+                            label=f'Download {file_name}',
+                            data=f,
+                            file_name=file_name,
+                            mime='audio/wav'
+                        )
+                else:
+                    st.error(f"Error: {file} not found.")