Update app.py

app.py

@@ -25,8 +25,8 @@ except ImportError:
     print("Warning: cv2 not available. Watermark feature will be disabled.")
 
 # API keys setup from environment variables
-API_KEY_T2V = os.getenv("WS_API_KEY_T2V", "946b77acd6a456dfde349aa0bac7b6d2bdf9c0c995fff072898c6d8734f866c4")
-API_KEY_I2V = os.getenv("WS_API_KEY_I2V", "5a90eaa7ded95066c07adf55b91185a83abfa8db3fdc74d12ba5ad66db1d68fe")
+API_KEY_T2V = os.getenv("WAVESPEED_API_KEY_T2V", "946b77acd6a456dfde349aa0bac7b6d2bdf9c0c995fff072898c6d8734f866c4")
+API_KEY_I2V = os.getenv("WAVESPEED_API_KEY_I2V", "5a90eaa7ded95066c07adf55b91185a83abfa8db3fdc74d12ba5ad66db1d68fe")
 
 # API endpoints
 API_BASE_URL = "https://api.wavespeed.ai/api/v3"
@@ -111,6 +111,43 @@ def add_watermark(input_video_path, output_video_path):
     except:
         return False
 
+def compress_image(image, max_size_mb=2, max_dimension=1920):
+    """Compress and resize image to meet size requirements"""
+    # Convert to RGB if necessary
+    if image.mode in ('RGBA', 'LA'):
+        rgb_image = Image.new('RGB', image.size, (255, 255, 255))
+        rgb_image.paste(image, mask=image.split()[-1] if image.mode == 'RGBA' else None)
+        image = rgb_image
+    elif image.mode not in ('RGB', 'L'):
+        image = image.convert('RGB')
+    
+    # Resize if dimensions are too large
+    width, height = image.size
+    if width > max_dimension or height > max_dimension:
+        ratio = min(max_dimension / width, max_dimension / height)
+        new_width = int(width * ratio)
+        new_height = int(height * ratio)
+        image = image.resize((new_width, new_height), Image.Resampling.LANCZOS)
+    
+    # Compress with varying quality to meet size requirement
+    quality = 95
+    while quality > 20:
+        buffered = io.BytesIO()
+        image.save(buffered, format="JPEG", quality=quality, optimize=True)
+        size_mb = len(buffered.getvalue()) / (1024 * 1024)
+        
+        if size_mb <= max_size_mb:
+            return buffered.getvalue()
+        
+        quality -= 5
+    
+    # If still too large, resize more aggressively
+    width, height = image.size
+    image = image.resize((width // 2, height // 2), Image.Resampling.LANCZOS)
+    buffered = io.BytesIO()
+    image.save(buffered, format="JPEG", quality=50, optimize=True)
+    return buffered.getvalue()
+
 def update_prompt_placeholder(mode):
     """Update prompt placeholder based on mode"""
     if mode == "Text to Video":
@@ -203,24 +240,42 @@ def generate_video(mode, prompt, image_url, image_file, aspect_ratio, seed, api_
                 # 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")
-                else:
-                    image_file.save(buffered, format="PNG")
-                
-                image_base64 = base64.b64encode(buffered.getvalue()).decode()
-                
-                # 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
+                # Compress image to reduce size
+                try:
+                    # Get original size
+                    orig_buffered = io.BytesIO()
+                    image_file.save(orig_buffered, format="PNG")
+                    orig_size_mb = len(orig_buffered.getvalue()) / (1024 * 1024)
+                    
+                    progress(0.1, desc=f"Compressing image (Original: {orig_size_mb:.1f}MB)...")
+                    
+                    compressed_data = compress_image(image_file, max_size_mb=2, max_dimension=1920)
+                    compressed_size_mb = len(compressed_data) / (1024 * 1024)
+                    
+                    progress(0.15, desc=f"Image compressed: {orig_size_mb:.1f}MB → {compressed_size_mb:.1f}MB")
+                    
+                    image_base64 = base64.b64encode(compressed_data).decode()
+                    
+                    # Check encoded size
+                    encoded_size_mb = len(image_base64) / (1024 * 1024)
+                    if encoded_size_mb > 20:  # Conservative limit for base64
+                        return None, f"""❌ Image too large even after compression.
+
+Original size: {orig_size_mb:.1f}MB
+Compressed size: {compressed_size_mb:.1f}MB
+Encoded size: {encoded_size_mb:.1f}MB (exceeds API limit)
+
+Please use an image hosting service:
+1. Go to [imgur.com](https://imgur.com)
+2. Upload your image (drag & drop)
+3. Right-click uploaded image → "Copy image address"
+4. Paste URL in the Image URL field above"""
+                    
+                    # Try data URL format
+                    final_image_url = f"data:image/jpeg;base64,{image_base64}"
+                    
+                except Exception as e:
+                    return None, f"❌ Error processing image: {str(e)}"
             else:
                 return None, "❌ No image provided."
             
@@ -248,15 +303,31 @@ def generate_video(mode, prompt, image_url, image_file, aspect_ratio, seed, api_
         else:
             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}
+                # Check if it's a size limit error
+                if "size exceeds" in error_detail or "30MB" in error_detail:
+                    return None, f"""❌ Image size exceeds API limit (30MB).
 
-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
+Your image was compressed but still too large for the API.
+
+**Please use an image hosting service instead:**
+1. Go to [imgur.com](https://imgur.com) (no account needed)
+2. Click "New post" → Drop your image
+3. After upload, right-click the image → "Copy image address"
+4. Paste the URL in the Image URL field above
+
+Alternative services:
+- [imgbb.com](https://imgbb.com) - Simple, no account needed
+- [postimages.org](https://postimages.org) - Direct links available
+
+Error: {response.status_code}"""
+                else:
+                    return None, f"""❌ API Error: {response.status_code}
+
+The API does not accept base64 data URLs. Please use a direct image URL instead:
+
+1. Upload your image to [imgur.com](https://imgur.com)
+2. Copy the direct image URL (ends with .jpg or .png)
+3. Paste it in the Image URL field above
 
 Error details: {error_detail}"""
             else:
@@ -312,17 +383,25 @@ Error details: {error_detail}"""
         
         # Generation info
         watermark_status = "Added" if watermark_added else "Not available (cv2 not installed)" if not VIDEO_PROCESSING_AVAILABLE else "Failed"
+        
+        # Add image info if applicable
+        image_info = ""
+        if mode == "Image to Video":
+            if image_url:
+                image_info = "\n- Image Source: URL provided"
+            elif image_file:
+                image_info = "\n- Image Source: Uploaded & compressed"
+        
         info = f"""✅ Video generated successfully!
 
 📊 Generation Info:
 - Mode: {mode}
-- Aspect Ratio: {aspect_ratio}
+- Aspect Ratio: {aspect_ratio if mode == "Text to Video" else "N/A"}{image_info}
 - Seed: {seed}
 - Duration: 5 seconds
 - Resolution: 480p
 - Watermark: {watermark_status}
 - API: WaveSpeed AI
-- Cost: $0.06
 - File: output.mp4"""
         
         return final_video_path, info
@@ -339,9 +418,8 @@ with gr.Blocks(title="Bytedance Seedance Video Free", theme=gr.themes.Soft()) as
     
     Generate videos from text or images using **WaveSpeed AI API**.
     
-    [![Powered by Ginigen](https://img.shields.io/badge/Powered%20by-WaveSpeed%20AI-blue)](https://ginigen.com/)
+    [![Powered by Ginigen](https://img.shields.io/badge/Powered%20by%20Ginigen)](https://ginigen.com/)
     
-    ⚠️ **Note**: Each generation costs $0.06
     """)
     
     with gr.Row():
@@ -381,15 +459,18 @@ with gr.Blocks(title="Bytedance Seedance Video Free", theme=gr.themes.Soft()) as
             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"
+                    label="📷 Image URL (Recommended) - Direct image URL works best",
+                    placeholder="https://example.com/image.jpg"
                 )
                 gr.Markdown("**OR**")
                 image_file_input = gr.Image(
-                    label="📤 Upload Image (Beta) - Converted to data URL (may not work)",
+                    label="📤 Upload Image (Will be compressed, max 2MB after compression)",
                     type="pil"
                 )
+                gr.Markdown("""
+                ⚠️ **Note**: API has a 30MB limit. Large images will be compressed.
+                If upload fails, use [imgur.com](https://imgur.com) to host your image.
+                """)
             
             # Aspect ratio (only for T2V)
             aspect_ratio = gr.Dropdown(
@@ -417,7 +498,6 @@ with gr.Blocks(title="Bytedance Seedance Video Free", theme=gr.themes.Soft()) as
             ### 📋 Fixed Settings
             - **Duration**: 5 seconds
             - **Resolution**: 480p
-            - **Cost**: $0.06 per video
             - **Watermark**: {watermark_info}
             """)
         
@@ -431,7 +511,7 @@ with gr.Blocks(title="Bytedance Seedance Video Free", theme=gr.themes.Soft()) as
             )
             
             # Generate button
-            generate_btn = gr.Button("🎬 Generate Video ($0.06)", variant="primary", size="lg")
+            generate_btn = gr.Button("🎬 Generate Video", variant="primary", size="lg")
             
             # Results display
             with gr.Column():
@@ -483,19 +563,24 @@ with gr.Blocks(title="Bytedance Seedance Video Free", theme=gr.themes.Soft()) as
         
         ### 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
+        **Recommended: Use Image URLs**
+        1. Upload your image to [imgur.com](https://imgur.com) (easiest, no account needed)
+        2. After upload, right-click image → "Copy image address"
+        3. Paste URL in the Image URL field
+        
+        **Image Upload Limitations:**
+        - Uploaded images are compressed to max 2MB
+        - Large images (>30MB) may still fail after compression
+        - If upload fails, use an image hosting service instead
+        
+        **Free Image Hosting Services:**
+        - [imgur.com](https://imgur.com) - No account needed, easiest
+        - [imgbb.com](https://imgbb.com) - Simple interface
+        - [postimages.org](https://postimages.org) - Direct links
         
         ### API Information
         
-        - **Provider**: WaveSpeed AI
-        - **Models**: ByteDance Seedance v1 Lite
-        - **Cost**: $0.06 per 5-second video
+        - **Models**: ByteDance Seedance v1
         - **Resolution**: 480p
         
         ### Prompt Writing Tips
@@ -512,7 +597,7 @@ with gr.Blocks(title="Bytedance Seedance Video Free", theme=gr.themes.Soft()) as
             ["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],
+            ["Image to Video", "In a wide shot, the camera slowly zooms in while maintaining focus. Natural movement brings the scene to life.", "https://d2g64w682n9w0w.cloudfront.net/media/images/1750659000869310688_vGUb740X.jpg", None, "16:9", -1],
         ],
         inputs=[mode, prompt, image_url_input, image_file_input, aspect_ratio, seed],
         label="Example Prompts"