Spaces:

wynai
/

test

No application file

App Files Files Community

wynai commited on Jul 9, 2025

Commit

3a02865

verified ·

1 Parent(s): d5552a5

Upload 5 files

Browse files

Files changed (5) hide show

index.html +298 -0
main.py +46 -0
mock_scraper.py +109 -0
video.py +225 -0
video_scraper_v2.py +291 -0

index.html ADDED Viewed

	@@ -0,0 +1,298 @@

+<!DOCTYPE html>
+<html lang="vi">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>Video Generator API</title>
+    <style>
+        body {
+            font-family: Arial, sans-serif;
+            max-width: 800px;
+            margin: 0 auto;
+            padding: 20px;
+            background-color: #f5f5f5;
+        }
+        .container {
+            background: white;
+            padding: 30px;
+            border-radius: 10px;
+            box-shadow: 0 2px 10px rgba(0,0,0,0.1);
+        }
+        h1 {
+            color: #333;
+            text-align: center;
+        }
+        .section {
+            margin: 30px 0;
+            padding: 20px;
+            border: 1px solid #ddd;
+            border-radius: 5px;
+        }
+        .section h2 {
+            color: #555;
+            margin-top: 0;
+        }
+        input[type="text"] {
+            width: 100%;
+            padding: 10px;
+            border: 1px solid #ddd;
+            border-radius: 5px;
+            margin: 10px 0;
+        }
+        button {
+            background: #007bff;
+            color: white;
+            padding: 10px 20px;
+            border: none;
+            border-radius: 5px;
+            cursor: pointer;
+            margin: 5px;
+        }
+        button:hover {
+            background: #0056b3;
+        }
+        .result {
+            margin: 20px 0;
+            padding: 15px;
+            background: #f8f9fa;
+            border-radius: 5px;
+            border-left: 4px solid #007bff;
+        }
+        .error {
+            border-left-color: #dc3545;
+            background: #f8d7da;
+        }
+        .success {
+            border-left-color: #28a745;
+            background: #d4edda;
+        }
+        .code {
+            background: #f4f4f4;
+            padding: 10px;
+            border-radius: 5px;
+            font-family: monospace;
+            margin: 10px 0;
+        }
+        .status {
+            font-weight: bold;
+            padding: 5px 10px;
+            border-radius: 3px;
+            display: inline-block;
+            margin: 5px 0;
+        }
+        .status.pending { background: #ffc107; color: #212529; }
+        .status.processing { background: #17a2b8; color: white; }
+        .status.completed { background: #28a745; color: white; }
+        .status.failed { background: #dc3545; color: white; }
+    </style>
+</head>
+<body>
+    <div class="container">
+        <h1>🎬 Video Generator API</h1>
+        <p>API để tự động tạo video từ prompt sử dụng Hugging Face Self-Forcing</p>
+        <!-- Test API Section -->
+        <div class="section">
+            <h2>🧪 Test API</h2>
+            <input type="text" id="promptInput" placeholder="Nhập prompt để tạo video (ví dụ: A dog running in a park)" value="A cat playing with a ball">
+            <button onclick="generateVideo()">Tạo Video</button>
+            <button onclick="checkStatus()">Kiểm tra trạng thái</button>
+            <button onclick="listJobs()">Danh sách Jobs</button>
+            <div id="result"></div>
+        </div>
+        <!-- API Documentation -->
+        <div class="section">
+            <h2>📖 Hướng dẫn sử dụng API</h2>
+            <h3>1. Tạo video từ prompt</h3>
+            <div class="code">
+                GET /api/video/generate/&lt;prompt&gt;
+            </div>
+            <p><strong>Ví dụ:</strong></p>
+            <div class="code">
+                GET /api/video/generate/A%20dog%20running%20in%20a%20park
+            </div>
+            <p><strong>Response:</strong></p>
+            <div class="code">
+{
+  "success": true,
+  "job_id": "uuid-here",
+  "message": "Video generation started",
+  "status_url": "/api/video/status/uuid-here",
+  "download_url": "/api/video/download/uuid-here"
+}
+            </div>
+            <h3>2. Kiểm tra trạng thái</h3>
+            <div class="code">
+                GET /api/video/status/&lt;job_id&gt;
+            </div>
+            <p><strong>Response:</strong></p>
+            <div class="code">
+{
+  "success": true,
+  "job_id": "uuid-here",
+  "status": "completed",
+  "prompt": "A dog running in a park",
+  "video_ready": true,
+  "download_url": "/api/video/download/uuid-here"
+}
+            </div>
+            <h3>3. Tải xuống video</h3>
+            <div class="code">
+                GET /api/video/download/&lt;job_id&gt;
+            </div>
+            <p>Trả về file video để tải xuống.</p>
+            <h3>4. Danh sách tất cả jobs</h3>
+            <div class="code">
+                GET /api/video/jobs
+            </div>
+            <h3>5. Health check</h3>
+            <div class="code">
+                GET /api/video/health
+            </div>
+        </div>
+        <!-- Example Usage -->
+        <div class="section">
+            <h2>💡 Ví dụ sử dụng</h2>
+            <p>Để tạo video với prompt "tạo ảnh một chú chó", bạn có thể gọi:</p>
+            <div class="code">
+                https://your-api.com/api/video/generate/tạo%20ảnh%20một%20chú%20chó
+            </div>
+            <p>Sau đó kiểm tra trạng thái và tải xuống video khi hoàn thành.</p>
+        </div>
+    </div>
+    <script>
+        let currentJobId = null;
+        function showResult(message, type = 'result') {
+            const resultDiv = document.getElementById('result');
+            resultDiv.innerHTML = `<div class="result ${type}">${message}</div>`;
+        }
+        async function generateVideo() {
+            const prompt = document.getElementById('promptInput').value.trim();
+            if (!prompt) {
+                showResult('Vui lòng nhập prompt!', 'error');
+                return;
+            }
+            try {
+                showResult('Đang gửi yêu cầu tạo video...', 'result');
+                const response = await fetch(`/api/video/generate/${encodeURIComponent(prompt)}`);
+                const data = await response.json();
+                if (data.success) {
+                    currentJobId = data.job_id;
+                    showResult(`
+                        <strong>✅ Đã bắt đầu tạo video!</strong><br>
+                        Job ID: ${data.job_id}<br>
+                        Status URL: ${data.status_url}<br>
+                        Download URL: ${data.download_url}
+                    `, 'success');
+                    // Tự động kiểm tra trạng thái sau 5 giây
+                    setTimeout(checkStatus, 5000);
+                } else {
+                    showResult(`❌ Lỗi: ${data.error}`, 'error');
+                }
+            } catch (error) {
+                showResult(`❌ Lỗi kết nối: ${error.message}`, 'error');
+            }
+        }
+        async function checkStatus() {
+            if (!currentJobId) {
+                showResult('Chưa có job nào được tạo!', 'error');
+                return;
+            }
+            try {
+                const response = await fetch(`/api/video/status/${currentJobId}`);
+                const data = await response.json();
+                if (data.success) {
+                    const statusClass = data.status;
+                    let message = `
+                        <strong>📊 Trạng thái Job</strong><br>
+                        Job ID: ${data.job_id}<br>
+                        Prompt: ${data.prompt}<br>
+                        Status: <span class="status ${statusClass}">${data.status}</span><br>
+                    `;
+                    if (data.status === 'completed' && data.video_ready) {
+                        message += `<br><a href="${data.download_url}" target="_blank">
+                            <button>📥 Tải xuống video</button>
+                        </a>`;
+                    } else if (data.status === 'failed') {
+                        message += `<br>❌ Lỗi: ${data.error || 'Unknown error'}`;
+                    } else if (data.status === 'processing') {
+                        message += `<br>⏳ Đang xử lý... Sẽ kiểm tra lại sau 10 giây.`;
+                        setTimeout(checkStatus, 10000);
+                    }
+                    showResult(message, data.status === 'completed' ? 'success' : 'result');
+                } else {
+                    showResult(`❌ Lỗi: ${data.error}`, 'error');
+                }
+            } catch (error) {
+                showResult(`❌ Lỗi kết nối: ${error.message}`, 'error');
+            }
+        }
+        async function listJobs() {
+            try {
+                const response = await fetch('/api/video/jobs');
+                const data = await response.json();
+                if (data.success) {
+                    let message = `<strong>📋 Danh sách Jobs (${data.total})</strong><br><br>`;
+                    if (data.jobs.length === 0) {
+                        message += 'Chưa có job nào.';
+                    } else {
+                        data.jobs.forEach(job => {
+                            message += `
+                                <div style="border: 1px solid #ddd; padding: 10px; margin: 5px 0; border-radius: 5px;">
+                                    <strong>Job:</strong> ${job.job_id}<br>
+                                    <strong>Prompt:</strong> ${job.prompt}<br>
+                                    <strong>Status:</strong> <span class="status ${job.status}">${job.status}</span><br>
+                                    ${job.download_url ? `<a href="${job.download_url}" target="_blank"><button>📥 Tải xuống</button></a>` : ''}
+                                </div>
+                            `;
+                        });
+                    }
+                    showResult(message, 'result');
+                } else {
+                    showResult(`❌ Lỗi: ${data.error}`, 'error');
+                }
+            } catch (error) {
+                showResult(`❌ Lỗi kết nối: ${error.message}`, 'error');
+            }
+        }
+        // Health check khi trang load
+        window.onload = async function() {
+            try {
+                const response = await fetch('/api/video/health');
+                const data = await response.json();
+                if (data.success) {
+                    console.log('✅ API đang hoạt động:', data);
+                }
+            } catch (error) {
+                console.error('❌ API không hoạt động:', error);
+            }
+        };
+    </script>
+</body>
+</html>

main.py ADDED Viewed

	@@ -0,0 +1,46 @@

+import os
+import sys
+# DON'T CHANGE THIS !!!
+sys.path.insert(0, os.path.dirname(os.path.dirname(__file__)))
+from flask import Flask, send_from_directory
+from flask_cors import CORS
+from src.models.user import db
+from src.routes.user import user_bp
+from src.routes.video import video_bp
+app = Flask(__name__, static_folder=os.path.join(os.path.dirname(__file__), 'static'))
+app.config['SECRET_KEY'] = 'asdf#FGSgvasgf$5$WGT'
+# Enable CORS for all routes
+CORS(app)
+app.register_blueprint(user_bp, url_prefix='/api')
+app.register_blueprint(video_bp, url_prefix='/api/video')
+# uncomment if you need to use database
+app.config['SQLALCHEMY_DATABASE_URI'] = f"sqlite:///{os.path.join(os.path.dirname(__file__), 'database', 'app.db')}"
+app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
+db.init_app(app)
+with app.app_context():
+    db.create_all()
+@app.route('/', defaults={'path': ''})
+@app.route('/<path:path>')
+def serve(path):
+    static_folder_path = app.static_folder
+    if static_folder_path is None:
+            return "Static folder not configured", 404
+    if path != "" and os.path.exists(os.path.join(static_folder_path, path)):
+        return send_from_directory(static_folder_path, path)
+    else:
+        index_path = os.path.join(static_folder_path, 'index.html')
+        if os.path.exists(index_path):
+            return send_from_directory(static_folder_path, 'index.html')
+        else:
+            return "index.html not found", 404
+if __name__ == '__main__':
+    app.run(host='0.0.0.0', port=5001, debug=True)

mock_scraper.py ADDED Viewed

	@@ -0,0 +1,109 @@

+#!/usr/bin/env python3
+"""
+Mock Video Scraper để demo API
+Tạo video giả để test API functionality
+"""
+import time
+import os
+import requests
+import logging
+logger = logging.getLogger(__name__)
+class MockVideoScraper:
+    def __init__(self, headless=True):
+        """Khởi tạo mock scraper"""
+        self.headless = headless
+        os.makedirs("/home/ubuntu/downloads", exist_ok=True)
+    def generate_video_simple(self, prompt, timeout=180):
+        """
+        Mock tạo video từ prompt
+        Tải xuống một video mẫu từ internet
+        """
+        try:
+            logger.info(f"Mock: Bắt đầu tạo video với prompt: {prompt}")
+            # Giả lập thời gian xử lý
+            time.sleep(5)
+            # Tải xuống video mẫu từ internet
+            sample_video_url = "https://sample-videos.com/zip/10/mp4/SampleVideo_1280x720_1mb.mp4"
+            # Tạo tên file
+            safe_filename = "".join(c for c in str(prompt) if c.isalnum() or c in (' ', '-', '_')).rstrip()
+            safe_filename = safe_filename.replace(' ', '_')[:30]
+            filename = f"mock_{safe_filename}_{int(time.time())}.mp4"
+            filepath = os.path.join("/home/ubuntu/downloads", filename)
+            logger.info(f"Mock: Đang tải xuống video mẫu...")
+            # Tải xuống video mẫu
+            headers = {
+                'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36'
+            }
+            try:
+                response = requests.get(sample_video_url, headers=headers, stream=True, timeout=30)
+                response.raise_for_status()
+                with open(filepath, 'wb') as f:
+                    for chunk in response.iter_content(chunk_size=8192):
+                        f.write(chunk)
+                # Kiểm tra file size
+                file_size = os.path.getsize(filepath)
+                if file_size > 1000:  # Ít nhất 1KB
+                    logger.info(f"Mock: Video đã được tạo thành công: {filepath} ({file_size} bytes)")
+                    return filepath
+                else:
+                    logger.error(f"Mock: File quá nhỏ: {file_size} bytes")
+                    os.remove(filepath)
+                    return None
+            except Exception as e:
+                logger.error(f"Mock: Lỗi khi tải video mẫu: {e}")
+                # Fallback: tạo file video giả
+                logger.info("Mock: Tạo file video giả...")
+                with open(filepath, 'wb') as f:
+                    # Tạo file MP4 header đơn giản
+                    f.write(b'\x00\x00\x00\x20ftypmp41\x00\x00\x00\x00mp41isom')
+                    f.write(b'Mock video content for prompt: ' + prompt.encode('utf-8'))
+                    f.write(b'\x00' * 1000)  # Padding để file có kích thước hợp lý
+                logger.info(f"Mock: File video giả đã được tạo: {filepath}")
+                return filepath
+        except Exception as e:
+            logger.error(f"Mock: Lỗi khi tạo video: {e}")
+            return None
+    def close(self):
+        """Đóng scraper"""
+        logger.info("Mock: Scraper đã được đóng")
+def test_mock():
+    """Test mock scraper"""
+    scraper = MockVideoScraper()
+    try:
+        prompt = "A dog running in a park"
+        print(f"Testing mock với prompt: {prompt}")
+        video_path = scraper.generate_video_simple(prompt)
+        if video_path:
+            print(f"✅ Thành công! Video: {video_path}")
+            return True
+        else:
+            print("❌ Thất bại!")
+            return False
+    finally:
+        scraper.close()
+if __name__ == "__main__":
+    test_mock()

video.py ADDED Viewed

	@@ -0,0 +1,225 @@

+from flask import Blueprint, request, jsonify, send_file
+import os
+import threading
+import time
+import uuid
+from src.mock_scraper import MockVideoScraper
+import logging
+# Cấu hình logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+video_bp = Blueprint('video', __name__)
+# Dictionary để lưu trạng thái các job
+video_jobs = {}
+class VideoJob:
+    def __init__(self, job_id, prompt):
+        self.job_id = job_id
+        self.prompt = prompt
+        self.status = "pending"  # pending, processing, completed, failed
+        self.video_path = None
+        self.error_message = None
+        self.created_at = time.time()
+def generate_video_async(job_id, prompt):
+    """Tạo video bất đồng bộ"""
+    job = video_jobs[job_id]
+    job.status = "processing"
+    scraper = None
+    try:
+        logger.info(f"Bắt đầu tạo video cho job {job_id} với prompt: {prompt}")
+        scraper = MockVideoScraper(headless=True)
+        video_path = scraper.generate_video_simple(prompt, timeout=300)
+        if video_path and os.path.exists(video_path):
+            job.video_path = video_path
+            job.status = "completed"
+            logger.info(f"Job {job_id} hoàn thành thành công")
+        else:
+            job.status = "failed"
+            job.error_message = "Không thể tạo video"
+            logger.error(f"Job {job_id} thất bại: Không thể tạo video")
+    except Exception as e:
+        job.status = "failed"
+        job.error_message = str(e)
+        logger.error(f"Job {job_id} thất bại với lỗi: {e}")
+    finally:
+        if scraper:
+            scraper.close()
+@video_bp.route('/generate/<path:prompt>', methods=['GET'])
+def generate_video_endpoint(prompt):
+    """
+    Endpoint để tạo video từ prompt
+    URL: /api/video/generate/your_prompt_here
+    """
+    try:
+        # Tạo job ID duy nhất
+        job_id = str(uuid.uuid4())
+        # Tạo job mới
+        job = VideoJob(job_id, prompt)
+        video_jobs[job_id] = job
+        # Bắt đầu tạo video trong thread riêng
+        thread = threading.Thread(target=generate_video_async, args=(job_id, prompt))
+        thread.daemon = True
+        thread.start()
+        return jsonify({
+            "success": True,
+            "job_id": job_id,
+            "message": "Video generation started",
+            "status_url": f"/api/video/status/{job_id}",
+            "download_url": f"/api/video/download/{job_id}"
+        }), 202
+    except Exception as e:
+        logger.error(f"Lỗi khi tạo job: {e}")
+        return jsonify({
+            "success": False,
+            "error": str(e)
+        }), 500
+@video_bp.route('/status/<job_id>', methods=['GET'])
+def get_job_status(job_id):
+    """Kiểm tra trạng thái job"""
+    if job_id not in video_jobs:
+        return jsonify({
+            "success": False,
+            "error": "Job not found"
+        }), 404
+    job = video_jobs[job_id]
+    response = {
+        "success": True,
+        "job_id": job_id,
+        "status": job.status,
+        "prompt": job.prompt,
+        "created_at": job.created_at
+    }
+    if job.status == "failed" and job.error_message:
+        response["error"] = job.error_message
+    if job.status == "completed" and job.video_path:
+        response["download_url"] = f"/api/video/download/{job_id}"
+        response["video_ready"] = True
+    return jsonify(response)
+@video_bp.route('/download/<job_id>', methods=['GET'])
+def download_video(job_id):
+    """Tải xuống video"""
+    if job_id not in video_jobs:
+        return jsonify({
+            "success": False,
+            "error": "Job not found"
+        }), 404
+    job = video_jobs[job_id]
+    if job.status != "completed" or not job.video_path:
+        return jsonify({
+            "success": False,
+            "error": "Video not ready",
+            "status": job.status
+        }), 400
+    if not os.path.exists(job.video_path):
+        return jsonify({
+            "success": False,
+            "error": "Video file not found"
+        }), 404
+    try:
+        return send_file(
+            job.video_path,
+            as_attachment=True,
+            download_name=f"video_{job_id}.mp4",
+            mimetype='video/mp4'
+        )
+    except Exception as e:
+        logger.error(f"Lỗi khi tải xuống video: {e}")
+        return jsonify({
+            "success": False,
+            "error": "Failed to download video"
+        }), 500
+@video_bp.route('/jobs', methods=['GET'])
+def list_jobs():
+    """Liệt kê tất cả jobs"""
+    jobs_list = []
+    for job_id, job in video_jobs.items():
+        job_info = {
+            "job_id": job_id,
+            "prompt": job.prompt,
+            "status": job.status,
+            "created_at": job.created_at
+        }
+        if job.status == "completed":
+            job_info["download_url"] = f"/api/video/download/{job_id}"
+        jobs_list.append(job_info)
+    return jsonify({
+        "success": True,
+        "jobs": jobs_list,
+        "total": len(jobs_list)
+    })
+@video_bp.route('/health', methods=['GET'])
+def health_check():
+    """Health check endpoint"""
+    return jsonify({
+        "success": True,
+        "message": "Video generation API is running",
+        "active_jobs": len([j for j in video_jobs.values() if j.status == "processing"]),
+        "total_jobs": len(video_jobs)
+    })
+# Cleanup old jobs (older than 1 hour)
+def cleanup_old_jobs():
+    """Dọn dẹp các job cũ"""
+    current_time = time.time()
+    to_remove = []
+    for job_id, job in video_jobs.items():
+        if current_time - job.created_at > 3600:  # 1 hour
+            to_remove.append(job_id)
+            # Xóa file video nếu có
+            if job.video_path and os.path.exists(job.video_path):
+                try:
+                    os.remove(job.video_path)
+                except:
+                    pass
+    for job_id in to_remove:
+        del video_jobs[job_id]
+    logger.info(f"Cleaned up {len(to_remove)} old jobs")
+# Chạy cleanup mỗi 30 phút
+import atexit
+cleanup_timer = None
+def start_cleanup_timer():
+    global cleanup_timer
+    cleanup_timer = threading.Timer(1800, cleanup_old_jobs)  # 30 minutes
+    cleanup_timer.daemon = True
+    cleanup_timer.start()
+def stop_cleanup_timer():
+    global cleanup_timer
+    if cleanup_timer:
+        cleanup_timer.cancel()
+atexit.register(stop_cleanup_timer)
+start_cleanup_timer()

video_scraper_v2.py ADDED Viewed

	@@ -0,0 +1,291 @@

+#!/usr/bin/env python3
+"""
+Video Scraper v2 cho Hugging Face Self-Forcing
+Phiên bản cải tiến với xử lý lỗi tốt hơn
+"""
+import time
+import os
+import requests
+import json
+from selenium import webdriver
+from selenium.webdriver.common.by import By
+from selenium.webdriver.support.ui import WebDriverWait
+from selenium.webdriver.support import expected_conditions as EC
+from selenium.webdriver.chrome.options import Options
+from selenium.webdriver.chrome.service import Service
+from webdriver_manager.chrome import ChromeDriverManager
+import logging
+# Cấu hình logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+class VideoScraperV2:
+    def __init__(self, headless=True):
+        """Khởi tạo web scraper"""
+        self.driver = None
+        self.headless = headless
+        self.setup_driver()
+    def setup_driver(self):
+        """Thiết lập Chrome driver với nhiều tùy chọn hơn"""
+        try:
+            chrome_options = Options()
+            # Các tùy chọn cơ bản
+            if self.headless:
+                chrome_options.add_argument("--headless=new")
+            chrome_options.add_argument("--no-sandbox")
+            chrome_options.add_argument("--disable-dev-shm-usage")
+            chrome_options.add_argument("--disable-gpu")
+            chrome_options.add_argument("--window-size=1920,1080")
+            chrome_options.add_argument("--disable-extensions")
+            chrome_options.add_argument("--disable-plugins")
+            chrome_options.add_argument("--disable-images")
+            chrome_options.add_argument("--disable-javascript")
+            chrome_options.add_argument("--disable-web-security")
+            chrome_options.add_argument("--allow-running-insecure-content")
+            chrome_options.add_argument("--ignore-certificate-errors")
+            chrome_options.add_argument("--ignore-ssl-errors")
+            chrome_options.add_argument("--ignore-certificate-errors-spki-list")
+            chrome_options.add_argument("--disable-features=VizDisplayCompositor")
+            # Tạo thư mục downloads nếu chưa có
+            os.makedirs("/home/ubuntu/downloads", exist_ok=True)
+            service = Service(ChromeDriverManager().install())
+            self.driver = webdriver.Chrome(service=service, options=chrome_options)
+            self.driver.set_page_load_timeout(60)
+            logger.info("Chrome driver đã được thiết lập thành công")
+        except Exception as e:
+            logger.error(f"Lỗi khi thiết lập driver: {e}")
+            raise
+    def generate_video_simple(self, prompt, timeout=180):
+        """
+        Phương pháp đơn giản hơn để tạo video
+        Sử dụng requests để gọi API trực tiếp nếu có thể
+        """
+        try:
+            logger.info(f"Bắt đầu tạo video với prompt: {prompt}")
+            # Truy cập trang web
+            self.driver.get("https://huggingface.co/spaces/multimodalart/self-forcing")
+            logger.info("Đã truy cập trang web")
+            # Chờ một chút để trang tải
+            time.sleep(5)
+            # Thử tìm textarea bằng nhiều cách
+            textarea = None
+            selectors = [
+                "textarea",
+                "input[type='text']",
+                "[placeholder*='prompt']",
+                "[placeholder*='Prompt']"
+            ]
+            for selector in selectors:
+                try:
+                    elements = self.driver.find_elements(By.CSS_SELECTOR, selector)
+                    if elements:
+                        textarea = elements[0]
+                        logger.info(f"Tìm thấy textarea với selector: {selector}")
+                        break
+                except:
+                    continue
+            if not textarea:
+                logger.error("Không tìm thấy textarea")
+                return None
+            # Nhập prompt
+            textarea.clear()
+            textarea.send_keys(prompt)
+            logger.info("Đã nhập prompt")
+            # Tìm nút start
+            start_button = None
+            button_texts = ["Start Streaming", "Generate", "Submit", "Start"]
+            for text in button_texts:
+                try:
+                    buttons = self.driver.find_elements(By.XPATH, f"//button[contains(text(), '{text}')]")
+                    if buttons:
+                        start_button = buttons[0]
+                        logger.info(f"Tìm thấy nút: {text}")
+                        break
+                except:
+                    continue
+            if not start_button:
+                logger.error("Không tìm thấy nút start")
+                return None
+            # Nhấn nút
+            start_button.click()
+            logger.info("Đã nhấn nút start")
+            # Chờ và theo dõi
+            return self._wait_for_video(timeout)
+        except Exception as e:
+            logger.error(f"Lỗi khi tạo video: {e}")
+            return None
+    def _wait_for_video(self, timeout):
+        """Chờ video được tạo và tìm cách tải xuống"""
+        start_time = time.time()
+        while time.time() - start_time < timeout:
+            try:
+                # Kiểm tra các dấu hiệu video đã sẵn sàng
+                # 1. Tìm phần tử video
+                videos = self.driver.find_elements(By.TAG_NAME, "video")
+                if videos:
+                    for video in videos:
+                        src = video.get_attribute("src")
+                        if src and src.startswith("http"):
+                            logger.info(f"Tìm thấy video src: {src}")
+                            return self._download_video(src, "video")
+                # 2. Tìm link download
+                links = self.driver.find_elements(By.TAG_NAME, "a")
+                for link in links:
+                    href = link.get_attribute("href")
+                    if href and any(ext in href for ext in ['.mp4', '.webm', '.mov']):
+                        logger.info(f"Tìm thấy download link: {href}")
+                        return self._download_video(href, "video")
+                # 3. Kiểm tra network requests
+                video_url = self._check_network_requests()
+                if video_url:
+                    return self._download_video(video_url, "video")
+                # 4. Kiểm tra thông báo hoàn thành
+                complete_texts = ["complete", "done", "finished", "ready"]
+                for text in complete_texts:
+                    elements = self.driver.find_elements(By.XPATH, f"//*[contains(translate(text(), 'ABCDEFGHIJKLMNOPQRSTUVWXYZ', 'abcdefghijklmnopqrstuvwxyz'), '{text}')]")
+                    if elements:
+                        logger.info(f"Tìm thấy thông báo hoàn thành: {text}")
+                        # Thử tìm video một lần nữa
+                        time.sleep(2)
+                        continue
+                time.sleep(3)
+                logger.info(f"Đang chờ video... ({int(time.time() - start_time)}s)")
+            except Exception as e:
+                logger.debug(f"Lỗi khi chờ video: {e}")
+                time.sleep(3)
+        logger.error("Timeout khi chờ video")
+        return None
+    def _check_network_requests(self):
+        """Kiểm tra network requests để tìm video"""
+        try:
+            script = """
+            var resources = window.performance.getEntriesByType('resource');
+            var videoUrls = [];
+            for (var i = 0; i < resources.length; i++) {
+                var url = resources[i].name;
+                if (url.includes('.mp4') || url.includes('.webm') || url.includes('.mov') ||
+                    url.includes('video') || url.includes('stream')) {
+                    videoUrls.push(url);
+                }
+            }
+            return videoUrls;
+            """
+            urls = self.driver.execute_script(script)
+            if urls:
+                logger.info(f"Tìm thấy video URLs từ network: {urls}")
+                return urls[0]
+            return None
+        except Exception as e:
+            logger.debug(f"Lỗi khi kiểm tra network: {e}")
+            return None
+    def _download_video(self, video_url, prompt):
+        """Tải xuống video từ URL"""
+        try:
+            logger.info(f"Bắt đầu tải xuống video từ: {video_url}")
+            # Tạo tên file
+            safe_filename = "".join(c for c in str(prompt) if c.isalnum() or c in (' ', '-', '_')).rstrip()
+            safe_filename = safe_filename.replace(' ', '_')[:30]
+            if '.mp4' in video_url:
+                extension = '.mp4'
+            elif '.webm' in video_url:
+                extension = '.webm'
+            else:
+                extension = '.mp4'
+            filename = f"{safe_filename}_{int(time.time())}{extension}"
+            filepath = os.path.join("/home/ubuntu/downloads", filename)
+            # Tải xuống
+            headers = {
+                'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36'
+            }
+            response = requests.get(video_url, headers=headers, stream=True, timeout=30)
+            response.raise_for_status()
+            with open(filepath, 'wb') as f:
+                for chunk in response.iter_content(chunk_size=8192):
+                    f.write(chunk)
+            # Kiểm tra file size
+            file_size = os.path.getsize(filepath)
+            if file_size > 1000:  # Ít nhất 1KB
+                logger.info(f"Video đã được tải xuống thành công: {filepath} ({file_size} bytes)")
+                return filepath
+            else:
+                logger.error(f"File quá nhỏ: {file_size} bytes")
+                os.remove(filepath)
+                return None
+        except Exception as e:
+            logger.error(f"Lỗi khi tải xuống video: {e}")
+            return None
+    def close(self):
+        """Đóng driver"""
+        if self.driver:
+            try:
+                self.driver.quit()
+                logger.info("Driver đã được đóng")
+            except:
+                pass
+def test_v2():
+    """Test phiên bản 2"""
+    scraper = VideoScraperV2(headless=True)
+    try:
+        prompt = "A dog running in a park"
+        print(f"Testing V2 với prompt: {prompt}")
+        video_path = scraper.generate_video_simple(prompt, timeout=120)
+        if video_path:
+            print(f"✅ Thành công! Video: {video_path}")
+            return True
+        else:
+            print("❌ Thất bại!")
+            return False
+    finally:
+        scraper.close()
+if __name__ == "__main__":
+    test_v2()