Update app.py

app.py

@@ -34,6 +34,7 @@ def generate(prompt):
     return audio_buffer
 
 st.title("Music Generator")
+st.header("Select an example or write a text prompt")
 
 text_prompt = st.text_input("Text Prompt", "")
 
@@ -42,18 +43,19 @@ examples = [
     "Earthy tones, environmentally conscious, ukulele-infused, harmonic, breezy, easygoing, organic instrumentation, gentle grooves",
     "90s rock song with loud guitars and heavy drums",
     "Heartful EDM with beautiful synths and chords",
+    "None"
 ]
 
 st.title("Examples")
 selected_example = st.radio("Select an example", examples)
 
 if st.button("Generate Audio"):
-    if selected_example or text_prompt:
+    if selected_example != "None" or text_prompt:
         prompt = ""
-        if selected_example:
-            prompt = selected_example
-        else:
+        if text_prompt:
             prompt = text_prompt
+        else:
+            prompt = selected_example
         with st.spinner("Generating audio..."):
             audio_output = generate(prompt)
             st.audio(audio_output, format='audio/wav')
@@ -62,6 +64,6 @@ if st.button("Generate Audio"):
 
 if st.checkbox("Show debug info"):
     if text_prompt:
-        st.write("Text Prompt:", text_prompt)
-    if selected_example:
-        st.write("Text Prompt:", selected_example)
+        st.write("Prompt:", text_prompt)
+    else:
+        st.write("Prompt:", selected_example)