Update app.py

app.py

@@ -148,22 +148,6 @@ def update_image_input(mode):
     else:
         return gr.update(visible=False)
 
-def wait_for_model_with_retry(model_name, max_retries=5, initial_wait=10):
-    """Wait for model to be ready with retry logic"""
-    for attempt in range(max_retries):
-        try:
-            # Try to get model info
-            model = replicate.models.get(model_name)
-            return True
-        except Exception as e:
-            if attempt < max_retries - 1:
-                wait_time = initial_wait * (attempt + 1)
-                print(f"Model not ready, waiting {wait_time} seconds... (Attempt {attempt + 1}/{max_retries})")
-                time.sleep(wait_time)
-            else:
-                return False
-    return False
-
 def generate_video(mode, prompt, image, aspect_ratio, seed, api_key_input, progress=gr.Progress()):
     """Main video generation function"""
     
@@ -210,15 +194,13 @@ def generate_video(mode, prompt, image, aspect_ratio, seed, api_key_input, progr
             
             input_params["image"] = f"data:image/png;base64,{image_base64}"
         
-        progress(0.2, desc="Checking model availability...")
+        progress(0.2, desc="Preparing API request...")
         
         # Set up Replicate with the API token
         replicate.api_token = token
         
-        # Wait for model to be ready
-        model_ready = wait_for_model_with_retry("bytedance/seedance-1-lite")
-        if not model_ready:
-            return None, "⏳ Model is still booting up. Please try again in a few minutes."
+        # Skip model availability check to avoid delays
+        # The actual run will handle cold start retries
         
         progress(0.3, desc="Calling Replicate API...")
         
@@ -231,51 +213,52 @@ def generate_video(mode, prompt, image, aspect_ratio, seed, api_key_input, progr
                 # Run Replicate - use the model directly without version specifier
                 start_time = time.time()
                 
-                # Create a generator for progress tracking
-                output_generator = replicate.run(
+                progress(0.3 + attempt * 0.1, desc=f"Generating video... (Attempt {attempt + 1}/{max_attempts})")
+                
+                # Run the model
+                output = replicate.run(
                     "bytedance/seedance-1-lite",
                     input=input_params
                 )
                 
-                # If it's a generator, iterate through it
-                if hasattr(output_generator, '__iter__') and not isinstance(output_generator, (str, bytes)):
-                    for event in output_generator:
-                        elapsed = time.time() - start_time
-                        progress_val = min(0.3 + (elapsed / 300) * 0.4, 0.7)
-                        progress(progress_val, desc=f"Generating video... ({int(elapsed)}s)")
-                        output = event  # Keep the last event as output
-                else:
-                    # If it's not a generator, use it directly
-                    output = output_generator
-                
                 # If we got output, break the retry loop
                 if output:
                     break
-                
-            except replicate.exceptions.ReplicateError as e:
+                    
+            except Exception as e:
                 error_str = str(e)
-                if "cold boot" in error_str.lower() or "starting" in error_str.lower():
+                # 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...")
+                        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'."
+                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, retrying... (Attempt {attempt + 2}/{max_attempts})")
+                        progress(0.3, desc=f"Model is starting up, waiting 30s before retry...")
                         time.sleep(30)
                         continue
-                elif "timeout" in error_str.lower() and attempt < max_attempts - 1:
-                    progress(0.3, desc=f"Timeout occurred, retrying... (Attempt {attempt + 2}/{max_attempts})")
-                    time.sleep(10)
-                    continue
-                else:
+                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)
+                        continue
+                elif "ReplicateError" in error_str:
                     return None, f"❌ Replicate API error: {error_str}"
-            except Exception as e:
-                if attempt < max_attempts - 1:
-                    progress(0.3, desc=f"Error occurred, retrying... (Attempt {attempt + 2}/{max_attempts})")
-                    time.sleep(5)
-                    continue
                 else:
-                    return None, f"❌ Unexpected error: {str(e)}"
+                    if attempt < max_attempts - 1:
+                        progress(0.3, desc=f"Error occurred, retrying... (Attempt {attempt + 2}/{max_attempts})")
+                        time.sleep(5)
+                        continue
+                    else:
+                        return None, f"❌ Unexpected error: {error_str}"
         
         # Check if we got output
         if not output:
-            return None, "❌ Failed to generate video after multiple attempts."
+            return None, "❌ Failed to generate video. The model might be cold starting. Please wait 2-3 minutes and try again with a simple prompt."
         
         progress(0.7, desc="Downloading video...")
         
@@ -350,6 +333,8 @@ 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.
     """)
     
     with gr.Row():
@@ -481,6 +466,8 @@ 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.
@@ -490,12 +477,13 @@ with gr.Blocks(title="Bytedance Seedance Video Free", theme=gr.themes.Soft()) as
     # Examples
     gr.Examples(
         examples=[
+            ["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"
+        label="Example Prompts (Start with the simple 'cat walking' for cold start)"
     )
     
     # Event handlers