Update app.py

app.py

@@ -284,19 +284,27 @@ if st.session_state.get("jwt_token"):
 else:
     is_free_tier = True  # Default to free tier if not logged in
 
-# Add this debug section before the duration slider
+# 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"
-    response = requests.get(subscription_url, headers=auth_headers)
+    subscription_response = requests.get(subscription_url, headers=auth_headers)
     
-    if response.status_code == 200:
-        data = response.json()
+    if subscription_response.status_code == 200:
+        subscription_data = subscription_response.json()
+        # Convert subscription_plan_id to string for comparison
+        subscription_plan_id = str(subscription_data.get("subscription_plan_id"))
+        is_free_tier = subscription_plan_id == "576" or subscription_data.get("status") == "no_subscription"
+        
+        # Debug info
         st.write("Debug Info:")
-        st.json(data)  # This will show the full API response
-        st.write(f"Is Free Tier: {not data.get('subscription_plan_id') or data.get('subscription_plan_id') == 576}")
+        st.json(subscription_data)
+        st.write(f"Subscription Plan ID: {subscription_plan_id}")
+        st.write(f"Is Free Tier: {is_free_tier}")
     else:
-        st.write(f"API Error: {response.status_code}")
+        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:
@@ -484,7 +492,7 @@ def generate_audio(genre, energy_level, tempo, description, duration):
         subscription_data = subscription_response.json()
         
         # Check for free tier duration restriction
-        is_free_tier = not subscription_plan_id or subscription_plan_id == 576
+        is_free_tier = subscription_plan_id == "576" or subscription_data.get("status") == "no_subscription"
         if is_free_tier:
             if duration > 30:
                 st.warning("⚠️ Free tier users are limited to 30-second generations.")