Update app.py

app.py

@@ -909,7 +909,7 @@ def show_sidebar():
                                 # Video "How It Works" - only on video tab
                                 st.markdown("---")
                                 st.markdown("#### ℹ️ How Video Credits Work")
-                                st.markdown("- 🎬 2 free preset video generations")
+                                st.markdown("- 🎬 1 free preset video generation")
                                 st.markdown("- 💳 Buy credits for custom videos & more generations")
                                 st.markdown("- 🎞️ **Each credit is tied to the FPS tier you purchased**")
                                 st.markdown("- 🔒 To use 24fps, buy credits with 24fps addon")
@@ -928,7 +928,7 @@ def show_sidebar():
                         col1, col2 = st.columns(2)
 
                         st.markdown("#### 💎 Audio Credit Packages")
-                        st.info("🎁 **Start FREE!** Get 2 free 30-second samples")
+                        st.info("🎁 **Start FREE!** Get 1 free sample on [Generate for Free](https://songlabai.com/generate-music-free/) page")
 
                         st.markdown("**Available Packages:**")
                         st.markdown("- 🎵 **Music with Lyrics**: $7.50 (up to 3 min)")
@@ -944,7 +944,7 @@ def show_sidebar():
                         # Audio "How It Works" - only on audio tabs
                         st.markdown("---")
                         st.markdown("#### ℹ️ How Audio Credits Work")
-                        st.markdown("- 🎁 **2 free 30-second samples** for every user")
+                        st.markdown("- 🎁 **1 free sample** (available on [Generate for Free](https://songlabai.com/generate-music-free/) page)")
                         st.markdown("- 💳 **Credits charged per generation**")
                         st.markdown("- ⏱️ **1 credit = any duration** up to 3 minutes")
                         st.markdown("- 🎵 **With or without lyrics** - same credit cost")
@@ -1778,15 +1778,23 @@ def display_audio_with_controls(audio_bytes, subscription_status, location="main
                 ):
                     st.success("✅ Download started!", icon="✅")
             else:
-                # Downloads available after generation
-                if st.download_button(
-                    label="⬇️ Download Audio",
-                    data=audio_bytes,
-                    file_name=f"songlab_track_{location}.wav",
-                    mime="audio/wav",
-                    key=f"download_btn_{location}"
-                ):
-                    st.success("✅ Download started!", icon="✅")
+                # Downloads available for credit purchasers only (Streamlit is paid-only)
+                remaining_credits = subscription_status.get('remaining_credits', 0)
+
+                if remaining_credits > 0 or subscription_status.get('total_credits', 0) > 0:
+                    # User has purchased credits - allow download
+                    if st.download_button(
+                        label="⬇️ Download Audio",
+                        data=audio_bytes,
+                        file_name=f"songlab_track_{location}.wav",
+                        mime="audio/wav",
+                        key=f"download_btn_{location}"
+                    ):
+                        st.success("✅ Download started!", icon="✅")
+                else:
+                    # No credits purchased - block download
+                    st.button("🔒 Download (Purchase Credits Required)", disabled=True, key=f"download_btn_{location}")
+                    st.info("💡 Download is included with credit purchases. Visit [Pricing Page](https://songlabai.com/credit-pricing/) to buy credits.")
     
     with col2:
         # User dashboard saving removed - only admin auto-save is active
@@ -4069,12 +4077,10 @@ with generation_tab3:
     # Check 3: Preset mode warning for free users
     if not is_custom_mode and video_credits:
         free_remaining = video_credits.get('free_generations_remaining', 0)
-        if free_remaining == 2:
-            st.info(f"🎁 Using free preset generation (1/2)")
-        elif free_remaining == 1:
-            st.warning(f"⚠️ Using your LAST free preset generation (2/2) - Purchase credits to continue after this!")
+        if free_remaining == 1:
+            st.info(f"🎁 Using your 1 free preset generation - Purchase credits to continue after this!")
         elif free_remaining == 0:
-            st.error(f"🔒 You've used all 2 free generations - Purchase credits to continue")
+            st.error(f"🔒 You've used your 1 free generation - Purchase credits to continue")
 
     # Generation Button - Label changes based on mode and eligibility
     if generation_mode == "🎵 Jingle + Video":