Update app.py

app.py

@@ -43,14 +43,22 @@ if generate_btn:
     cover2 = generate_image("Generate a cover for this song:" + prompt)
     cover3 = generate_image("Generate a thumbnail for this song:" + prompt)
 
-    st.image(cover1)
-    st.header(title1)
-    st.audio(audio1, format="audio/wav")
+    col1, col2, col3 = st.columns(3)
 
-    st.image(cover2)
-    st.header(title2)
-    st.audio(audio2, format="audio/wav")
+    with col1:
+        st.image(cover1)
+        st.header(title1)
+        st.audio(audio1, format="audio/wav")
+        st.download_button("Download Audio 1", audio1, file_format="wav")
 
-    st.image(cover3)
-    st.header(title3)
-    st.audio(audio3, format="audio/wav")
+    with col2:
+        st.image(cover2)
+        st.header(title2)
+        st.audio(audio2, format="audio/wav")
+        st.download_button("Download Audio 2", audio2, file_format="wav")
+
+    with col3:
+        st.image(cover3)
+        st.header(title3)
+        st.audio(audio3, format="audio/wav")
+        st.download_button("Download Audio 3", audio3, file_format="wav")