Update app.py

app.py

@@ -6,42 +6,53 @@ import logging
 from PIL import Image
 import mimetypes
 
-# Configure logging
+# Cấu hình logging
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
-# Vidu API configuration
-VIDU_API_KEY = os.getenv("VIDU_API_KEY")
-VIDU_API_URL = "https://api.vidu.com"
-POLL_INTERVAL = 5  # Seconds between status checks
-TIMEOUT = 300  # Max seconds to wait for video generation
+# Cấu hình API TDNM
+VIDU_API_KEY = os.getenv("VIDU_API_KEY")  # Giữ tên biến để tương thích với API
+TDNM_KEY = os.getenv("TDNM_KEY")  # Khóa bí mật để xác thực người dùng
+VIDU_API_URL = "https://api.vidu.com"  # URL API thực tế
+POLL_INTERVAL = 5  # Giây giữa các lần kiểm tra trạng thái
+TIMEOUT = 300  # Thời gian chờ tối đa để tạo video
 
-# Function to validate image requirements
+# Hàm kiểm tra TDNM_KEY
+def validate_tdn_key(user_key):
+    if not TDNM_KEY:
+        return False, "Lỗi: Khóa bí mật chưa được cấu hình trên máy chủ."
+    if not user_key:
+        return False, "Lỗi: Vui lòng nhập khóa bí mật."
+    if user_key != TDNM_KEY:
+        return False, "Lỗi: Khóa bí mật không hợp lệ."
+    return True, "Khóa bí mật được xác thực thành công."
+
+# Hàm kiểm tra yêu cầu hình ảnh
 def validate_image(image_path):
     if not image_path:
-        return False, "No image provided."
+        return False, "Lỗi: Chưa cung cấp hình ảnh."
     
-    # Check file size (<10MB)
+    # Kiểm tra kích thước tệp (<10MB)
     if os.path.getsize(image_path) > 10 * 1024 * 1024:
-        return False, "Image size exceeds 10MB."
+        return False, "Lỗi: Kích thước hình ảnh vượt quá 10MB."
     
-    # Check format
+    # Kiểm tra định dạng
     mime_type, _ = mimetypes.guess_type(image_path)
     if mime_type not in ["image/png", "image/webp", "image/jpeg", "image/jpg"]:
-        return False, "Unsupported image format. Use PNG, WebP, JPEG, or JPG."
+        return False, "Lỗi: Định dạng hình ảnh không được hỗ trợ. Sử dụng PNG, WebP, JPEG hoặc JPG."
     
-    # Check aspect ratio
+    # Kiểm tra tỷ lệ khung hình
     try:
         img = Image.open(image_path)
         width, height = img.size
         aspect_ratio = width / height
         if aspect_ratio < 0.25 or aspect_ratio > 4:
-            return False, "Image aspect ratio must be between 1:4 and 4:1."
+            return False, "Lỗi: Tỷ lệ khung hình phải nằm trong khoảng 1:4 đến 4:1."
         return True, None
     except Exception as e:
-        return False, f"Error validating image: {str(e)}"
+        return False, f"Lỗi khi kiểm tra hình ảnh: {str(e)}"
 
-# Function to validate pixel density ratio for two images
+# Hàm kiểm tra tỷ lệ mật độ điểm ảnh giữa hai hình ảnh
 def validate_pixel_density(start_image, end_image):
     try:
         start_img = Image.open(start_image)
@@ -50,22 +61,22 @@ def validate_pixel_density(start_image, end_image):
         end_pixels = end_img.size[0] * end_img.size[1]
         ratio = start_pixels / end_pixels
         if not (0.8 <= ratio <= 1.25):
-            return False, "Pixel density ratio between start and end images must be between 0.8 and 1.25."
+            return False, "Lỗi: Tỷ lệ mật độ điểm ảnh giữa hai hình ảnh phải nằm trong khoảng 0.8 đến 1.25."
         return True, None
     except Exception as e:
-        return False, f"Error validating pixel density: {str(e)}"
+        return False, f"Lỗi khi kiểm tra mật độ điểm ảnh: {str(e)}"
 
-# Function to upload image to Vidu
+# Hàm tải hình ảnh lên TDNM
 def upload_image_to_vidu(image_path):
     if not VIDU_API_KEY:
-        return None, "Error: Vidu API key not configured."
+        return None, "Lỗi: Khóa API TDNM chưa được cấu hình."
     
-    # Validate image
+    # Kiểm tra hình ảnh
     valid, error_message = validate_image(image_path)
     if not valid:
         return None, error_message
     
-    # Step 1: Create upload link
+    # Bước 1: Tạo liên kết tải lên
     url = f"{VIDU_API_URL}/tools/v2/files/uploads"
     headers = {
         "Authorization": f"Token {VIDU_API_KEY}",
@@ -74,68 +85,73 @@ def upload_image_to_vidu(image_path):
     payload = {"scene": "vidu"}
     
     try:
-        logger.info("Creating upload link")
+        logger.info("Tạo liên kết tải lên")
         response = requests.post(url, json=payload, headers=headers)
         response.raise_for_status()
         result = response.json()
         put_url = result.get("put_url")
         resource_id = result.get("id")
         if not put_url or not resource_id:
-            return None, "Error: No put_url or resource_id returned."
+            return None, "Lỗi: Không nhận được put_url hoặc resource_id."
     except requests.exceptions.RequestException as e:
-        logger.error(f"Create upload link error: {str(e)}")
-        return None, f"Error creating upload link: {str(e)}"
+        logger.error(f"Lỗi tạo liên kết tải lên: {str(e)}")
+        return None, f"Lỗi khi tạo liên kết tải lên: {str(e)}"
     
-    # Step 2: Upload image using put_url
+    # Bước 2: Tải hình ảnh lên put_url
     mime_type, _ = mimetypes.guess_type(image_path)
     with open(image_path, "rb") as f:
         image_data = f.read()
     
     try:
-        logger.info("Uploading image to put_url")
+        logger.info("Tải hình ảnh lên put_url")
         response = requests.put(put_url, data=image_data, headers={"Content-Type": mime_type})
         response.raise_for_status()
         etag = response.headers.get("etag")
         if not etag:
-            return None, "Error: No etag returned."
+            return None, "Lỗi: Không nhận được etag."
     except requests.exceptions.RequestException as e:
-        logger.error(f"Upload image error: {str(e)}")
-        return None, f"Error uploading image: {str(e)}"
+        logger.error(f"Lỗi tải hình ảnh: {str(e)}")
+        return None, f"Lỗi khi tải hình ảnh: {str(e)}"
     
-    # Step 3: Complete upload
+    # Bước 3: Hoàn tất tải lên
     url = f"{VIDU_API_URL}/tools/v2/files/uploads/{resource_id}/finish"
     payload = {"etag": etag.strip('"')}
     
     try:
-        logger.info("Completing upload")
+        logger.info("Hoàn tất tải lên")
         response = requests.put(url, json=payload, headers=headers)
         response.raise_for_status()
         result = response.json()
         uri = result.get("uri")
         if not uri:
-            return None, "Error: No URI returned."
-        return uri, "Image uploaded successfully."
+            return None, "Lỗi: Không nhận được URI."
+        return uri, "Tải hình ảnh lên thành công."
     except requests.exceptions.RequestException as e:
-        logger.error(f"Complete upload error: {str(e)}")
-        return None, f"Error completing upload: {str(e)}"
+        logger.error(f"Lỗi hoàn tất tải lên: {str(e)}")
+        return None, f"Lỗi khi hoàn tất tải lên: {str(e)}"
 
-# Function to call Vidu API for Start-End to Video
-def start_end_to_video(start_image, end_image, prompt, model="vidu2.0", resolution="720p", duration=4, movement_amplitude="auto", seed=None):
+# Hàm gọi API TDNM cho Start-End to Video
+def start_end_to_video(start_image, end_image, prompt, model="vidu2.0", resolution="720p", duration=4, movement_amplitude="auto", seed=None, user_key=None):
+    # Kiểm tra TDNM_KEY
+    valid_key, key_message = validate_tdn_key(user_key)
+    if not valid_key:
+        return None, key_message
+    
     if not VIDU_API_KEY:
-        return None, "Error: Vidu API key not configured."
+        return None, "Lỗi: Khóa API TDNM chưa được cấu hình."
     
     if not start_image or not end_image:
-        return None, "Error: Both start and end images are required."
+        return None, "Lỗi: Cần cung cấp cả hai hình ảnh đầu và cuối."
     
     if prompt and len(prompt) > 1500:
-        return None, "Error: Prompt must be 1500 characters or less."
+        return None, "Lỗi: Mô tả văn bản không được vượt quá 1500 ký tự."
     
-    # Validate pixel density
+    # Kiểm tra mật độ điểm ảnh
     valid, error_message = validate_pixel_density(start_image, end_image)
     if not valid:
         return None, error_message
     
-    # Upload images to Vidu
+    # Tải hình ảnh lên TDNM
     start_uri, start_message = upload_image_to_vidu(start_image)
     if not start_uri:
         return None, start_message
@@ -161,28 +177,32 @@ def start_end_to_video(start_image, end_image, prompt, model="vidu2.0", resoluti
         payload["seed"] = seed
     
     try:
-        logger.info(f"Sending request to Start-End to Video API: {payload}")
+        logger.info(f"Gửi yêu cầu đến API TDNM Start-End to Video: {payload}")
         response = requests.post(url, json=payload, headers=headers)
         response.raise_for_status()
         result = response.json()
         task_id = result.get("task_id")
         if not task_id:
-            return None, "Error: No task ID returned."
-        return task_id, f"Task created successfully. Task ID: {task_id}"
+            return None, "Lỗi: Không nhận được ID tác vụ."
+        return task_id, f"Tác vụ được tạo thành công. ID tác vụ: {task_id}"
     except requests.exceptions.RequestException as e:
-        logger.error(f"API error: {str(e)}")
+        logger.error(f"Lỗi API: {str(e)}")
         if response.text:
-            logger.error(f"API response: {response.text}")
-        return None, f"Error: {str(e)} - {response.text}"
+            logger.error(f"Phản hồi API: {response.text}")
+        return None, f"Lỗi: {str(e)} - {response.text}"
 
-# Placeholder for References to Video
-def references_to_video(reference_images, prompt, resolution="720p", duration=4):
-    return None, "Error: References to Video not implemented. Please provide API details."
+# Placeholder cho References to Video
+def references_to_video(reference_images, prompt, resolution="720p", duration=4, user_key=None):
+    # Kiểm tra TDNM_KEY
+    valid_key, key_message = validate_tdn_key(user_key)
+    if not valid_key:
+        return None, key_message
+    return None, "Lỗi: Chức năng References to Video chưa được triển khai. Vui lòng cung cấp chi tiết API."
 
-# Function to check task status
+# Hàm kiểm tra trạng thái tác vụ
 def check_task_status(task_id):
     if not task_id:
-        return None, "Error: Invalid task ID."
+        return None, "Lỗi: ID tác vụ không hợp lệ."
     
     url = f"{VIDU_API_URL}/ent/v2/tasks/{task_id}/creations"
     headers = {
@@ -193,7 +213,7 @@ def check_task_status(task_id):
     start_time = time.time()
     while time.time() - start_time < TIMEOUT:
         try:
-            logger.info(f"Checking task status for task_id: {task_id}")
+            logger.info(f"Kiểm tra trạng thái tác vụ cho task_id: {task_id}")
             response = requests.get(url, headers=headers)
             response.raise_for_status()
             result = response.json()
@@ -204,89 +224,93 @@ def check_task_status(task_id):
             if state == "success" and creations:
                 video_url = creations[0].get("url")
                 if video_url:
-                    return video_url, "Video generated successfully!"
-                return None, "Error: No video URL in creations."
+                    return video_url, "Tạo video thành công!"
+                return None, "Lỗi: Không có URL video trong kết quả."
             elif state == "failed":
-                return None, f"Error: Video generation failed. Error code: {err_code or 'Unknown'}"
+                return None, f"Lỗi: Tạo video thất bại. Mã lỗi: {err_code or 'Không xác định'}"
             elif state in ["created", "queueing", "processing"]:
                 time.sleep(POLL_INTERVAL)
             else:
-                return None, f"Error: Unknown state {state}."
+                return None, f"Lỗi: Trạng thái không xác định {state}."
         except requests.exceptions.RequestException as e:
-            logger.error(f"Status check error: {str(e)}")
+            logger.error(f"Lỗi kiểm tra trạng thái: {str(e)}")
             if response.text:
-                logger.error(f"Status check response: {response.text}")
-            return None, f"Error checking task status: {str(e)} - {response.text}"
+                logger.error(f"Phản hồi kiểm tra trạng thái: {response.text}")
+            return None, f"Lỗi khi kiểm tra trạng thái tác vụ: {str(e)} - {response.text}"
     
-    return None, f"Error: Video generation timed out. Task ID: {task_id}"
+    return None, f"Lỗi: Hết thời gian tạo video. ID tác vụ: {task_id}"
 
-# Gradio interface function for Start-End to Video
-def gradio_start_end_to_video(start_image, end_image, prompt, model, resolution, duration, movement_amplitude, seed):
+# Hàm giao diện Gradio cho Start-End to Video
+def gradio_start_end_to_video(start_image, end_image, prompt, model, resolution, duration, movement_amplitude, seed, user_key):
     if not start_image or not end_image:
-        return None, "Error: Both start and end images are required."
-    task_id, message = start_end_to_video(start_image, end_image, prompt, model, resolution, duration, movement_amplitude, seed)
+        return None, "Lỗi: Cần cung cấp cả hai hình ảnh đầu và cuối."
+    task_id, message = start_end_to_video(start_image, end_image, prompt, model, resolution, duration, movement_amplitude, seed, user_key)
     if not task_id:
         return None, message
     video_url, status_message = check_task_status(task_id)
     return video_url, status_message
 
-# Gradio interface function for References to Video
-def gradio_references_to_video(image1, image2, image3, prompt, resolution, duration):
+# Hàm giao diện Gradio cho References to Video
+def gradio_references_to_video(image1, image2, image3, prompt, resolution, duration, user_key):
     reference_images = [img for img in [image1, image2, image3] if img]
     if not reference_images:
-        return None, "Error: At least one reference image is required."
+        return None, "Lỗi: Cần cung cấp ít nhất một hình ảnh tham chiếu."
     if not prompt:
-        return None, "Error: Prompt is required."
-    task_id, message = references_to_video(reference_images, prompt, resolution, duration)
+        return None, "Lỗi: Cần cung cấp mô tả văn bản."
+    task_id, message = references_to_video(reference_images, prompt, resolution, duration, user_key)
     if not task_id:
         return None, message
     video_url, status_message = check_task_status(task_id)
     return video_url, status_message
 
-# Gradio interface
-with gr.Blocks(title="LÀM HOẠT HÌNH @PRO") as demo:
-    gr.Markdown("# AI APPS | by TDNM")
-    gr.Markdown("SẢN XUẤT HOẠT HÌNH CHUYÊN NGHIỆP")
-    # Tab for Start-End to Video
-    with gr.Tab("PHƯƠNG PHÁP ĐẦU - CUỐI"):
-        gr.Markdown("Tải lên hình ảnh bắt đầu và kết thúc và lời nhắc để tạo video chuyển tiếp. Hình ảnh phải là PNG, WebP, JPEG hoặc JPG, <10MB, với tỷ lệ khung hình từ 1:4 đến 4:1.")
-        start_image = gr.Image(type="filepath", label="Hình ảnh bắt đầu")
-        end_image = gr.Image(type="filepath", label="Hình ảnh kết thúc")
-        prompt_se = gr.Textbox(label="Prompt (Optional)", placeholder="E.g., 'Smoothly transition from a car chassis to a complete car.'")
-        model_se = gr.Dropdown(choices=["vidu2.0", "vidu1.5"], label="Model", value="vidu2.0")
-        resolution_se = gr.Dropdown(choices=["360p", "720p", "1080p"], label="Resolution", value="720p")
-        duration_se = gr.Dropdown(choices=[4, 8], label="Duration (seconds)", value=4)
-        movement_amplitude_se = gr.Dropdown(choices=["auto", "small", "medium", "large"], label="Movement Amplitude", value="auto")
-        seed_se = gr.Number(label="Seed (Optional)", value=None, precision=0)
-        se_button = gr.Button("Generate Video")
-        se_video_output = gr.Video(label="Generated Video")
-        se_message = gr.Textbox(label="Status")
+# Giao diện Gradio
+with gr.Blocks(title="Trình Tạo Video TDNM") as demo:
+    gr.Markdown("# Trình Tạo Video TDNM")
+    gr.Markdown("Tạo video với TDNM. Vui lòng nhập khóa bí mật (TDNM_KEY) để sử dụng ứng dụng.")
+
+    # Trường nhập khóa bí mật (áp dụng cho cả hai tab)
+    user_key = gr.Textbox(label="Khóa Bí Mật (TDNM_KEY)", type="password", placeholder="Nhập khóa bí mật của bạn")
+
+    # Tab cho Start-End to Video
+    with gr.Tab("Video Chuyển Đổi Hình Ảnh"):
+        gr.Markdown("Tải lên hai hình ảnh (đầu và cuối) và mô tả văn bản để tạo video chuyển đổi. Hình ảnh phải là PNG, WebP, JPEG hoặc JPG, kích thước dưới 10MB, tỷ lệ khung hình từ 1:4 đến 4:1.")
+        start_image = gr.Image(type="filepath", label="Hình Ảnh Đầu")
+        end_image = gr.Image(type="filepath", label="Hình Ảnh Cuối")
+        prompt_se = gr.Textbox(label="Mô Tả Văn Bản (Tùy Chọn)", placeholder="Ví dụ: 'Chuyển đổi mượt mà từ khung xe thành xe hoàn chỉnh.'")
+        model_se = gr.Dropdown(choices=["vidu2.0", "vidu1.5"], label="Mô Hình", value="vidu2.0")
+        resolution_se = gr.Dropdown(choices=["360p", "720p", "1080p"], label="Độ Phân Giải", value="720p")
+        duration_se = gr.Dropdown(choices=[4, 8], label="Thời Lượng (giây)", value=4)
+        movement_amplitude_se = gr.Dropdown(choices=["auto", "small", "medium", "large"], label="Biên Độ Chuyển Động", value="auto")
+        seed_se = gr.Number(label="Hạt Giống (Tùy Chọn)", value=None, precision=0)
+        se_button = gr.Button("Tạo Video")
+        se_video_output = gr.Video(label="Video Được Tạo")
+        se_message = gr.Textbox(label="Trạng Thái")
 
         se_button.click(
             fn=gradio_start_end_to_video,
-            inputs=[start_image, end_image, prompt_se, model_se, resolution_se, duration_se, movement_amplitude_se, seed_se],
+            inputs=[start_image, end_image, prompt_se, model_se, resolution_se, duration_se, movement_amplitude_se, seed_se, user_key],
             outputs=[se_video_output, se_message]
         )
 
-    # Tab for References to Video (Placeholder)
-    with gr.Tab("References to Video"):
-        gr.Markdown("Upload 1–3 reference images and a prompt to generate a video. (Not implemented yet.)")
-        image1 = gr.Image(type="filepath", label="Reference Image 1")
-        image2 = gr.Image(type="filepath", label="Reference Image 2 (Optional)")
-        image3 = gr.Image(type="filepath", label="Reference Image 3 (Optional)")
-        prompt_ref = gr.Textbox(label="Prompt", placeholder="E.g., 'A character walking in a forest.'")
-        resolution_ref = gr.Dropdown(choices=["360p", "720p", "1080p"], label="Resolution", value="720p")
-        duration_ref = gr.Dropdown(choices=[4, 8], label="Duration (seconds)", value=4)
-        ref_button = gr.Button("Generate Video")
-        ref_video_output = gr.Video(label="Generated Video")
-        ref_message = gr.Textbox(label="Status")
+    # Tab cho References to Video (Placeholder)
+    with gr.Tab("Video Từ Hình Ảnh Tham Chiếu"):
+        gr.Markdown("Tải lên 1–3 hình ảnh tham chiếu và mô tả văn bản để tạo video. (Chưa được triển khai.)")
+        image1 = gr.Image(type="filepath", label="Hình Ảnh Tham Chiếu 1")
+        image2 = gr.Image(type="filepath", label="Hình Ảnh Tham Chiếu 2 (Tùy Chọn)")
+        image3 = gr.Image(type="filepath", label="Hình Ảnh Tham Chiếu 3 (Tùy Chọn)")
+        prompt_ref = gr.Textbox(label="Mô Tả Văn Bản", placeholder="Ví dụ: 'Một nhân vật đi bộ trong rừng.'")
+        resolution_ref = gr.Dropdown(choices=["360p", "720p", "1080p"], label="Độ Phân Giải", value="720p")
+        duration_ref = gr.Dropdown(choices=[4, 8], label="Thời Lượng (giây)", value=4)
+        ref_button = gr.Button("Tạo Video")
+        ref_video_output = gr.Video(label="Video Được Tạo")
+        ref_message = gr.Textbox(label="Trạng Thái")
 
         ref_button.click(
             fn=gradio_references_to_video,
-            inputs=[image1, image2, image3, prompt_ref, resolution_ref, duration_ref],
+            inputs=[image1, image2, image3, prompt_ref, resolution_ref, duration_ref, user_key],
             outputs=[ref_video_output, ref_message]
         )
 
-# Launch the app (not needed for HF Spaces, included for local testing)
+# Khởi chạy ứng dụng (không cần cho HF Spaces, dùng để kiểm tra cục bộ)
 if __name__ == "__main__":
     demo.launch()