Update app.py

app.py

@@ -328,6 +328,7 @@ def generate_audio(genre, energy_level, tempo, description, duration):
     progress_placeholder = st.empty()
     subscription_info_placeholder = st.empty()
     server_message_placeholder = st.empty()
+    stage_message_placeholder = st.empty()
 
     # Define the subscription tiers information
     subscription_info = """
@@ -351,17 +352,68 @@ def generate_audio(genre, energy_level, tempo, description, duration):
 
     ### 🚀 **Why Subscribe?**
     Premium subscribers enjoy priority access to dedicated servers, 
-    significantly reducing wait times and server initialization delays.
+    significantly reducing wait times from 7 minutes to just 30 seconds!
     """
 
-    # Start the server initialization phase
-    for seconds in range(30):  # 30-second initial countdown
-        remaining_seconds = 30 - seconds
-        status_placeholder.markdown("### 🔄 Initializing AI Music Server...")
-        progress_placeholder.progress(seconds / 30)
-        server_message_placeholder.info(f"⏳ Server warmup: {remaining_seconds}s remaining. Premium users get priority access!")
+    # Start server warmup with an initial request
+    warmup_payload = {"inputs": {"prompt": "warmup", "duration": 1}}
+    api_headers = get_api_headers()
+    
+    # Create a thread for the initial warmup request
+    def make_warmup_request():
+        try:
+            return time_post_request(API_URL, headers=api_headers, payload=warmup_payload, timeout=600)
+        except:
+            return None
+
+    warmup_thread = threading.Thread(target=make_warmup_request)
+    warmup_thread.start()
+
+    # Start the countdown while server is warming up
+    start_time = time.time()
+    total_warmup_time = 420  # 7 minutes in seconds
+    
+    # Define server initialization stages
+    stages = [
+        {"time": 0, "message": "🚀 Initializing server resources..."},
+        {"time": 60, "message": "⚙️ Loading AI models into memory..."},
+        {"time": 180, "message": "🔧 Optimizing model parameters..."},
+        {"time": 300, "message": "📊 Configuring generation settings..."},
+        {"time": 360, "message": "✨ Finalizing server preparation..."}
+    ]
+    current_stage = 0
+
+    while time.time() - start_time < total_warmup_time and warmup_thread.is_alive():
+        elapsed = time.time() - start_time
+        remaining = total_warmup_time - elapsed
+        progress = elapsed / total_warmup_time
+        
+        # Update current stage based on elapsed time
+        while current_stage < len(stages) - 1 and elapsed > stages[current_stage + 1]["time"]:
+            current_stage += 1
+
+        minutes_remaining = int(remaining // 60)
+        seconds_remaining = int(remaining % 60)
+        
+        status_placeholder.markdown("### 🔄 AI Server Initialization in Progress")
+        progress_placeholder.progress(progress)
+        
+        stage_message_placeholder.info(stages[current_stage]["message"])
+        
+        server_message_placeholder.warning(
+            f"⏳ Estimated time remaining: {minutes_remaining}m {seconds_remaining}s\n\n"
+            f"🎵 Free tier server initialization takes about 7 minutes\n\n"
+            "💡 Premium users enjoy dramatically faster startup times (30s or less)!"
+        )
+        
         subscription_info_placeholder.markdown(subscription_info)
-        time.sleep(1)
+        time.sleep(0.1)
+
+    # Clear initialization messages
+    status_placeholder.empty()
+    progress_placeholder.empty()
+    stage_message_placeholder.empty()
+    server_message_placeholder.empty()
 
     # Check if the user is logged in
     if st.session_state.get("jwt_token") is None:
@@ -401,19 +453,10 @@ def generate_audio(genre, energy_level, tempo, description, duration):
         )
         return
 
-    # Clear initial countdown placeholders
-    status_placeholder.empty()
-    progress_placeholder.empty()
-    subscription_info_placeholder.empty()
-    server_message_placeholder.empty()
-
     # Prepare the prompt and payload for the music generation API
     prompt = f"Genre: {genre}, Energy Level: {energy_level}, Tempo: {tempo}, Description: {description}"
     payload = {"inputs": {"prompt": prompt, "duration": duration}}
 
-    # Headers for the music generation API
-    api_headers = get_api_headers()
-
     # Enhanced retry logic with better feedback
     max_retries = 3
     retry_delay = 30  # increased initial delay
@@ -426,7 +469,7 @@ def generate_audio(genre, energy_level, tempo, description, duration):
 
                 if response is None:
                     server_message_placeholder.warning(
-                        "🔄 Server is warming up. This may take a few minutes on first request. "
+                        "🔄 Finalizing server preparation. "
                         f"Retrying in {current_delay} seconds... \n\n"
                         "💡 Premium subscribers enjoy faster access with dedicated servers!"
                     )