Update app.py

app.py

@@ -54,11 +54,16 @@ background-size: cover;
 '''
 st.markdown(page_bg_img, unsafe_allow_html=True)
 
-def save_to_wordpress(channel1, channel2, sample_rate):
 
-    channel1 = np.array(channel1).astype(np.float32)
-    channel2 = np.array(channel2).astype(np.float32)
-    stereo_audio = np.column_stack((channel1, channel2))
+
+def save_to_wordpress(audio_array, sample_rate):
+    # Convert audio_array to float32 if not already
+    audio_array = audio_array.astype(np.float32)
+
+    # Save the audio to a BytesIO buffer
+    wav_bytes = BytesIO()
+    sf.write(wav_bytes, audio_array, samplerate=sample_rate, format='WAV')
+    wav_bytes.seek(0)
 
     # Define your WordPress site URL and authentication credentials
     wordpress_url = "https://songlabai.com/xmlrpc.php"
@@ -69,8 +74,6 @@ def save_to_wordpress(channel1, channel2, sample_rate):
     password = "um^VdaNK0H8Vw2*KNJlYABkh"
 
     # Authenticate with WordPress XML-RPC API
-    wav_bytes = BytesIO()
-    sf.write(wav_bytes, stereo_audio, samplerate=sample_rate, format="WAV")
     title = f"generated_audio_{datetime.now().timestamp()}.wav"
     file_data = {
         "name": title,
@@ -81,7 +84,6 @@ def save_to_wordpress(channel1, channel2, sample_rate):
     for _ in range(4):
         try:
             # Upload the file to WordPress Media Library
-
             media_response = wp_client.call(media.UploadFile(file_data))
 
             # Handle the response
@@ -151,6 +153,10 @@ def save_to_wordpress(channel1, channel2, sample_rate):
             break
         except Exception as e:
             print("Error:", e)
+            continue  # Retry on error
+
+    # If upload fails, return placeholders
+    return "https://songlabai.com/contact_us/", "N/A"
 
 
 
@@ -525,9 +531,15 @@ def load_and_play_generated_audio(response):
     # Play the audio
     st.audio(audio_buffer.getvalue())
 
+    # Save the audio to WordPress and get perm_link and product_code
+    perm_link, product_code = save_to_wordpress(audio_array, sample_rate)
+
+    # Define col_btn and col_text
+    col_btn, col_text = st.columns([2, 4])
+
     with col_btn:
-        st.link_button("Publish your Song", "https://songlabai.com/contact_us/")
-        st.link_button("Download Song", "https://songlabai.com/download-music/")
+        st.markdown("[Publish your Song](https://songlabai.com/contact_us/)")
+        st.markdown("[Download Song](https://songlabai.com/download-music/)")
 
     with col_text:
         st.write(