Update app.py

app.py

@@ -127,9 +127,11 @@ def update_prompt_placeholder(mode):
 def update_image_input(mode):
     """Show/hide image input based on mode"""
     if mode == "Image to Video":
-        return gr.update(visible=True)
+        # Show image input, hide aspect ratio for I2V
+        return gr.update(visible=True), gr.update(visible=False)
     else:
-        return gr.update(visible=False)
+        # Hide image input, show aspect ratio for T2V
+        return gr.update(visible=False), gr.update(visible=True)
 
 def poll_for_result(request_id, api_key, progress_callback=None):
     """Poll WaveSpeed API for video generation result"""
@@ -167,15 +169,16 @@ def poll_for_result(request_id, api_key, progress_callback=None):
         except Exception as e:
             return False, f"Connection error: {str(e)}"
 
-def generate_video(mode, prompt, image, aspect_ratio, seed, api_key_override_t2v, api_key_override_i2v, progress=gr.Progress()):
+def generate_video(mode, prompt, image_url, image_file, aspect_ratio, seed, api_key_override_t2v, api_key_override_i2v, progress=gr.Progress()):
     """Main video generation function using WaveSpeed API"""
     
     # Input validation
     if not prompt:
         return None, "❌ Please enter a prompt."
     
-    if mode == "Image to Video" and image is None:
-        return None, "❌ Please upload an image."
+    if mode == "Image to Video":
+        if not image_url and image_file is None:
+            return None, "❌ Please provide an image URL or upload an image."
     
     try:
         progress(0, desc="Preparing request...")
@@ -195,30 +198,35 @@ def generate_video(mode, prompt, image, aspect_ratio, seed, api_key_override_t2v
             api_key = api_key_override_i2v or API_KEY_I2V
             endpoint = I2V_ENDPOINT
             
-            # Handle image upload
-            if isinstance(image, str) and (image.startswith('http://') or image.startswith('https://')):
-                # If it's already a URL
-                image_url = image
-            else:
-                # Convert PIL Image to base64 for upload
-                progress(0.1, desc="Processing image...")
-                if isinstance(image, str):
-                    with Image.open(image) as img:
-                        buffered = io.BytesIO()
+            # Handle image input
+            if image_url:
+                # Use provided URL directly
+                final_image_url = image_url
+            elif image_file is not None:
+                # Convert PIL image to data URL
+                progress(0.1, desc="Converting image to data URL...")
+                
+                # Convert to base64
+                buffered = io.BytesIO()
+                if isinstance(image_file, str):
+                    with Image.open(image_file) as img:
                         img.save(buffered, format="PNG")
-                        image_base64 = base64.b64encode(buffered.getvalue()).decode()
                 else:
-                    buffered = io.BytesIO()
-                    image.save(buffered, format="PNG")
-                    image_base64 = base64.b64encode(buffered.getvalue()).decode()
+                    image_file.save(buffered, format="PNG")
+                
+                image_base64 = base64.b64encode(buffered.getvalue()).decode()
                 
-                # For this demo, we'll assume the image needs to be uploaded somewhere
-                # In production, you'd upload to a CDN and get a URL
-                return None, "❌ Image upload to CDN not implemented. Please use an image URL."
+                # Try data URL format first
+                final_image_url = f"data:image/png;base64,{image_base64}"
+                
+                # Note: If the API doesn't accept data URLs, you'll need to upload to a CDN
+                # For now, we'll try the data URL and provide guidance if it fails
+            else:
+                return None, "❌ No image provided."
             
             payload = {
                 "duration": 5,
-                "image": image_url,
+                "image": final_image_url,
                 "prompt": prompt,
                 "seed": seed if seed >= 0 else -1
             }
@@ -238,7 +246,21 @@ def generate_video(mode, prompt, image, aspect_ratio, seed, api_key_override_t2v
             request_id = result["id"]
             progress(0.3, desc=f"Request submitted. ID: {request_id}")
         else:
-            return None, f"❌ API Error: {response.status_code} - {response.text}"
+            error_detail = response.text
+            if mode == "Image to Video" and image_file is not None and not image_url:
+                return None, f"""❌ API Error: {response.status_code}
+
+The API may not accept base64 data URLs. Please try one of these options:
+1. Use a direct image URL (e.g., https://example.com/image.jpg)
+2. Upload your image to an image hosting service like:
+   - imgur.com
+   - imgbb.com
+   - postimages.org
+3. Use the image URL in the Image URL field
+
+Error details: {error_detail}"""
+            else:
+                return None, f"❌ API Error: {response.status_code} - {error_detail}"
         
         # Poll for result
         success, result = poll_for_result(request_id, api_key, lambda p, desc: progress(p, desc=desc))
@@ -355,19 +377,28 @@ with gr.Blocks(title="Bytedance Seedance Video Free", theme=gr.themes.Soft()) as
                 value="Text to Video"
             )
             
-            # Image upload
-            image_input = gr.Image(
-                label="📷 Upload Image or Provide URL",
-                type="pil",
-                visible=False
-            )
+            # Image input - URL or file
+            with gr.Column(visible=False) as image_input_group:
+                gr.Markdown("### Image Input")
+                image_url_input = gr.Textbox(
+                    label="📷 Image URL",
+                    placeholder="https://example.com/image.jpg",
+                    info="Enter a direct image URL"
+                )
+                gr.Markdown("**OR**")
+                image_file_input = gr.Image(
+                    label="📤 Upload Image (Beta)",
+                    type="pil",
+                    info="Note: Uploaded images will be converted to data URL (may not work with all APIs)"
+                )
             
             # Aspect ratio (only for T2V)
             aspect_ratio = gr.Dropdown(
-                label="📐 Aspect Ratio",
+                label="📐 Aspect Ratio (Text to Video only)",
                 choices=list(ASPECT_RATIOS.keys()),
                 value="16:9",
-                info="Choose ratio optimized for social media platforms"
+                info="Choose ratio optimized for social media platforms",
+                visible=True
             )
             
             # Ratio description
@@ -443,12 +474,24 @@ with gr.Blocks(title="Bytedance Seedance Video Free", theme=gr.themes.Soft()) as
         
         ### Features
         
-        - **Text to Video**: Generate video from text description only
-        - **Image to Video**: Transform uploaded image into animated video (requires image URL)
-        - **Aspect Ratios**: Choose ratios optimized for various social media platforms
+        - **Text to Video**: Generate video from text description with aspect ratio selection
+        - **Image to Video**: Transform image into animated video
+          - Option 1: Provide direct image URL (recommended)
+          - Option 2: Upload image file (converted to data URL - may not work)
+        - **Aspect Ratios**: Available for Text to Video only
         - **Seed Value**: Use -1 for random or set specific value for reproducible results
         - **Watermark**: Automatically adds "ginigen.com" watermark (requires opencv-python)
         
+        ### Image to Video Tips
+        
+        For best results with Image to Video:
+        1. Use direct image URLs (https://...)
+        2. If uploading files doesn't work, upload your image to:
+           - [imgur.com](https://imgur.com)
+           - [imgbb.com](https://imgbb.com)
+           - [postimages.org](https://postimages.org)
+        3. Copy the direct image URL and paste it in the Image URL field
+        
         ### API Information
         
         - **Provider**: WaveSpeed AI
@@ -467,12 +510,12 @@ with gr.Blocks(title="Bytedance Seedance Video Free", theme=gr.themes.Soft()) as
     # Examples
     gr.Examples(
         examples=[
-            ["Text to Video", "A cat walking gracefully across a sunlit room", None, "16:9", -1],
-            ["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],
+            ["Text to Video", "A cat walking gracefully across a sunlit room", "", None, "16:9", -1],
+            ["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],
+            ["Image to Video", "The camera slowly zooms in while clouds drift across the sky", "https://images.unsplash.com/photo-1506905925346-21bda4d32df4", None, "16:9", -1],
         ],
-        inputs=[mode, prompt, image_input, aspect_ratio, seed],
+        inputs=[mode, prompt, image_url_input, image_file_input, aspect_ratio, seed],
         label="Example Prompts"
     )
     
@@ -486,7 +529,7 @@ with gr.Blocks(title="Bytedance Seedance Video Free", theme=gr.themes.Soft()) as
     mode.change(
         fn=update_image_input,
         inputs=[mode],
-        outputs=[image_input]
+        outputs=[image_input_group, aspect_ratio]
     )
     
     aspect_ratio.change(
@@ -497,7 +540,7 @@ with gr.Blocks(title="Bytedance Seedance Video Free", theme=gr.themes.Soft()) as
     
     generate_btn.click(
         fn=generate_video,
-        inputs=[mode, prompt, image_input, aspect_ratio, seed, api_key_override_t2v, api_key_override_i2v],
+        inputs=[mode, prompt, image_url_input, image_file_input, aspect_ratio, seed, api_key_override_t2v, api_key_override_i2v],
         outputs=[output_video, output_info]
     )