Update app.py

app.py

@@ -582,57 +582,31 @@ def download_audio():
         on_click=update_download_count
     )
 
-def notify_admin_of_generation(user_id, genre, energy_level, tempo, description, duration, perm_link, product_code):
+def notify_admin_of_generation(perm_link, product_code):
     """
-    Send a notification email to [email protected] when a song is generated
+    Notify admin of new song generation using WordPress REST API
     """
-    sender_email = "[email protected]"  # Your email
-    sender_password = os.getenv("EMAIL_PASSWORD")  # Store this in your environment variables
-    
-    # Create the email
-    msg = MIMEMultipart()
-    msg["From"] = sender_email
-    msg["To"] = sender_email  # Sending to yourself
-    msg["Subject"] = f"New Song Generated - {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}"
-    
-    # Create the email body with generation details
-    body = f"""
-    New Song Generation Details:
-    
-    Time: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}
-    User ID: {user_id}
-    
-    Song Parameters:
-    - Genre: {genre}
-    - Energy Level: {energy_level}
-    - Tempo: {tempo}
-    - Duration: {duration} seconds
-    - Description: {description}
-    
-    Links:
-    - Permanent Link: {perm_link}
-    - Product Code: {product_code}
-    
-    --
-    Automated notification from SongLab AI
-    """
-    
-    msg.attach(MIMEText(body, "plain"))
-    
     try:
-        # Connect to your email server (update these settings based on your email provider)
-        server = smtplib.SMTP("smtp.gmail.com", 587)  # Update with your SMTP server
-        server.starttls()
-        server.login(sender_email, sender_password)
+        notification_url = "https://songlabai.com/wp-json/songlab/v1/notify"
         
-        # Send the email
-        server.send_message(msg)
-        server.quit()
+        # Prepare the notification data
+        payload = {
+            "link": perm_link,
+            "product_code": product_code
+        }
+        
+        # Send the notification
+        response = requests.post(notification_url, json=payload)
         
-        print("✓ Admin notification email sent successfully")
+        # Check if notification was successful
+        if response.status_code == 200:
+            print("✓ Admin notification sent successfully")
+        else:
+            print("× Failed to send admin notification")
+            
     except Exception as e:
-        print(f"× Failed to send admin notification email: {str(e)}")
-    
+        print(f"× Error sending notification: {str(e)}")
+        
 def update_download_count():
     auth_headers = get_auth_headers()
     update_download_url = "https://songlabai.com/wp-json/custom-api/v1/update-download-count"
@@ -689,17 +663,9 @@ def load_and_play_generated_audio(response, genre=None, energy_level=None, tempo
         except:
             user_id = 'Unknown'
 
-    # Send email notification using the separate function
-    notify_admin_of_generation(
-        user_id=user_id,
-        genre=genre,
-        energy_level=energy_level,
-        tempo=tempo,
-        description=description,
-        duration=duration,
-        perm_link=perm_link,
-        product_code=product_code
-    )
+    # Send notification using simplified function
+    notify_admin_of_generation(perm_link, product_code)
+
 
     # Define col_btn and col_text
     col_btn, col_text = st.columns([2, 4])