Update app.py

app.py

@@ -86,36 +86,35 @@ 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 ......."""))
     
     if st.sidebar.button('Generate !'):
-    with left_co:
-        st.text('')
-        st.text('')
-        st.text('')
-        st.text('')
-        st.text('')
-        st.text('')
-        st.subheader("Generated Music")
-
-        # Generate audio
-        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
-        idx = 0
-        music_tensor = music_tensors[idx]
-        save_music_file = save_audio(music_tensor)
-        audio_filepath = f'audio_output/audio_{idx}.wav'
-        audio_file = open(audio_filepath, 'rb')
-        audio_bytes = audio_file.read()
-
-        # 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)
-
+        with left_co:
+            st.text('')
+            st.text('')
+            st.text('')
+            st.text('')
+            st.text('')
+            st.text('')
+            st.subheader("Generated Music")
+
+            # Generate audio
+            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
+            idx = 0
+            music_tensor = music_tensors[idx]
+            save_music_file = save_audio(music_tensor)
+            audio_filepath = f'audio_output/audio_{idx}.wav'
+            audio_file = open(audio_filepath, 'rb')
+            audio_bytes = audio_file.read()
+
+            # 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)
 
 if __name__ == "__main__":
     main()