Update app.py

app.py

@@ -6,7 +6,6 @@ import os
 import numpy as np
 import base64
 
-genres = ["Pop", "Rock", "Jazz", "Electronic", "Hip-Hop", "Classical", "Lofi", "Chillpop"]
 
 @st.cache_resource()
 def load_model():
@@ -23,19 +22,24 @@ def generate_music_tensors(descriptions, duration: int):
     )
 
     with st.spinner("Generating Music..."):
+        st.markdown("### Generating Music... 🎵🎶🎹")
+
         output = model.generate(
             descriptions=descriptions,
             progress=True,
             return_tokens=True
         )
 
-    st.success("Music Generation Complete!")
+    st.success("Music Generation Complete! 🎉")
     return output
 
-
 def save_audio(samples: torch.Tensor):
     sample_rate = 30000
-    save_path = "/tmp/audio_output"
+    save_path = "/tmp/audio_output"  # Use /tmp directory
+
+    if not os.path.exists(save_path):
+        os.makedirs(save_path)
+
     assert samples.dim() == 2 or samples.dim() == 3
 
     samples = samples.detach().cpu()
@@ -44,23 +48,36 @@ def save_audio(samples: torch.Tensor):
 
     for idx, audio in enumerate(samples):
         audio_path = os.path.join(save_path, f"audio_{idx}.wav")
-        torchaudio.save(audio_path, audio, sample_rate)
+        try:
+            torchaudio.save(audio_path, audio, sample_rate)
+        except Exception as e:
+            st.error(f"Error saving audio file: {e}")
+            return None
+
+    return save_path
 
+
+    
+
+# Define the genres list
+genres = ["Pop", "Rock", "Jazz", "Electronic", "Hip-Hop", "Classical", "Lofi", "Chillpop"]
+
+
+# Add this function for downloading binary files
 def get_binary_file_downloader_html(bin_file, file_label='File'):
     with open(bin_file, 'rb') as f:
         data = f.read()
-    bin_str = base64.b64encode(data).decode()
-    href = f'<a href="data:application/octet-stream;base64,{bin_str}" download="{os.path.basename(bin_file)}">Download {file_label}</a>'
+        bin_str = base64.b64encode(data).decode()
+        href = f'<a href="data:application/octet-stream;base64,{bin_str}" download="{file_label}">Download {file_label}</a>'
     return href
 
 st.set_page_config(
     page_icon= "musical_note",
     page_title= "Music Gen"
 )
-
 def main():
     with st.sidebar:
-        st.header("""⚙️Generate Music ⚙️""",divider="rainbow")
+        st.header("""⚙️Generate Music ⚙️""", divider="rainbow")
         st.text("")
         st.subheader("1. Enter your music description.......")
         bpm = st.number_input("Enter Speed in BPM", min_value=60)
@@ -75,7 +92,7 @@ def main():
 
     st.title("""🎵 Song Lab AI 🎵""")
     st.text('')
-    left_co,right_co = st.columns(2)
+    left_co, right_co = st.columns(2)
     left_co.write("""Music Generation through a prompt""")
     left_co.write(("""PS : First generation may take some time ......."""))
     
@@ -93,7 +110,7 @@ def main():
             descriptions = [f"{text_area} {selected_genre} {bpm} BPM" for _ in range(5)]  # Adjust the batch size (5 in this case)
             music_tensors = generate_music_tensors(descriptions, time_slider)
 
-             # Only play the full audio for index 0
+            # Only play the full audio for index 0
             idx = 0
             music_tensor = music_tensors[idx]
             save_music_file = save_audio(music_tensor)
@@ -103,8 +120,9 @@ def main():
 
             # Play the full audio
             st.audio(audio_bytes, format='audio/wav')
-            st.markdown(get_binary_file_downloader_html(audio_filepath, f'Audio_{idx}'), unsafe_allow_html=True)
 
+            # Add download link
+            st.markdown(get_binary_file_downloader_html(audio_filepath, f'Audio_{idx}'), unsafe_allow_html=True)
 
 if __name__ == "__main__":
     main()