Update app.py

app.py

@@ -268,13 +268,50 @@ 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)
 
-# Duration input
-duration = st.slider(
-    "Duration (in seconds):", min_value=15, max_value=300, value=30, step=1
-)
-
-prompt2 = "Classical ,Low, 40, slow calm"
-
+# Also update the duration slider to show the restriction
+if "jwt_token" in st.session_state:
+    auth_headers = get_auth_headers()
+    subscription_url = "https://songlabai.com/wp-json/custom-api/v1/subscription"
+    try:
+        subscription_response = requests.get(subscription_url, headers=auth_headers)
+        if subscription_response.status_code == 200:
+            subscription_data = subscription_response.json()
+            is_free_tier = subscription_data.get("subscription_tier") == "free"
+            
+            if is_free_tier:
+                duration = st.slider(
+                    "Duration (in seconds):",
+                    min_value=15,
+                    max_value=30,  # Restrict 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
+                )
+    except Exception as e:
+        st.error(f"Error checking subscription status: {str(e)}")
+        duration = st.slider(
+            "Duration (in seconds):",
+            min_value=15,
+            max_value=300,
+            value=30,
+            step=1
+        )
+else:
+    duration = st.slider(
+        "Duration (in seconds):",
+        min_value=15,
+        max_value=300,
+        value=30,
+        step=1
+    )
 
 
 
@@ -438,6 +475,14 @@ def generate_audio(genre, energy_level, tempo, description, duration):
             return
                 
         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. Your duration has been automatically adjusted.")
+            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. "