Update app.py

app.py

@@ -563,35 +563,34 @@ def get_current_subscription_status():
                 print(f"DEBUG - free_samples_remaining field: {credits_data.get('free_samples_remaining', 'FIELD_MISSING')}")
                 print(f"DEBUG - total_credits field: {credits_data.get('total_credits', 'FIELD_MISSING')}")
 
-                # NEW AUDIO CREDIT SYSTEM
+                # NEW AUDIO CREDIT SYSTEM - Streamlit app is PAID ONLY (no free samples)
                 # IMPORTANT: API returns values as strings, convert to int
                 remaining_credits = int(credits_data.get('remaining_credits', 0))
-                free_samples_remaining = int(credits_data.get('free_samples_remaining', 0))
                 total_credits = int(credits_data.get('total_credits', 0))
-                
-                print(f"DEBUG - Parsed values -> remaining: {remaining_credits}, free_samples: {free_samples_remaining}, total: {total_credits}")
+
+                print(f"DEBUG - Parsed values -> remaining: {remaining_credits}, total: {total_credits}")
 
                 result = {
                     # Legacy compatibility fields
-                    'subscription_plan_id': 'credits' if (total_credits > 0 or free_samples_remaining > 0) else 'free',
-                    'is_free_tier': total_credits == 0 and free_samples_remaining == 0,
+                    'subscription_plan_id': 'credits' if total_credits > 0 else 'free',
+                    'is_free_tier': total_credits == 0,
                     'has_exceeded_download_limit': False,
                     'has_exceeded_generation_limit': False,
 
-                    # NEW AUDIO CREDIT SYSTEM FIELDS
+                    # NEW AUDIO CREDIT SYSTEM FIELDS (Streamlit = paid only)
                     'is_credit_based': True,
                     'unlimited_generation': True,
                     'remaining_credits': remaining_credits,
-                    'free_samples_remaining': free_samples_remaining,
+                    'free_samples_remaining': 0,  # Streamlit app has no free samples
                     'total_credits': total_credits,
                     'used_credits': int(credits_data.get('used_credits', 0)),
-                    'free_samples_used': int(credits_data.get('free_samples_used', 0))
+                    'free_samples_used': 0  # Streamlit app has no free samples
                 }
 
                 # Cache the result
                 st.session_state[cache_key] = result
                 st.session_state[cache_time_key] = current_time
-                print(f"Cached subscription status - credits: {remaining_credits}, free samples: {free_samples_remaining}")
+                print(f"Cached subscription status - credits: {remaining_credits} (Streamlit = paid only, no free samples)")
                 print(f"DEBUG - Final result object: {result}")
                 return result
             else:
@@ -820,27 +819,24 @@ def show_sidebar():
                     st.markdown("- 📥 **Downloads:** All formats available")
                     
                 else:
-                    # Get credit balances from the new system
+                    # Get credit balances from the new system (Streamlit = paid only)
                     remaining_credits = subscription_status.get('remaining_credits', 0)
-                    free_samples = subscription_status.get('free_samples_remaining', 0)
 
                     # Display what user can do across all tabs
                     st.markdown("#### 🎵 What You Can Do")
-                    st.markdown("- 🎁 **2 Free 30s Samples:** Try before you buy!")
                     st.markdown("- 🎵 **Audio Generation:** Credits required")
-                    st.markdown("- 🎬 **Videos:** 2 Free Presets")
-                    st.markdown("- 📥 **Downloads:** Available after generation")
+                    st.markdown("- 🎬 **Video Generation:** Credits required")
+                    st.markdown("- 📥 **Downloads:** Included with credit purchases")
+                    st.markdown("- 💡 **Free Samples:** Available in Gradio app only")
 
                     # Display audio credit balances
                     st.markdown("---")
                     st.markdown("#### 💰 Audio Generation Credits")
-                    if remaining_credits > 0 or free_samples > 0:
+                    if remaining_credits > 0:
                         st.markdown(f"- 🎵 **Audio Credits:** {remaining_credits} (any duration)")
-                        st.markdown(f"- 🆓 **Free Samples:** {free_samples} (30s max)")
                     else:
                         st.markdown("- 🎵 **Audio Credits:** 0")
-                        st.markdown("- 🆓 **Free Samples:** 0")
-                        st.info("💡 Purchase credits below to download your tracks!")
+                        st.info("💡 Purchase credits below to generate and download your tracks!")
 
                     # Show video credits ONLY when on Commercial Ad tab
                     if st.session_state.get('current_tab') == 'commercial_ad':
@@ -1825,7 +1821,6 @@ def download_button_no_rerun(audio_bytes, subscription_status, location="main"):
     
     # Get credit balances
     remaining_credits = subscription_status.get('remaining_credits', 0)
-    free_samples = subscription_status.get('free_samples_remaining', 0)
 
     # Get audio metadata
     metadata = st.session_state.get('audio_metadata', {})
@@ -1836,11 +1831,10 @@ def download_button_no_rerun(audio_bytes, subscription_status, location="main"):
 
     # 10-second download
     st.markdown("**🥉 10-Second Download**")
-    can_use_10s_free_sample = (10 <= 30 and free_samples > 0)
     can_use_10s_credit = remaining_credits > 0
 
-    if can_use_10s_free_sample or can_use_10s_credit:
-        resource_label = "Free Sample" if can_use_10s_free_sample else "1 Credit"
+    if can_use_10s_credit:
+        resource_label = "1 Credit"
         if st.button(f"⬇️ Download (10s) - {resource_label}", key=f"download_10s_{location}", type="secondary"):
             # Use credit/sample and download
             success = use_download_credit(jwt_token, user_id, 10, has_lyrics, {'location': location, 'duration': '10s'})
@@ -1866,24 +1860,23 @@ def download_button_no_rerun(audio_bytes, subscription_status, location="main"):
                         key=f"download_10s_final_{location}"
                     )
 
-                    st.success(f"✅ {resource_label} used! Credits: {remaining_credits}, Samples: {free_samples}", icon="✅")
+                    st.success(f"✅ {resource_label} used! Credits: {remaining_credits}", icon="✅")
                 except Exception as e:
                     st.error(f"❌ Download failed: {str(e)}")
             else:
                 st.error("❌ Failed to process download. Please try again.")
     else:
-        st.button("🔒 No Credits/Samples", disabled=True, key=f"no_credit_10s_{location}")
+        st.button("🔒 No Credits", disabled=True, key=f"no_credit_10s_{location}")
         st.caption("Purchase credits to download")
 
     st.markdown("---")
 
     # 20-second download
     st.markdown("**🥇 20-Second Download**")
-    can_use_20s_free_sample = (20 <= 30 and free_samples > 0)
     can_use_20s_credit = remaining_credits > 0
 
-    if can_use_20s_free_sample or can_use_20s_credit:
-        resource_label = "Free Sample" if can_use_20s_free_sample else "1 Credit"
+    if can_use_20s_credit:
+        resource_label = "1 Credit"
         if st.button(f"⬇️ Download (20s) - {resource_label}", key=f"download_20s_{location}", type="primary"):
             # Use credit/sample and download
             success = use_download_credit(jwt_token, user_id, 20, has_lyrics, {'location': location, 'duration': '20s'})
@@ -1909,17 +1902,17 @@ def download_button_no_rerun(audio_bytes, subscription_status, location="main"):
                         key=f"download_20s_final_{location}"
                     )
                     
-                    st.success(f"✅ {resource_label} used! Credits: {remaining_credits}, Samples: {free_samples}", icon="✅")
+                    st.success(f"✅ {resource_label} used! Credits: {remaining_credits}", icon="✅")
                 except Exception as e:
                     st.error(f"❌ Download failed: {str(e)}")
             else:
                 st.error("❌ Failed to process download. Please try again.")
     else:
-        st.button("🔒 No Credits/Samples", disabled=True, key=f"no_credit_20s_{location}")
+        st.button("🔒 No Credits", disabled=True, key=f"no_credit_20s_{location}")
         st.caption("Purchase credits to download")
-    
-    # Show purchase links if no credits or samples
-    if remaining_credits == 0 and free_samples == 0:
+
+    # Show purchase links if no credits
+    if remaining_credits == 0:
         st.markdown("---")
         st.info("💡 **Need Credits?** Purchase audio credits to download your tracks!", icon="ℹ️")
         st.markdown("[🛒 Buy Audio Credits](https://songlabai.com/songlab-melody/)")