Update app.py

app.py

@@ -329,100 +329,99 @@ def time_post_request(api_url, headers=None, payload=None):
 
     return response
 
-    def generate_audio(genre, energy_level, tempo, description, duration):
-        # Start the countdown
-        count_down = 10
-        count_down_placeholder = st.empty()
-        for seconds in range(count_down):
-            count_down_placeholder.info(
-                f"Preparing the server... Please wait ⏳ {count_down - seconds} seconds."
-            )
-            time.sleep(1)
-        count_down_placeholder.empty()
+def generate_audio(genre, energy_level, tempo, description, duration):
+    # Start the countdown
+    count_down = 10
+    count_down_placeholder = st.empty()
+    for seconds in range(count_down):
+        count_down_placeholder.info(
+            f"Preparing the server... Please wait ⏳ {count_down - seconds} seconds."
+        )
+        time.sleep(1)
+    count_down_placeholder.empty()
 
-        # Check if the user is logged in
-        if st.session_state.get("jwt_token") is None:
-            st.error("You must be logged in to generate audio.")
-            return
+    # Check if the user is logged in
+    if st.session_state.get("jwt_token") is None:
+        st.error("You must be logged in to generate audio.")
+        return
 
-        jwt_token = st.session_state["jwt_token"]
-
-        # Headers for authenticated requests
-        auth_headers = {
-            "Authorization": f"Bearer {jwt_token}"
-        }
-
-        # Get user ID
-        user_id_url = "https://songlabai.com/wp-json/custom-api/v1/current-user-id"
-        user_response = requests.get(user_id_url, headers=auth_headers)
-        if user_response.status_code == 200:
-            user_data = user_response.json()
-            user_id = user_data["user_id"]
-            st.write("User Data:", user_data)
-            st.write("user_id:", user_id)
-
-            # Get subscription status
-            subscription_url = "https://songlabai.com/wp-json/custom-api/subscription"
-            subscription_response = requests.get(subscription_url, headers=auth_headers)
-            if subscription_response.status_code == 200:
-                subscription_data = subscription_response.json()
-                subscription_plan_id = subscription_data["subscription_plan_id"]
-                st.write("subscription_plan_id:", subscription_plan_id)
-                has_exceeded_generation_limit = subscription_data["has_exceeded_generation_limit"]
-                st.write("has_exceeded_generation_limit:", has_exceeded_generation_limit)
-
-                # Check subscription plan and limits
-                if subscription_plan_id in [None, ""]:  # Free user
-                    if has_exceeded_generation_limit:
-                        st.error("You have reached your generation limit.")
-                        return
-                    else:
-                        # Update user's generation count
-                        update_generation_url = "https://songlabai.com/wp-json/custom-api/update-generation-count"
-                        update_response = requests.post(update_generation_url, headers=auth_headers)
-                        if update_response.status_code != 200:
-                            st.error("Failed to update your generation count.")
-                            return
-                        # Proceed to generate audio
-                elif subscription_plan_id in ["304", "305", "306"]:
-                    if has_exceeded_generation_limit:
-                        st.error("You have reached your daily generation limit.")
-                        return
-                    else:
-                        # Update user's generation count
-                        update_generation_url = "https://songlabai.com/wp-json/custom-api/update-generation-count"
-                        update_response = requests.post(update_generation_url, headers=auth_headers)
-                        if update_response.status_code != 200:
-                            st.error("Failed to update your generation count.")
-                            return
-                        # Proceed to generate audio
+    jwt_token = st.session_state["jwt_token"]
+
+    # Headers for authenticated requests
+    auth_headers = {
+        "Authorization": f"Bearer {jwt_token}"
+    }
+
+    # Get user ID
+    user_id_url = "https://songlabai.com/wp-json/custom-api/v1/current-user-id"
+    user_response = requests.get(user_id_url, headers=auth_headers)
+    if user_response.status_code == 200:
+        user_data = user_response.json()
+        user_id = user_data["user_id"]
+        st.write("User Data:", user_data)
+        st.write("user_id:", user_id)
+
+        # Get subscription status
+        subscription_url = "https://songlabai.com/wp-json/custom-api/subscription"
+        subscription_response = requests.get(subscription_url, headers=auth_headers)
+        if subscription_response.status_code == 200:
+            subscription_data = subscription_response.json()
+            subscription_plan_id = subscription_data["subscription_plan_id"]
+            st.write("subscription_plan_id:", subscription_plan_id)
+            has_exceeded_generation_limit = subscription_data["has_exceeded_generation_limit"]
+            st.write("has_exceeded_generation_limit:", has_exceeded_generation_limit)
+
+            # Check subscription plan and limits
+            if subscription_plan_id in [None, ""]:  # Free user
+                if has_exceeded_generation_limit:
+                    st.error("You have reached your generation limit.")
+                    return
                 else:
-                    st.error("Your subscription does not allow you to generate music.")
+                    # Update user's generation count
+                    update_generation_url = "https://songlabai.com/wp-json/custom-api/update-generation-count"
+                    update_response = requests.post(update_generation_url, headers=auth_headers)
+                    if update_response.status_code != 200:
+                        st.error("Failed to update your generation count.")
+                        return
+                    # Proceed to generate audio
+            elif subscription_plan_id in ["304", "305", "306"]:
+                if has_exceeded_generation_limit:
+                    st.error("You have reached your daily generation limit.")
                     return
+                else:
+                    # Update user's generation count
+                    update_generation_url = "https://songlabai.com/wp-json/custom-api/update-generation-count"
+                    update_response = requests.post(update_generation_url, headers=auth_headers)
+                    if update_response.status_code != 200:
+                        st.error("Failed to update your generation count.")
+                        return
+                    # Proceed to generate audio
             else:
-                st.error("Failed to retrieve your subscription status.")
+                st.error("Your subscription does not allow you to generate music.")
                 return
         else:
-            st.error("Failed to retrieve your user information.")
+            st.error("Failed to retrieve your subscription status.")
             return
+    else:
+        st.error("Failed to retrieve your user information.")
+        return
 
-        # Proceed to generate audio after checks
-        prompt = f"Genre: {genre}, Energy Level: {energy_level}, Tempo: {tempo}, Description: {description},"
-        payload = {"inputs": {"prompt": prompt, "duration": duration}}
-
-        # Use the original headers for the API request
-        with st.spinner("Generating audio ..."):
-            response = time_post_request(API_URL, headers, payload)
+    # Proceed to generate audio after checks
+    prompt = f"Genre: {genre}, Energy Level: {energy_level}, Tempo: {tempo}, Description: {description},"
+    payload = {"inputs": {"prompt": prompt, "duration": duration}}
 
-        placeholder1 = st.empty()
-        if response.status_code != 200:
-            st.error("Failed to generate audio.")
-            return
-        else:
-            placeholder1.success(f"✔️ Audio Generated, Loading...")
-            load_and_play_generated_audio(response)
-            placeholder1.empty()
+    # Use the original headers for the API request
+    with st.spinner("Generating audio ..."):
+        response = time_post_request(API_URL, headers, payload)
 
+    placeholder1 = st.empty()
+    if response.status_code != 200:
+        st.error("Failed to generate audio.")
+        return
+    else:
+        placeholder1.success(f"✔️ Audio Generated, Loading...")
+        load_and_play_generated_audio(response)
+        placeholder1.empty()
 
 
 
@@ -467,11 +466,11 @@ def load_and_play_generated_audio(response):
         st.write(f"To download use the following product code: {product_code}")
 
 
-    if st.button("Generate Audio", key="generate_audio_button"):
-        if genre and energy_level and description and tempo:
-            generate_audio(genre, energy_level, tempo, description, duration)
-        else:
-            st.info("Description field is required.")
+if st.button("Generate Audio", key="generate_audio_button"):
+    if genre and energy_level and description and tempo:
+         generate_audio(genre, energy_level, tempo, description, duration)
+    else:
+        st.info("Description field is required.")
 
 
 # Post-processing options