updated UI

app.py

@@ -8,25 +8,30 @@ from audio_utils import play_audio
 from fx import get_fx
 
 # Streamlit UI
-st.title("Generate Drum Kits with Text")
+st.title("text2kit")
+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!")
 
-# User Inputs
-prompt = st.text_input("Describe your drum kit:", "warm vintage acoustic")
-kit_size = st.slider("Number of sounds per instrument:", 1, 10, 4)
-use_fx = st.toggle("Apply audio effects?")
-if use_fx:
-    if st.toggle("Use a different prompt for audio effects?"):
-        fx_prompt = st.text_input("Describe your desired FX tone:", "soft and ethereal telephone")
-    else:
-        fx_prompt = prompt
-
-# Run the inference
-if st.button("Generate Drum Kit"):
-    drum_kit = generate_drum_kit(prompt, kit_size)
+with st.container(border=True):
+    # User Inputs
+    prompt = st.text_input("Describe your drum kit:", "warm vintage acoustic")
+    kit_size = st.slider("Number of sounds per instrument:", 1, 10, 4)
+    use_fx = st.toggle("Apply audio effects?", value=True)
     if use_fx:
-        drum_kit, fx_params = get_fx(drum_kit, fx_prompt)
-        st.write(fx_params)
-    st.session_state["drum_kit"] = drum_kit  # Store results
+        if st.toggle("Use a different prompt for audio effects?", value=True):
+            fx_prompt = st.text_input("Describe your desired FX tone:", "soft and ethereal space")
+        else:
+            fx_prompt = prompt
+
+    # Run the inference
+    if st.button("Generate Drum Kit"):
+        drum_kit = generate_drum_kit(prompt, kit_size)
+        st.session_state["dry_kit"] = drum_kit
+        if use_fx:
+            drum_kit, fx_params = get_fx(drum_kit, fx_prompt)
+            st.session_state["fx_params"] = fx_params
+        st.session_state["drum_kit"] = drum_kit  # Store results
 
 # Display results
 if "drum_kit" in st.session_state:
@@ -40,4 +45,11 @@ if "drum_kit" in st.session_state:
         for i, sound_file in enumerate(sounds):
             with cols[i]:
                 if st.button(f"▶️ {os.path.basename(sound_file)}", key=sound_file):
-                    play_audio(sound_file)
+                    play_audio(sound_file)
+    
+    if st.toggle("Show parameters?"):
+        if "fx_params" in st.session_state:
+            st.subheader("FX Parameters")
+            st.write(st.session_state["fx_params"])
+        if "dry_kit" in st.session_state:
+            st.write(st.session_state["dry_kit"])

fx.py

@@ -87,7 +87,7 @@ search_space = [
     Real(5000, 15000, name="lowpass"),
     Real(50, 1000, name="highpass"),
     Real(0.0, 0.8, name="reverb_size"),
-    Real(0.0, 0.8, name="reverb_wet"),
+    Real(0.0, 1.0, name="reverb_wet"),
     Real(0.0, 20.0, name="drive_db"),
     Real(6.0, 32.0, name="bit_depth")
 ]