Jaman commited on
Commit
15cf5ef
·
verified ·
1 Parent(s): 0bef748

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +94 -0
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)