Update app.py

app.py

@@ -268,15 +268,40 @@ energy_level = st.radio("Energy Level:", energy_levels, horizontal=True)
 description = st.text_input("Description:", "")
 tempo = st.slider("Tempo (in bpm):", min_value=40, max_value=100, value=60, step=5)
 
-# Always show the full duration slider initially
-duration = st.slider(
-    "Duration (in seconds):",
-    min_value=15,
-    max_value=300,
-    value=30,
-    step=1
-)
-
+# First, check subscription status if user is logged in
+if st.session_state.get("jwt_token"):
+    auth_headers = get_auth_headers()
+    subscription_url = "https://songlabai.com/wp-json/custom-api/v1/subscription"
+    subscription_response = requests.get(subscription_url, headers=auth_headers)
+    
+    if subscription_response.status_code == 200:
+        subscription_data = subscription_response.json()
+        subscription_plan_id = subscription_data.get("subscription_plan_id")
+        # Free tier has plan ID 576 or null
+        is_free_tier = not subscription_plan_id or subscription_plan_id == 576
+    else:
+        is_free_tier = True  # Default to free tier if unable to verify
+else:
+    is_free_tier = True  # Default to free tier if not logged in
+
+# Show appropriate duration slider based on subscription tier
+if is_free_tier:
+    duration = st.slider(
+        "Duration (in seconds):",
+        min_value=15,
+        max_value=30,  # Restricted to 30 seconds for free tier
+        value=30,
+        step=1,
+        help="Free tier users are limited to 30-second generations. Upgrade to create longer tracks!"
+    )
+else:
+    duration = st.slider(
+        "Duration (in seconds):",
+        min_value=15,
+        max_value=300,
+        value=30,
+        step=1
+    )
 
 
 def convert_audio_segment_to_float_array(audio_pydub):
@@ -351,28 +376,32 @@ def wait_for_server_warmup(status_placeholder, progress_placeholder, subscriptio
     ### 📋 **Subscription Tiers**
 
     **Free User**
-    - 🎵 **Generations:** 3 music generations per month.
-    - 📥 **Downloads:** No downloads allowed.
-    - ⏳ **Wait Time:** Up to 7 minutes when the server is inactive.
-
-    **Tier 2 - $19.99/month**
-    - 🎵 **Generations:** Up to 3 music generations per day.
-    - 📥 **Downloads:** 3 downloads per month.
-    - ⚡ **Reduced Wait Times:** The server stays active longer, minimizing wait times.
-
-    **Tier 3 - $99/month**
-    - 🎵 **Generations:** Up to 5 music generations per day.
-    - 📥 **Downloads:** 5 downloads per month.
-    - ⚡ **Minimal Wait Times:** Increased server uptime for faster access.
-
-    **Tier 4 - $499/month**
-    - 🎵 **Generations:** Up to 20 music generations per day.
-    - 📥 **Downloads:** 20 downloads per month.
-    - 🚀 **Instant Access:** Server is always active for immediate use.
+    - 🎵 **Generations:** 3 music generations
+    - 📥 **Downloads:** No downloads allowed
+    - 🎼 **Submit your track from dashboard**
+    - 💰 **Receive 50% commission**
+    - 💵 **Tracks sale for $1 on website**
+
+    **Tier 2 - $24.99/month**
+    - 🎵 **Generations:** Up to 3 music generations per day
+    - 📥 **Downloads:** 1 download per month
+    - 🔒 **Private Tracks:** Create unpublished, private tracks
+    - 📊 **Dashboard:** Access your profile and save up to 2 tracks
+
+    **Tier 3 - $99.99/month**
+    - 🎵 **Generations:** Up to 5 music generations per day
+    - 📥 **Downloads:** 5 downloads per month
+    - 📊 **Dashboard:** Full profile access for managing and saving your tracks
+    - 💾 **Storage:** Save up to 10 tracks
+
+    **Tier 4 - $269.97/month**
+    - 🎵 **Generations:** Up to 20 music generations per day
+    - 📥 **Downloads:** 15 downloads per month
+    - 🏢 **Commercial Use:** Includes rights for commercial purposes
+    - 📊 **Dashboard:** Advanced profile management for commercial accounts
 
     ### 🚀 **Why Subscribe?**
-    Subscribing helps keep the server active, reducing or eliminating wait times due to server startup.
-    Enjoy faster access and a smoother experience with a premium subscription!
+    Upgrade now to unlock more features and enhance your music creation experience!
     """
 
     start_time = time.time()
@@ -441,19 +470,17 @@ def generate_audio(genre, energy_level, tempo, description, duration):
         subscription_data = subscription_response.json()
         
         # Check for free tier duration restriction
-        is_free_tier = subscription_data.get("subscription_tier") == "free"
-        if is_free_tier and duration > 30:
-            st.warning("⚠️ Free tier users are limited to 30-second generations.")
-            st.info("💡 Upgrade to a premium plan to generate longer tracks!\n\n👉 [Upgrade Now](https://songlabai.com/subscribe/)")
-            duration = 30  # Force duration to 30 seconds for free tier
-        
-        if subscription_data["has_exceeded_generation_limit"]:
-            st.error(
-                "You have reached your generation limit. "
-                "Please upgrade your subscription for additional access.\n\n"
-                "��� [Upgrade Now](https://songlabai.com/subscribe/)"
-            )
-            return
+        is_free_tier = not subscription_plan_id or subscription_plan_id == 576
+        if is_free_tier:
+            if duration > 30:
+                st.warning("⚠️ Free tier users are limited to 30-second generations.")
+                st.info("💡 Upgrade to a premium plan to generate longer tracks!\n\n" + 
+                       "**Available Plans:**\n" +
+                       "- **$24.99/month:** Up to 3 generations per day, 1 download per month\n" +
+                       "- **$99.99/month:** Up to 5 generations per day, 5 downloads per month\n" +
+                       "- **$269.97/month:** Up to 20 generations per day, 15 downloads per month\n\n" +
+                       "👉 [Upgrade Now](https://songlabai.com/subscribe/)")
+                duration = 30  # Force duration to 30 seconds for free tier
                 
         # Check if server is cold
         if not check_server_status():