Update app.py

app.py

@@ -148,6 +148,60 @@ def update_image_input(mode):
     else:
         return gr.update(visible=False)
 
+def warmup_model(api_key_input, progress=gr.Progress()):
+    """Warm up the model with a simple request"""
+    token = api_key_input or api_token
+    if not token:
+        return gr.update(visible=True, value="❌ API token required. Please set it in API Settings.")
+    
+    replicate.api_token = token
+    
+    try:
+        progress(0, desc="Starting model warm-up...")
+        
+        # Very simple warm-up request with minimal parameters
+        warm_up_params = {
+            "prompt": "red",  # Extremely simple prompt
+            "duration": 1,    # Minimum duration
+            "resolution": "480p",
+            "aspect_ratio": "16:9",
+            "seed": 42
+        }
+        
+        progress(0.5, desc="Sending warm-up request (this may fail, that's normal)...")
+        
+        try:
+            output = replicate.run("bytedance/seedance-1-lite", input=warm_up_params)
+            if output:
+                return gr.update(visible=True, value="""✅ **Warm-up successful!** 
+
+The model is now awake. You can generate videos normally.
+Note: The model stays active for ~5 minutes, so generate quickly!""")
+        except Exception as e:
+            if "PA" in str(e) or "interrupted" in str(e).lower():
+                return gr.update(visible=True, value="""⏰ **Warm-up request sent!**
+
+The model is waking up. This is normal for a cold start.
+
+**Next steps:**
+1. Wait 2-3 minutes for the model to fully initialize
+2. Try a simple prompt: "a cat walking"
+3. Once that works, use your actual prompt
+
+The wake-up process takes time, but subsequent generations will be fast!""")
+            else:
+                return gr.update(visible=True, value=f"""⚠️ **Warm-up attempted**
+
+Error: {str(e)[:200]}...
+
+**What to do:**
+1. Wait 2-3 minutes
+2. Try generating with a simple prompt
+3. The model might already be warming up!""")
+            
+    except Exception as e:
+        return gr.update(visible=True, value=f"❌ Warm-up error: {str(e)[:200]}... Try generating anyway after waiting a minute.")
+
 def generate_video(mode, prompt, image, aspect_ratio, seed, api_key_input, progress=gr.Progress()):
     """Main video generation function"""
     
@@ -171,7 +225,7 @@ def generate_video(mode, prompt, image, aspect_ratio, seed, api_key_input, progr
         # Input parameters setup
         input_params = {
             "prompt": prompt,
-            "duration": 5,
+            "duration": 5,  # Fixed at 5 seconds
             "resolution": "480p",
             "aspect_ratio": aspect_ratio,
             "seed": seed
@@ -204,15 +258,13 @@ def generate_video(mode, prompt, image, aspect_ratio, seed, api_key_input, progr
         
         progress(0.3, desc="Calling Replicate API...")
         
-        # Run Replicate with retry logic
-        max_attempts = 3
+        # Run Replicate with retry logic and cold start handling
+        max_attempts = 3  # Reduced attempts, focus on user guidance
         output = None
         
         for attempt in range(max_attempts):
             try:
-                # Run Replicate - use the model directly without version specifier
-                start_time = time.time()
-                
+                # Run the actual request
                 progress(0.3 + attempt * 0.1, desc=f"Generating video... (Attempt {attempt + 1}/{max_attempts})")
                 
                 # Run the model
@@ -227,38 +279,78 @@ def generate_video(mode, prompt, image, aspect_ratio, seed, api_key_input, progr
                     
             except Exception as e:
                 error_str = str(e)
-                # Handle specific error patterns
-                if "Prediction interrupted" in error_str or "code: PA" in error_str:
-                    if attempt < max_attempts - 1:
-                        wait_time = 45 + (attempt * 20)  # Longer wait for interrupted predictions
-                        progress(0.3, desc=f"Model is cold starting. This is normal for first use. Waiting {wait_time}s...")
+                print(f"Attempt {attempt + 1} error: {error_str}")  # Debug logging
+                
+                # Check for cold start indicators
+                if "Prediction interrupted" in error_str or "code: PA" in error_str or "PA" in error_str:
+                    if attempt == 0:
+                        # First cold start - provide immediate guidance
+                        return None, """❌ **Model is cold starting (code: PA)**
+
+**Quick Fix - Follow these steps:**
+
+1️⃣ Click the **"🔥 Warm Up Model"** button above
+2️⃣ Wait for the warm-up to complete (it may also fail, that's OK)
+3️⃣ **Wait 2-3 minutes** for the model to fully initialize
+4️⃣ Try this simple prompt first: **"a cat walking"**
+5️⃣ Once that works, try your original prompt
+
+**Why this happens:** The model goes to sleep after inactivity to save resources. The first request wakes it up, which takes time.
+
+**Alternative:** Try again in 5 minutes when the model should be fully awake."""
+                    elif attempt < max_attempts - 1:
+                        # Subsequent attempts - wait and retry
+                        wait_time = 30 + (attempt * 15)
+                        progress(0.3, desc=f"Model still warming up. Waiting {wait_time}s... (Attempt {attempt + 2}/{max_attempts})")
                         time.sleep(wait_time)
                         continue
                     else:
-                        return None, "❌ Model is still warming up. Please wait 2-3 minutes and try again with a simple prompt like 'a cat walking'."
+                        return None, """❌ Model is taking longer than usual to start. 
+
+**Please try one of these options:**
+1. Use the "🔥 Warm Up Model" button first
+2. Wait 5 minutes and try again
+3. Try a very simple prompt: "a red circle"
+4. Try during off-peak hours (evening/night US time)
+
+The model needs to 'warm up' after being idle."""
+                        
                 elif "cold boot" in error_str.lower() or "starting" in error_str.lower():
                     if attempt < max_attempts - 1:
-                        progress(0.3, desc=f"Model is starting up, waiting 30s before retry...")
-                        time.sleep(30)
+                        progress(0.3, desc=f"Model is booting up, waiting 60s...")
+                        time.sleep(60)
                         continue
+                        
                 elif "timeout" in error_str.lower():
                     if attempt < max_attempts - 1:
-                        progress(0.3, desc=f"Timeout occurred, retrying... (Attempt {attempt + 2}/{max_attempts})")
-                        time.sleep(10)
+                        progress(0.3, desc=f"Request timed out, retrying... (Attempt {attempt + 2}/{max_attempts})")
+                        time.sleep(15)
                         continue
-                elif "ReplicateError" in error_str:
-                    return None, f"❌ Replicate API error: {error_str}"
+                        
                 else:
+                    # For any other error, still retry a few times
                     if attempt < max_attempts - 1:
-                        progress(0.3, desc=f"Error occurred, retrying... (Attempt {attempt + 2}/{max_attempts})")
-                        time.sleep(5)
+                        wait_time = 10 + (attempt * 5)
+                        progress(0.3, desc=f"Error occurred, retrying in {wait_time}s... (Attempt {attempt + 2}/{max_attempts})")
+                        time.sleep(wait_time)
                         continue
                     else:
-                        return None, f"❌ Unexpected error: {error_str}"
+                        return None, f"❌ Error after {max_attempts} attempts: {error_str}"
         
         # Check if we got output
         if not output:
-            return None, "❌ Failed to generate video. The model might be cold starting. Please wait 2-3 minutes and try again with a simple prompt."
+            return None, """❌ Failed to generate video after multiple attempts.
+
+**This usually happens when:**
+- The model is cold starting (hasn't been used recently)
+- The server is under heavy load
+
+**Please try:**
+1. Wait 3-5 minutes for the model to warm up
+2. Start with a simple prompt: "a cat walking" or "a ball bouncing"
+3. Once that works, try your original prompt
+
+The first generation often takes longer, but subsequent ones will be faster."""
         
         progress(0.7, desc="Downloading video...")
         
@@ -333,10 +425,27 @@ with gr.Blocks(title="Bytedance Seedance Video Free", theme=gr.themes.Soft()) as
     Generate videos from text or images using **Replicate API**.
     
     [![Powered by Ginigen](https://img.shields.io/badge/Powered%20by-Replicate-blue)](https://ginigen.com/)
-    
-    ⚠️ **Note**: First generation may take longer (2-3 minutes) as the model warms up.
     """)
     
+    # Cold start warning box
+    with gr.Accordion("⚠️ Getting 'Prediction interrupted (PA)' error? Click here!", open=False):
+        gr.Markdown("""
+        ### 🔥 Model Cold Start Guide
+        
+        If you're getting **"Prediction interrupted (code: PA)"** errors, the model is sleeping and needs to wake up.
+        
+        **Quick Solution:**
+        1. Click the **"🔥 Warm Up Model"** button
+        2. Wait 2-3 minutes after warm-up
+        3. Try a simple prompt: **"a cat walking"**
+        4. Once that works, use your actual prompt
+        
+        **Why this happens:** Free tier models sleep after ~5 minutes of inactivity to save resources.
+        The first request wakes them up, which can take 2-5 minutes.
+        
+        **Pro tip:** Once warmed up, the model stays active for ~5 minutes. Generate multiple videos quickly!
+        """)
+    
     with gr.Row():
         with gr.Column(scale=1):
             # API Settings
@@ -409,11 +518,16 @@ with gr.Blocks(title="Bytedance Seedance Video Free", theme=gr.themes.Soft()) as
                 value=""
             )
             
-            # Generate button
-            generate_btn = gr.Button("🎬 Generate Video", variant="primary", size="lg")
+            # Generate and warm-up buttons
+            with gr.Row():
+                generate_btn = gr.Button("🎬 Generate Video", variant="primary", size="lg", scale=2)
+                warmup_btn = gr.Button("🔥 Warm Up Model First", variant="secondary", size="lg", scale=1)
+            
+            gr.Markdown("*Use Warm Up if you're getting PA errors or it's your first generation today*")
             
             # Results display
             with gr.Column():
+                warmup_status = gr.Textbox(label="Status", visible=False)
                 output_video = gr.Video(
                     label="📹 Generated Video",
                     autoplay=True
@@ -466,24 +580,38 @@ with gr.Blocks(title="Bytedance Seedance Video Free", theme=gr.themes.Soft()) as
         
         ### Troubleshooting
         
-        - **"Prediction interrupted (code: PA)"**: The model is warming up. Wait 2-3 minutes and try again, or try a simpler prompt first.
-        - **Repeated failures**: Try with a shorter, simpler prompt to warm up the model.
-        - **Timeout errors**: The model might be cold starting. Wait 1-2 minutes and try again.
-        - **Model booting**: First requests after inactivity may take longer as the model boots up.
-        - **Extended wait times**: Complex prompts or server load may cause longer generation times.
-        - **Watermark not showing**: Install opencv-python for watermark support.
+        #### If you get "Prediction interrupted (code: PA)":
+        
+        This means the model is cold starting. Follow these steps:
+        
+        1. **First Try**: Click the "🔥 Warm Up Model" button and wait for completion
+        2. **Wait**: After warm-up, wait 2-3 minutes for the model to fully initialize
+        3. **Simple Test**: Try generating with a very simple prompt: "a cat walking"
+        4. **Your Prompt**: Once the simple test works, try your actual prompt
+        
+        **Other Issues:**
+        - **Repeated failures**: The model may be under heavy load. Try during off-peak hours
+        - **Timeout errors**: Increase wait time between attempts
+        - **Watermark not showing**: Install opencv-python for watermark support
+        
+        **Pro Tips:**
+        - First generation of the day always takes longer
+        - Keep prompts under 100 words for best results
+        - Simple prompts work better for warming up the model
+        - Once warmed up, generate multiple videos quickly before the model sleeps again!
         """)
     
     # Examples
     gr.Examples(
         examples=[
-            ["Text to Video", "A cat walking", None, "16:9", 42],  # Simple warm-up prompt
+            ["Text to Video", "a red ball bouncing", None, "16:9", 42],  # Very simple warm-up
+            ["Text to Video", "a cat walking", None, "16:9", 42],  # Simple warm-up prompt
             ["Text to Video", "A serene lake at sunrise with mist rolling over the water. Camera slowly pans across the landscape as birds fly overhead.", None, "16:9", 42],
             ["Text to Video", "Urban street scene at night with neon lights reflecting on wet pavement. People walking with umbrellas, camera tracking forward.", None, "9:16", 123],
             ["Text to Video", "Close-up of a flower blooming in time-lapse, soft natural lighting, shallow depth of field.", None, "1:1", 789],
         ],
         inputs=[mode, prompt, image_input, aspect_ratio, seed],
-        label="Example Prompts (Start with the simple 'cat walking' for cold start)"
+        label="Example Prompts (⚡ Start with 'red ball' or 'cat walking' if model is cold)"
     )
     
     # Event handlers
@@ -505,6 +633,12 @@ with gr.Blocks(title="Bytedance Seedance Video Free", theme=gr.themes.Soft()) as
         outputs=[ratio_info]
     )
     
+    warmup_btn.click(
+        fn=warmup_model,
+        inputs=[api_key_input],
+        outputs=[warmup_status]
+    )
+    
     generate_btn.click(
         fn=generate_video,
         inputs=[mode, prompt, image_input, aspect_ratio, seed, api_key_input],