Update app.py

app.py

@@ -13,6 +13,7 @@ import io
 import requests
 from datetime import datetime
 import tempfile
+import time
 
 # API token setup
 api_token = os.getenv("RAPI_TOKEN")
@@ -30,12 +31,22 @@ ASPECT_RATIOS = {
     "9:21": "9:21 (Ultra Vertical)"
 }
 
+# Default prompts
+DEFAULT_TEXT_PROMPT = ""
+DEFAULT_IMAGE_PROMPT = "Generate a video with smooth and natural movement. Objects should have visible motion while maintaining fluid transitions."
+
 def update_prompt_placeholder(mode):
     """Update prompt placeholder based on mode"""
     if mode == "Text to Video":
-        return gr.update(placeholder="Describe the video you want to create.\nExample: The sun rises slowly between tall buildings. [Ground-level follow shot] Bicycle tires roll over a dew-covered street at dawn.")
+        return gr.update(
+            placeholder="Describe the video you want to create.\nExample: The sun rises slowly between tall buildings. [Ground-level follow shot] Bicycle tires roll over a dew-covered street at dawn.",
+            value=""
+        )
     else:
-        return gr.update(placeholder="Describe how the image should move.\nExample: Camera slowly zooms in while clouds move across the sky. The subject's hair gently moves in the wind.")
+        return gr.update(
+            placeholder="Describe how the image should move.\nExample: Camera slowly zooms in while clouds move across the sky. The subject's hair gently moves in the wind.",
+            value=DEFAULT_IMAGE_PROMPT
+        )
 
 def update_image_input(mode):
     """Show/hide image input based on mode"""
@@ -44,6 +55,22 @@ def update_image_input(mode):
     else:
         return gr.update(visible=False)
 
+def wait_for_model_with_retry(replicate_client, 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_client.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"""
     
@@ -90,22 +117,87 @@ 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.3, desc="Calling Replicate API...")
+        progress(0.2, desc="Checking model availability...")
         
-        # Run Replicate
-        output = replicate.run(
-            "bytedance/seedance-1-lite",
-            input=input_params
+        # Create Replicate client with extended timeout
+        client = replicate.Client(
+            api_token=token,
+            timeout=300  # 5 minutes timeout
         )
         
+        # Wait for model to be ready
+        model_ready = wait_for_model_with_retry(client, "bytedance/seedance-1-lite")
+        if not model_ready:
+            return None, "⏳ Model is still booting up. Please try again in a few minutes."
+        
+        progress(0.3, desc="Calling Replicate API...")
+        
+        # Run Replicate with retry logic
+        max_attempts = 3
+        for attempt in range(max_attempts):
+            try:
+                # Use prediction API for better control
+                prediction = client.predictions.create(
+                    version="bytedance/seedance-1-lite:latest",
+                    input=input_params,
+                    webhook_completed=None
+                )
+                
+                # Poll for completion with extended timeout
+                start_time = time.time()
+                timeout_seconds = 300  # 5 minutes
+                
+                while prediction.status not in ["succeeded", "failed", "canceled"]:
+                    if time.time() - start_time > timeout_seconds:
+                        return None, "⏱️ Generation timed out. The server might be under heavy load. Please try again."
+                    
+                    time.sleep(2)  # Poll every 2 seconds
+                    prediction.reload()
+                    
+                    # Update progress
+                    elapsed = time.time() - start_time
+                    progress_val = min(0.3 + (elapsed / timeout_seconds) * 0.4, 0.7)
+                    progress(progress_val, desc=f"Generating video... ({int(elapsed)}s)")
+                
+                if prediction.status == "failed":
+                    error_msg = getattr(prediction, 'error', 'Unknown error')
+                    if "cold boot" in str(error_msg).lower() or "starting" in str(error_msg).lower():
+                        if attempt < max_attempts - 1:
+                            progress(0.3, desc=f"Model is starting up, retrying... (Attempt {attempt + 2}/{max_attempts})")
+                            time.sleep(30)  # Wait 30 seconds before retry
+                            continue
+                    return None, f"❌ Generation failed: {error_msg}"
+                
+                elif prediction.status == "canceled":
+                    return None, "❌ Generation was canceled."
+                
+                # Success - get output
+                output = prediction.output
+                break
+                
+            except replicate.exceptions.ReplicateError as e:
+                if "timeout" in str(e).lower() and attempt < max_attempts - 1:
+                    progress(0.3, desc=f"Timeout occurred, retrying... (Attempt {attempt + 2}/{max_attempts})")
+                    time.sleep(10)
+                    continue
+                else:
+                    return None, f"❌ Replicate API error: {str(e)}"
+            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)}"
+        
         progress(0.7, desc="Downloading video...")
         
         # Get video data
         if hasattr(output, 'read'):
             video_data = output.read()
         else:
-            # Download from URL
-            response = requests.get(output)
+            # Download from URL with timeout
+            response = requests.get(output, timeout=60)
             video_data = response.content
         
         # Save to temporary file
@@ -132,8 +224,12 @@ def generate_video(mode, prompt, image, aspect_ratio, seed, api_key_input, progr
         
         return video_path, info
         
+    except requests.exceptions.Timeout:
+        return None, "⏱️ Request timed out. The server might be under heavy load. Please try again in a few minutes."
     except Exception as e:
         error_msg = f"❌ Error occurred: {str(e)}"
+        if "timeout" in str(e).lower():
+            error_msg += "\n\n💡 Tip: The model might be cold starting. Please wait a minute and try again."
         return None, error_msg
 
 # Gradio interface
@@ -212,7 +308,8 @@ with gr.Blocks(title="Bytedance Seedance Video Free", theme=gr.themes.Soft()) as
             prompt = gr.Textbox(
                 label="✍️ Prompt",
                 lines=5,
-                placeholder="Describe the video you want to create.\nExample: The sun rises slowly between tall buildings. [Ground-level follow shot] Bicycle tires roll over a dew-covered street at dawn."
+                placeholder="Describe the video you want to create.\nExample: The sun rises slowly between tall buildings. [Ground-level follow shot] Bicycle tires roll over a dew-covered street at dawn.",
+                value=""
             )
             
             # Generate button
@@ -263,6 +360,12 @@ with gr.Blocks(title="Bytedance Seedance Video Free", theme=gr.themes.Soft()) as
         - Specify camera movements (e.g., zoom in, pan left, tracking shot)
         - Describe lighting and atmosphere (e.g., golden hour, dramatic lighting)
         - Indicate movement speed (e.g., slowly, rapidly, gently)
+        
+        ### Troubleshooting
+        
+        - **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.
         """)
     
     # Examples