adding the removed code for song push to wordpress

app.py

@@ -13,6 +13,14 @@ from io import BytesIO
 import soundfile as sf
 from datetime import datetime
 from tempfile import NamedTemporaryFile
+from wordpress_xmlrpc import Client
+from wordpress_xmlrpc.methods import media
+from xmlrpc.client import Binary
+from woocommerce import API
+from wordpress_xmlrpc.compat import xmlrpc_client
+from io import BytesIO
+import soundfile as sf
+import uuid
 
 # Try to get API_URL from environment variables, if not found set to a default value
 try:
@@ -28,6 +36,84 @@ except KeyError:
     st.error("BEARER_TOKEN environment variable is not set.")
     st.stop()
 
+def save_to_wordpress(audio, sample_rate):
+    # Define your WordPress site URL and authentication credentials
+    wordpress_url = 'https://songlabai.com/xmlrpc.php'
+    woocommerce_url = 'https://songlabai.com'
+    consumer_key = 'ck_93d516ba12289a6fd0eced56bbc0b05ecbf98735'
+    consumer_secret = 'cs_9d5eb716d631db408a4c47796b5d18b0313d8559'
+    username = 'admin_h2ibbgql'
+    password = 'um^VdaNK0H8Vw2*KNJlYABkh'
+
+    # Authenticate with WordPress XML-RPC API
+    wav_bytes = BytesIO()
+    sf.write(wav_bytes, audio, samplerate=sample_rate, format='WAV')
+    title = f"generated_audio_{datetime.now().timestamp()}.wav"
+    file_data = {
+        'name': title,
+        'type': 'audio/x-wav',  # Change the MIME type according to your file type
+        'bits': xmlrpc_client.Binary(wav_bytes.getvalue()),
+    }
+    wp_client = Client(wordpress_url, username, password)
+    for _ in range(4):
+        try:
+            # Upload the file to WordPress Media Library
+
+            media_response = wp_client.call(media.UploadFile(file_data))
+
+            # Handle the response
+            if media_response:
+                print("File successfully uploaded to WordPress with attachment ID:", media_response)
+
+                # Create product data for WooCommerce
+                product_data = {
+                    'status':'pending',
+                    'name': title,
+                    'type': 'simple',
+                    'regular_price': '1.00',  # Set the price as needed
+                    'sku': str(uuid.uuid4()),
+                    'downloadable': True,
+                    'download_limit': -1,
+                    'download_expiry': -1,
+                }
+
+                # Authenticate with WooCommerce API
+                wc_api = API(
+                    url=woocommerce_url,
+                    consumer_key=consumer_key,
+                    consumer_secret=consumer_secret,
+                    version="wc/v3"
+                )
+
+                # Create the product
+                response = wc_api.post("products", product_data)
+
+                # Handle the response
+                if response.status_code == 201:
+                    print("Product successfully created in WooCommerce:", response.json())
+                    # Update product to add downloadable file URL
+                    product_update_data = {
+                        'downloads': [{
+                            'name': media_response['title'],
+                            'file': media_response['link']
+                        }]
+                    }
+                    product_id = response.json().get('id')
+                    response = wc_api.put(f"products/{product_id}", product_update_data)
+
+                    if response.status_code == 200:
+                        print("Downloadable file URL added to product:", response.json())
+                        return response.json()['permalink'], response.json()['permalink'].split('p=')[-1]
+                    else:
+                        print("Error adding downloadable file URL to product:", response.text)
+                else:
+                    print("Error creating product in WooCommerce:", response.text)
+            else:
+                print("Error uploading file to WordPress.")
+            break
+        except Exception as e:
+            print("Error:", e)
+
 headers = {
     "Authorization": f"Bearer {BEARER_TOKEN}",
     "Content-Type": "application/json"
@@ -126,9 +212,13 @@ if st.button("Generate Audio") and genre and energy_level and description and te
             st_state.audio_pydub_sample_rate = audio_pydub.frame_rate
         st_state.audio = np.array(output[0]).astype(np.float32)
         sample_rate = eval(response.content)[0]['sample_rate']
+        perm_link, product_code = save_to_wordpress(np.array(output[0]).astype(np.float32), sample_rate)
         st_state.audio_sample_rate = sample_rate
         st_state.augmented_audio_pydub = st_state.audio_pydub
         st.audio(st_state.audio_pydub.export().read())
+        st.write(f"To download use the following product code: {product_code}\nTo Publish, please contact the admin by sending the following link: {perm_link}")
+        st.link_button("Contact Us", "https://songlabai.com/subcribe/")
+        st.link_button("Download Song", "https://songlabai.com/download-music/")
 
 # Post-processing options
 st.header("Post-processing Options")