Update app.py

app.py

@@ -8,17 +8,17 @@ import os
 import gradio as gr
 import replicate
 import base64
-from PIL import Image
+from PIL import Image, ImageDraw, ImageFont
 import io
 import requests
 from datetime import datetime
 import tempfile
 import time
+from moviepy.editor import VideoFileClip, TextClip, CompositeVideoClip
+import numpy as np
 
 # API token setup
 api_token = os.getenv("RAPI_TOKEN")
-if api_token:
-    os.environ["REPLICATE_API_TOKEN"] = api_token
 
 # Aspect ratio options
 ASPECT_RATIOS = {
@@ -35,6 +35,64 @@ ASPECT_RATIOS = {
 DEFAULT_TEXT_PROMPT = ""
 DEFAULT_IMAGE_PROMPT = "Generate a video with smooth and natural movement. Objects should have visible motion while maintaining fluid transitions."
 
+def add_watermark(input_video_path, output_video_path):
+    """Add watermark to video using moviepy"""
+    try:
+        # Load the video
+        video = VideoFileClip(input_video_path)
+        
+        # Get video dimensions
+        width, height = video.size
+        
+        # Calculate watermark size and position
+        watermark_text = "ginigen.com"
+        font_size = max(12, int(height * 0.025))  # 2.5% of video height, minimum 12px
+        
+        # Create text clip for watermark
+        watermark = TextClip(
+            watermark_text,
+            fontsize=font_size,
+            color='white',
+            font='Arial',
+            stroke_color='black',
+            stroke_width=1,
+            method='caption',
+            size=(None, None)
+        )
+        
+        # Position watermark at bottom right with padding
+        padding = int(width * 0.02)  # 2% padding
+        watermark = watermark.set_position((width - watermark.w - padding, height - watermark.h - padding))
+        watermark = watermark.set_duration(video.duration)
+        
+        # Set opacity (semi-transparent)
+        watermark = watermark.set_opacity(0.7)
+        
+        # Composite the watermark onto the video
+        final_video = CompositeVideoClip([video, watermark])
+        
+        # Write the output video
+        final_video.write_videofile(
+            output_video_path,
+            codec='libx264',
+            audio_codec='aac',
+            temp_audiofile=tempfile.mktemp('.m4a'),
+            remove_temp=True,
+            logger=None  # Suppress moviepy output
+        )
+        
+        # Clean up
+        video.close()
+        final_video.close()
+        
+        return True
+    except Exception as e:
+        print(f"Watermark error: {str(e)}")
+        # If watermarking fails, just copy the original
+        import shutil
+        shutil.copy2(input_video_path, output_video_path)
+        return False
+
 def update_prompt_placeholder(mode):
     """Update prompt placeholder based on mode"""
     if mode == "Text to Video":
@@ -203,11 +261,31 @@ def generate_video(mode, prompt, image, aspect_ratio, seed, api_key_input, progr
         # Save to temporary file
         with tempfile.NamedTemporaryFile(delete=False, suffix='.mp4') as tmp_file:
             tmp_file.write(video_data)
-            video_path = tmp_file.name
+            temp_video_path = tmp_file.name
+        
+        progress(0.8, desc="Adding watermark...")
+        
+        # Create final output path
+        final_video_path = tempfile.mktemp(suffix='.mp4')
+        
+        # Add watermark
+        watermark_added = add_watermark(temp_video_path, final_video_path)
         
         # Also save as output.mp4
-        with open("output.mp4", "wb") as file:
-            file.write(video_data)
+        if os.path.exists(final_video_path):
+            with open(final_video_path, "rb") as f:
+                watermarked_data = f.read()
+            with open("output.mp4", "wb") as file:
+                file.write(watermarked_data)
+        else:
+            # Fallback if watermarking failed
+            with open("output.mp4", "wb") as file:
+                file.write(video_data)
+            final_video_path = temp_video_path
+        
+        # Clean up temp file if different
+        if temp_video_path != final_video_path and os.path.exists(temp_video_path):
+            os.unlink(temp_video_path)
         
         progress(1.0, desc="Complete!")
         
@@ -220,9 +298,10 @@ def generate_video(mode, prompt, image, aspect_ratio, seed, api_key_input, progr
 - Seed: {seed}
 - Duration: 5 seconds
 - Resolution: 480p
+- Watermark: {"Added" if watermark_added else "Failed (original saved)"}
 - File: output.mp4"""
         
-        return video_path, info
+        return final_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."
@@ -301,6 +380,7 @@ with gr.Blocks(title="Bytedance Seedance Video Free", theme=gr.themes.Soft()) as
             ### 📋 Fixed Settings
             - **Duration**: 5 seconds
             - **Resolution**: 480p
+            - **Watermark**: ginigen.com
             """)
         
         with gr.Column(scale=2):
@@ -334,7 +414,7 @@ with gr.Blocks(title="Bytedance Seedance Video Free", theme=gr.themes.Soft()) as
         
         1. **Install required packages**:
         ```bash
-        pip install gradio replicate pillow requests
+        pip install gradio replicate pillow requests moviepy
         ```
         
         2. **Set environment variable** (optional):
@@ -353,6 +433,7 @@ with gr.Blocks(title="Bytedance Seedance Video Free", theme=gr.themes.Soft()) as
         - **Image to Video**: Transform uploaded image into animated video
         - **Aspect Ratios**: Choose ratios optimized for various social media platforms
         - **Seed Value**: Use same seed to reproduce identical results
+        - **Watermark**: Automatically adds "ginigen.com" watermark to generated videos
         
         ### Prompt Writing Tips
         
@@ -366,6 +447,7 @@ with gr.Blocks(title="Bytedance Seedance Video Free", theme=gr.themes.Soft()) as
         - **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 issues**: If watermark fails, the original video will be saved without watermark.
         """)
     
     # Examples