Update app.py

app.py

@@ -24,6 +24,9 @@ from wordpress_xmlrpc import Client
 from wordpress_xmlrpc.compat import xmlrpc_client
 from wordpress_xmlrpc.methods import media
 
+import smtplib
+from email.mime.text import MIMEText
+from email.mime.multipart import MIMEMultipart
 
 
 # Try to get API_URL from environment variables, if not found set to a default value
@@ -478,7 +481,7 @@ def generate_audio(genre, energy_level, tempo, description, duration):
         # First, verify user is logged in and has permissions
         if st.session_state.get("jwt_token") is None:
             st.error("You must be logged in to generate audio.")
-            return
+            return None
                 
         auth_headers = get_auth_headers()
         
@@ -487,7 +490,7 @@ def generate_audio(genre, energy_level, tempo, description, duration):
         subscription_response = requests.get(subscription_url, headers=auth_headers)
         if subscription_response.status_code != 200:
             st.error("Failed to retrieve your subscription status.")
-            return
+            return None
                 
         subscription_data = subscription_response.json()
         
@@ -528,7 +531,7 @@ def generate_audio(genre, energy_level, tempo, description, duration):
                     "Consider upgrading to premium for reliable, fast access.\n\n"
                     "👉 [Upgrade Now](https://songlabai.com/subscribe/)"
                 )
-                return
+                return None
                     
         # Clear warmup messages
         for placeholder in [status_placeholder, progress_placeholder, subscription_placeholder, stage_placeholder]:
@@ -545,11 +548,21 @@ def generate_audio(genre, energy_level, tempo, description, duration):
             
             if response and response.status_code == 200:
                 st.success("✨ Music generated successfully!")
-                load_and_play_generated_audio(response)
+                # Pass all parameters to load_and_play_generated_audio
+                load_and_play_generated_audio(
+                    response=response,
+                    genre=genre,
+                    energy_level=energy_level,
+                    tempo=tempo,
+                    description=description,
+                    duration=duration
+                )
                 
                 # Update generation count
                 update_generation_url = "https://songlabai.com/wp-json/custom-api/v1/update-generation-count"
                 requests.post(update_generation_url, headers=auth_headers)
+                
+                return response  # Return the response for email notification
             else:
                 st.error(
                     "❌ Failed to generate audio.\n\n"
@@ -557,8 +570,10 @@ def generate_audio(genre, energy_level, tempo, description, duration):
                     "💡 **Tip:** Premium users experience fewer failures and faster generation times.\n\n"
                     "👉 [Upgrade Now](https://songlabai.com/subscribe/)"
                 )
+                return None
     except Exception as e:
         st.error(f"An unexpected error occurred: {str(e)}")
+        return None
 
 def download_audio():
     if st_state.audio_pydub is None:
@@ -576,6 +591,57 @@ def download_audio():
         mime="audio/wav",
         on_click=update_download_count
     )
+
+def notify_admin_of_generation(user_id, genre, energy_level, tempo, description, duration, perm_link, product_code):
+    """
+    Send a notification email to [email protected] when a song is generated
+    """
+    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)
+        
+        # Send the email
+        server.send_message(msg)
+        server.quit()
+        
+        print("✓ Admin notification email sent successfully")
+    except Exception as e:
+        print(f"× Failed to send admin notification email: {str(e)}")
     
 def update_download_count():
     auth_headers = get_auth_headers()
@@ -586,8 +652,8 @@ def update_download_count():
         return
 
 
-def load_and_play_generated_audio(response):
-    import soundfile as sf  # Ensure soundfile is imported
+def load_and_play_generated_audio(response, genre=None, energy_level=None, tempo=None, description=None, duration=None):
+    import soundfile as sf
     from io import BytesIO
 
     response_eval = json.loads(response.content)
@@ -596,13 +662,11 @@ def load_and_play_generated_audio(response):
 
     # Check if the audio is stereo or mono
     if isinstance(generated_audio[0], list):  # Stereo
-        # Transpose the list to get channels in columns
         audio_array = np.array(generated_audio).T
     else:  # Mono
         audio_array = np.array(generated_audio)
 
     # Convert the float32 audio data to int16
-    # Assuming the audio data is in range [-1.0, 1.0], scale accordingly
     max_val = np.max(np.abs(audio_array))
     if max_val > 0:
         audio_array = audio_array / max_val  # Normalize to [-1.0, 1.0]
@@ -611,7 +675,7 @@ def load_and_play_generated_audio(response):
     # Write the audio data to a BytesIO buffer using soundfile
     audio_buffer = BytesIO()
     sf.write(audio_buffer, int_audio, sample_rate, format='WAV', subtype='PCM_16')
-    audio_buffer.seek(0)  # Reset buffer position to the beginning
+    audio_buffer.seek(0)
 
     # Read the audio data into an AudioSegment
     audio_segment = AudioSegment.from_file(audio_buffer, format="wav")
@@ -626,6 +690,27 @@ def load_and_play_generated_audio(response):
     # Save the audio to WordPress and get perm_link and product_code
     perm_link, product_code = save_to_wordpress(audio_array, sample_rate)
 
+    # Get user ID from JWT token
+    user_id = None
+    if st.session_state.get("jwt_token"):
+        try:
+            decoded = jwt.decode(st.session_state["jwt_token"], options={"verify_signature": False})
+            user_id = decoded.get('user_id', 'Unknown')
+        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
+    )
+
     # Define col_btn and col_text
     col_btn, col_text = st.columns([2, 4])
 
@@ -639,7 +724,6 @@ def load_and_play_generated_audio(response):
         )
         st.write(f"To download use the following product code: {product_code}")
 
-
 if st.button("Generate Audio", key="generate_audio_button"):
     if genre and energy_level and description and tempo:
         if st.session_state.get("jwt_token") is None: