Update app.py

app.py

@@ -268,50 +268,14 @@ 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)
 
-# 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
-    )
+# Always show the full duration slider initially
+duration = st.slider(
+    "Duration (in seconds):",
+    min_value=15,
+    max_value=300,
+    value=30,
+    step=1
+)
 
 
 
@@ -479,7 +443,7 @@ def generate_audio(genre, energy_level, tempo, description, duration):
         # 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.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