Create app.py
Browse files
app.py
ADDED
@@ -0,0 +1,94 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import streamlit as st
|
2 |
+
import librosa
|
3 |
+
import soundfile as sf
|
4 |
+
import os
|
5 |
+
import time
|
6 |
+
import numpy as np
|
7 |
+
|
8 |
+
# Set the home directory path for moonarch
|
9 |
+
home_directory = os.getcwd()
|
10 |
+
|
11 |
+
# Function to ensure directory exists
|
12 |
+
def ensure_directory_exists(directory):
|
13 |
+
if not os.path.exists(directory):
|
14 |
+
os.makedirs(directory)
|
15 |
+
|
16 |
+
# Function to process the audio
|
17 |
+
def process_audio(file_path, choice):
|
18 |
+
# Ensure to navigate to the home directory first
|
19 |
+
os.chdir(home_directory)
|
20 |
+
|
21 |
+
file_name_without_ext = os.path.splitext(os.path.basename(file_path))[0]
|
22 |
+
|
23 |
+
|
24 |
+
|
25 |
+
if choice == 'Basic Split':
|
26 |
+
output_dir = os.path.join('basic_splits', file_name_without_ext)
|
27 |
+
ensure_directory_exists(output_dir)
|
28 |
+
os.chdir('basic_splits')
|
29 |
+
from moonarch_basic import BasicSplitter
|
30 |
+
splitter = BasicSplitter(file_path)
|
31 |
+
splitter.run()
|
32 |
+
st.write("Basic split process completed.")
|
33 |
+
os.chdir(home_directory)
|
34 |
+
ensure_directory_exists(os.path.join('vocal_remover', file_name_without_ext))
|
35 |
+
os.chdir('vocal_remover')
|
36 |
+
from moonarch_vocal_remover import VocalRemover
|
37 |
+
music_sep = VocalRemover(file_path)
|
38 |
+
music_sep.run()
|
39 |
+
st.write("Vocal remover process completed.")
|
40 |
+
os.chdir(home_directory)
|
41 |
+
return [
|
42 |
+
os.path.join('basic_splits', file_name_without_ext, 'vocals.wav'),
|
43 |
+
os.path.join('vocal_remover', file_name_without_ext, 'accompaniment.wav'),
|
44 |
+
os.path.join('basic_splits', file_name_without_ext, 'bass.wav'),
|
45 |
+
os.path.join('basic_splits', file_name_without_ext, 'drums.wav')
|
46 |
+
]
|
47 |
+
|
48 |
+
|
49 |
+
# Streamlit app
|
50 |
+
st.title('Monarch Music Analysis: Basic Splits')
|
51 |
+
|
52 |
+
task_choice = st.selectbox('Choose a task', ['Basic Split'])
|
53 |
+
|
54 |
+
uploaded_file = st.file_uploader('Upload an audio file', type=['mp3', 'wav'])
|
55 |
+
|
56 |
+
if uploaded_file is not None:
|
57 |
+
# Save uploaded file to a temporary location
|
58 |
+
audio_file_path = os.path.join(home_directory, f'{uploaded_file.name}')
|
59 |
+
with open(audio_file_path, 'wb') as f:
|
60 |
+
f.write(uploaded_file.getbuffer())
|
61 |
+
|
62 |
+
y, sr = librosa.load(audio_file_path, sr=None) # sr=None to use the native sample rate
|
63 |
+
|
64 |
+
if st.button('Start'):
|
65 |
+
progress_bar = st.progress(0)
|
66 |
+
|
67 |
+
# Simulate progress bar for processing
|
68 |
+
for percent_complete in range(100):
|
69 |
+
time.sleep(0.01)
|
70 |
+
progress_bar.progress(percent_complete + 1)
|
71 |
+
|
72 |
+
output_files = process_audio(audio_file_path, task_choice)
|
73 |
+
|
74 |
+
st.success('Audio processed successfully!')
|
75 |
+
|
76 |
+
for file in output_files:
|
77 |
+
file_name = os.path.basename(file)
|
78 |
+
if 'vocals.wav' in file:
|
79 |
+
st.write('Vocal')
|
80 |
+
elif 'accompaniment.wav' in file:
|
81 |
+
st.write('Music')
|
82 |
+
elif 'bass.wav' in file:
|
83 |
+
st.write('Bass')
|
84 |
+
elif 'drums.wav' in file:
|
85 |
+
st.write('Drums')
|
86 |
+
elif 'piano.wav' in file:
|
87 |
+
st.write('Piano')
|
88 |
+
elif 'strings_or_pads.wav' in file:
|
89 |
+
st.write('Strings or Pads')
|
90 |
+
|
91 |
+
st.audio(file)
|
92 |
+
with open(file, 'rb') as f:
|
93 |
+
new_file_name = 'music.wav' if file_name == 'accompaniment.wav' else ('strings_or_pads.wav' if file_name == 'other.wav' else file_name)
|
94 |
+
st.download_button(label=f'Download {new_file_name}', data=f, file_name=new_file_name)
|