changed UI, FX params and sample prompts

app.py

@@ -1,21 +1,19 @@
 import streamlit as st
 import os
-import librosa
-import soundfile as sf
 import numpy as np
 from inference import generate_drum_kit
 from audio_utils import play_audio
 from fx import get_fx
 
 # Streamlit UI
-st.title("text2kit")
+st.title("semantic spaces: kit generator")
 st.subheader("generate drum kits and audio effects with text prompts")
 st.write("uses publicly available samples from [freesound](https://zenodo.org/records/4687854) and [CLAP embeddings](https://github.com/LAION-AI/CLAP) for text-based querying")
 st.write("hint: turn audio effects on! try weird prompts!")
 
 with st.container(border=True):
     # User Inputs
-    prompt = st.text_input("Describe your drum kit:", "warm vintage acoustic")
+    prompt = st.text_input("Describe your drum kit:", "warm vintage organic percussion")
     kit_size = st.slider("Number of sounds per instrument:", 1, 10, 4)
     use_fx = st.toggle("Apply audio effects?", value=True)
     if use_fx:

fx.py

@@ -41,7 +41,9 @@ def apply_fx(audio_path, params, write_wav=True, output_dir="processed_audio"):
         Bitcrush(bit_depth=params['bit_depth']),
         Reverb(room_size=params['reverb_size'], wet_level=params['reverb_wet'])
     ])
+
     processed_audio = board(audio, sr)
+    
     if write_wav:
         # Determine output directory dynamically
         base_dir = os.path.dirname(os.path.dirname(audio_path))  # Get 'dataset' level
@@ -84,12 +86,12 @@ def get_params_dict(params_list):
 
 # Define parameter search space
 search_space = [
-    Real(5000, 15000, name="lowpass"),
+    Real(4000, 20000, name="lowpass"),
     Real(50, 1000, name="highpass"),
     Real(0.0, 0.8, name="reverb_size"),
     Real(0.0, 1.0, name="reverb_wet"),
-    Real(0.0, 20.0, name="drive_db"),
-    Real(6.0, 32.0, name="bit_depth")
+    Real(0.0, 10.0, name="drive_db"),
+    Real(4.0, 32.0, name="bit_depth")
 ]
 
 ##### Main function #####