updating uploading to wordpress function

app.py

@@ -10,38 +10,92 @@ import soundfile as sf
 from scipy.signal import butter, sosfilt
  
 from wordpress_xmlrpc.compat import xmlrpc_client
-from xmlrpc.client import ServerProxy
 from io import BytesIO
 import soundfile as sf
 from datetime import datetime
 
+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  # This library is used for saving audio data as WAV file
+from datetime import datetime
+import uuid
+
 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
-    proxy = ServerProxy(wordpress_url)
     wav_bytes = BytesIO()
     sf.write(wav_bytes, audio, samplerate=sample_rate, format='WAV')
-
+    title = f"generated_audio_{datetime.now().timestamp()}.wav"
     file_data = {
-        'name': f"generated_audio_{datetime.now().timestamp()}.wav",
+        '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
-            response_upload = proxy.wp.uploadFile(0, username, password, file_data)
+
+            media_response = wp_client.call(media.UploadFile(file_data))
 
             # Handle the response
-            if response_upload['id']:
-                print(response_upload)
-                attachment_id = response_upload['id']
-                print("File successfully uploaded to WordPress with attachment ID:", attachment_id)
+            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']
+                    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
@@ -114,9 +168,9 @@ if st.button("Generate Audio") and genre and energy_level and description and te
     audio = np.array(response['generated_audio'], dtype=np.float32)
     sample_rate = response['sample_rate']
     st_state.audio = audio
-    save_to_wordpress(audio, sample_rate)
+    permalink = save_to_wordpress(audio, sample_rate)
     st.audio(st_state.audio, format="audio/wav", sample_rate=sample_rate, start_time=0)
-
+    st.text(f"For downloading/publishing please contact the administrator, from your loged in accout. Remember to send the following link:\n{permalink}")
 # Post-processing options
 st.header("Post-processing Options")