Update webui/run_app.py

webui/run_app.py

@@ -1,31 +1,27 @@
-# =========================================================
-# VERSION 3.1.0 - COMMERCIAL WORKSHOP DISPATCHER
-# APPLICATION ENTRYPOINT: run_app.py
-# FIX: LOG LIVE TERMINAL, SELECT DROPDOWNS, COUNTERS & CANCEL ENGINE
-# AUTHOR: Abu Alone © 2026
-# =========================================================
 
 import os
 import sys
-import asyncio
+import time
+import shutil
 import subprocess
 from uuid import uuid4
 from loguru import logger
-from fastapi import FastAPI, Request, Form, HTTPException
+from fastapi import FastAPI, Request, Form, HTTPException, BackgroundTasks
 from fastapi.responses import HTMLResponse, JSONResponse
 from fastapi.staticfiles import StaticFiles
 from fastapi.middleware.cors import CORSMiddleware
 
-# Giữ nguyên vẹn cấu trúc lõi của ông từ ứng dụng chính
+# Import các logic cốt lõi tạo video (Giữ nguyên vẹn 100% cấu trúc của ông)
 from app.config import config
 from app.models.schema import VideoParams, VideoAspect, VideoConcatMode 
+from app.services import voice
 from app.services import task as tm
 from app.utils import utils
 
-# Nhúng module điều phối bảo mật vừa đồng bộ ở trên
+# Đồng bộ hóa chặt chẽ với module quản lý luồng bảo mật
 import licensing_client
 
-app = FastAPI(title="AI Video Engine Production Hub")
+app = FastAPI(title="AI Video Engine Commercial")
 
 app.add_middleware(
     CORSMiddleware,
@@ -35,41 +31,42 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
+# --- 1. SECRETS & CLOUDFLARE DEFAULTS ---
+PEXELS_KEY = os.getenv("PEXELS_KEY", "").strip()
+CF_TOKEN = os.getenv("CF_API_TOKEN", "").strip()
+CF_ID = os.getenv("CF_ACCOUNT_ID", "").strip()
+
+if PEXELS_KEY:
+    config.app["pexels_api_keys"] = [PEXELS_KEY]
+if CF_TOKEN and CF_ID:
+    config.app["llm_provider"] = "cloudflare"
+    config.app["cloudflare_account_id"] = CF_ID
+    config.app["cloudflare_api_key"] = CF_TOKEN
+    config.app["cloudflare_model_name"] = "@cf/meta/llama-3-8b-instruct"
+
 OUTPUT_DIR = utils.storage_dir("videos", True)
 if os.path.exists(OUTPUT_DIR):
     app.mount("/static_videos", StaticFiles(directory=OUTPUT_DIR), name="static_videos")
 
-# Cấu trúc lưu trữ quản lý tín hiệu dừng (Stop Signal) và Log trực tiếp cho Web Terminal
-ACTIVE_TASKS_SIGNALS = {}
-TASK_LIVE_LOGS = {}
-
-def write_live_log(task_id: str, msg: str):
-    if task_id not in TASK_LIVE_LOGS:
-        TASK_LIVE_LOGS[task_id] = []
-    log_line = f"[{time.strftime('%H:%M:%S')}] {msg}"
-    TASK_LIVE_LOGS[task_id].append(log_line)
-    logger.info(f"Task {task_id[:6]}: {msg}")
-
 def get_voice_list():
     path = os.path.join(os.path.dirname(__file__), "voice-list.txt")
     if os.path.exists(path):
         try:
             with open(path, "r", encoding="utf-8") as f:
-                return [line.split("Name:")[-1].strip() for line in f if "Name:" in line]
-        except Exception:
-            pass
+                voices = [line.split("Name:")[-1].strip() for line in f if "Name:" in line]
+                if voices: return voices
+        except Exception as e:
+            logger.error(f"Error reading voice list: {e}")
     return ["en-US-AvaNeural", "vi-VN-HoaiMyNeural"]
 
-# --- API ĐỒNG BỘ TRẠNG THÁI LUỒNG VÀ CHÈN KEY MỖI GIÂY ---
-import time
+# --- API CẬP NHẬT TRẠNG THÁI LUỒNG THỜI GIAN THỰC ĐỒNG BỘ MỖI GIÂY ---
 @app.get("/api/system-status")
-async def get_system_status(key: str = "", device_id: str = "MOBILE_DEFAULT_NODE"):
+async def get_system_status(key: str = "", request: Request = None):
+    device_id = request.client.host if request else "MOBILE_DEFAULT_NODE"
     try:
-        # 1. Gọi bộ dữ liệu luồng tổng hợp từ module nâng cao mới để lấy trạng thái thực tế
         thread_data = licensing_client.get_thread_status_json()
-        active_str = thread_data["busy_channels"]  # Bốc ra chuỗi định dạng dạng "0/6" hoặc "1/6"
+        active_str = thread_data["busy_channels"]  # Trả về chuỗi động thực tế ví dụ "2/6"
         
-        # 2. Thực hiện đối soát phân tầng bảo mật từ Secret hoặc Server công khai
         is_valid, key_info = licensing_client.verify_and_get_license_info(key.strip(), device_id)
         
         if is_valid:
@@ -89,129 +86,142 @@ async def get_system_status(key: str = "", device_id: str = "MOBILE_DEFAULT_NODE
         else:
             return {
                 "thread_string": active_str,
-                "key_info": {
-                    "status": "failed",
-                    "type": "free",
-                    "show_test_panel": False
-                }
-            }
-            
-    except Exception as sys_err:
-        # Biên chặn an toàn chống sập Container: Nếu có lỗi phát sinh, tự động trả về trạng thái Free Tier để chạy tiếp
-        return {
-            "thread_string": "0/6",
-            "key_info": {
-                "status": "failed",
-                "type": "free",
-                "show_test_panel": False,
-                "error_debug": str(sys_err)
+                "key_info": { "status": "failed", "type": "free", "show_test_panel": False }
             }
-        }
-
-
-# --- API XUẤT TERMINAL LOGS TỰ ĐỘNG CUỘN ---
-@app.get("/api/logs/{task_id}")
-async def get_task_logs(task_id: str):
-    logs = TASK_LIVE_LOGS.get(task_id, ["Pipeline standing by..."])
-    return {"logs": "\n".join(logs)}
+    except Exception:
+        return { "thread_string": "0/6", "key_info": { "status": "failed", "type": "free", "show_test_panel": False } }
 
-# --- API NGẮT TIẾN TRÌNH TẠO VIDEO LẬP TỨC (NÚT STOP) ---
-@app.post("/api/stop-task/{task_id}")
-async def stop_task_execution(task_id: str):
-    ACTIVE_TASKS_SIGNALS[task_id] = "STOP_SIGNALED"
-    write_live_log(task_id, "⚠️ ABORT SIGNAL: Process interruption requested by user. Killing task pipeline...")
-    return {"status": "terminated"}
-
-# --- API GENERATE VIDEO CHÍNH THỨC ---
+# --- API NGẮT TIẾN TRÌNH KHẨN CẤP KHI NHẤN NÚT STOP ---
+@app.post("/api/cancel-task")
+async def cancel_task(license_key: str = Form(""), request: Request = None):
+    device_id = request.client.host if request else "MOBILE_NODE_2026"
+    token = license_key.strip()
+    try:
+        # Tra cứu xem thiết bị hoặc key này đang giữ slot nào để giải phóng khẩn cấp
+        released = licensing_client.force_abort_user_session(token, device_id)
+        if released:
+            return {"status": "success", "msg": "Task rendering forcefully terminated."}
+        return {"status": "error", "msg": "No active tasks found for this session."}
+    except Exception as e:
+        return {"status": "error", "msg": str(e)}
+
+# --- 3. CORE PROCESSING INTERFACE WITH WATERMARK & LIMITS ---
 @app.post("/api/generate")
-async def generate_video_api(
+async def api_generate(
     request: Request,
     video_script: str = Form(...),
     clip_duration: int = Form(10),
-    voice_speed: str = Form("1.0"),
-    voice_engine: str = Form("af-ZA-AdriNeural"),
-    render_subtitles: bool = Form(True),
-    background_music: bool = Form(True),
+    voice_rate: float = Form(1.0),
+    selected_voice: str = Form("en-US-AvaNeural"),
+    enable_subtitles: bool = Form(True),
+    enable_bgm: bool = Form(True),
     license_key: str = Form("")
 ):
-    # 1. Khởi tạo hoặc định danh Node thiết bị di động dựa trên IP để kiểm soát thiết bị Free
-    device_id = request.client.host if request.client else "MOBILE_NODE_2026"
+    if not video_script.strip():
+        return JSONResponse(status_code=400, content={"status": "error", "msg": "Please enter your script before generating!"})
     
-    # Chuẩn hóa chuỗi Key nhập vào (loại bỏ khoảng trắng thừa)
+    # -------------------------------------------------------------------------
+    # LAYER 1: KIỂM TRA ĐIỀU PHỐI LUỒNG & PHÂN TẦNG KEY (CHẠY NGOÀI LOGIC TẠO VIDEO)
+    # -------------------------------------------------------------------------
     token = license_key.strip()
-
-    # 2. Thẩm định phân tầng quyền lực của Key (Admin / VIP / Commercial / Free)
-    is_key_valid, key_info = licensing_client.verify_and_get_license_info(token, device_id)
-    
-    # Xác định tài khoản có phải phân lớp Premium (VIP hoặc ADMIN) hay không
-    is_vip = (key_info.get("tier") in ["VIP", "ADMIN"]) if is_key_valid else False
+    device_id = request.client.host if request.client else "MOBILE_NODE_2026"
     
-    # Kiểm tra đặc quyền bypass (bỏ qua hoàn toàn giới hạn) đối với Admin Key hoặc Promo Key từ Secret
+    is_key_valid, key_info = licensing_client.verify_and_get_license_info(token, device_id)
+    is_vip_key = (key_info.get("tier") in ["VIP", "ADMIN"]) if is_key_valid else False
     bypass_limits = key_info.get("bypass_limits", False)
-
-    # 3. Áp dụng biên chặn hạn mức tạo video trong ngày (Nếu không có quyền đặc quyền bypass)
+    
     if not bypass_limits:
-        allowed, limit_res = licensing_client.check_generation_limits(token, device_id, is_vip)
+        allowed, limit_res = licensing_client.check_generation_limits(token, device_id, is_vip_key)
         if not allowed:
-            # Trả thông báo lỗi hạn mức trực tiếp lên khung tương tác UI
-            return {"status": "failed", "msg": limit_res}
-
-    # 4. Kích hoạt bộ điều phối: Cấp phát và xếp hàng vào 1 trong 6 luồng động của Server Cluster
-    success_alloc, slot_or_err = licensing_client.allocate_render_thread(token, device_id, is_vip)
+            return JSONResponse(status_code=400, content={"status": "error", "msg": limit_res})
+            
+    success_alloc, slot_or_err = licensing_client.allocate_render_thread(token, device_id, is_vip_key)
     if not success_alloc:
-        # Trường hợp cả 6 kênh render đều đạt đỉnh tải, báo bận để người dùng chờ
-        return {"status": "failed", "msg": slot_or_err}
+        return JSONResponse(status_code=400, content={"status": "error", "msg": slot_or_err})
         
     allocated_slot = slot_or_err
-    task_id = f"TASK-{int(time.time())}-SLOT{allocated_slot}"
+    current_pid = os.getpid()
+    licensing_client.register_process_to_slot(allocated_slot, token, device_id, is_vip_key, current_pid)
+    
+    # -------------------------------------------------------------------------
+    # LAYER 2: TIẾN TRÌNH TẠO VIDEO GỐC NGUYÊN BẢN 100% CỦA ÔNG (XỬ LÝ THỰC TẾ)
+    # -------------------------------------------------------------------------
+    task_id = str(uuid4())
+    params = VideoParams(
+        video_subject=video_script[:30].strip(),
+        video_script=video_script,
+        video_aspect=VideoAspect.portrait, 
+        video_concat_mode=VideoConcatMode.random,
+        video_clip_duration=clip_duration,
+        voice_name=selected_voice,
+        voice_rate=voice_rate,
+        subtitle_enabled=enable_subtitles,
+        bgm_type="random" if enable_bgm else "",
+        n_threads=2
+    )
     
     try:
-        # 5. Đăng ký tiến trình vào sơ đồ luồng hệ thống
-        # Sử dụng PID của chính worker hiện tại để ghim quyền kiểm soát tài nguyên
-        current_pid = os.getpid()
-        licensing_client.register_process_to_slot(allocated_slot, token, device_id, is_vip, current_pid)
+        logger.info(f"Starting Video Pipeline for task {task_id}...")
+        result = tm.start(task_id=task_id, params=params)
         
-        logger.info(f"🚀 [PIPELINE INITIALIZED] - Task: {task_id} assigned to Slot {allocated_slot} (VIP: {is_vip})")
-
-        # =========================================================================
-        # 🔥 KHU VỰC LÕI: GIỮ NGUYÊN 100% CẤU TRÚC LOGIC TẠO VIDEO VÀ THAM SỐ GỐC CỦA ÔNG
-        # Tuyệt đối không có bất kỳ sự thay đổi nào vào hệ thống sinh phim phía dưới này
-        # =========================================================================
-        
-        # [Đoạn code kết nối sang file main.py hoặc tiến trình chạy render video của ông]
-        # Ví dụ các biến đầu vào của ông đã sẵn sàng để nạp:
-        # video_script, clip_duration, voice_speed, voice_engine, render_subtitles, background_music
-        
-        # Giả lập thời gian xử lý render hệ thống (Đoạn này sẽ chạy code tạo video thực tế của ông)
-        # -------------------------------------------------------------------------
-        
-        # -------------------------------------------------------------------------
-
-        # 6. GHI NHẬN THÀNH CÔNG: Chỉ khi video đã xuất xưởng 100%, mới tiến hành trừ lượt cắn hạn mức
-        # Nếu tài khoản Free bị VIP đẩy tiến trình giữa chừng -> sập hàm -> không bị trừ lượt oan
-        licensing_client.commit_generation_success(token, device_id, is_vip)
+        if result and "videos" in result and len(result["videos"]) > 0:
+            video_path = result["videos"][0]
+            if os.path.exists(video_path):
+                filename = os.path.basename(video_path)
+                
+                # CHÈN WATERMARK ĐỘNG NẾU KHÔNG CÓ KEY HỢP LỆ (FREE TIER TRỪ VIP/ADMIN)
+                if not is_vip_key:
+                    wm_filename = f"wm_{filename}"
+                    watermarked_path = os.path.join(OUTPUT_DIR, wm_filename)
+                    
+                    drawtext_filter = "drawtext=text='Hugging/AbuAlone09':x='mod(t*35,w)':y='mod(t*15,h)':fontsize=24:fontcolor=white@0.35"
+                    ffmpeg_cmd = [
+                        'ffmpeg', '-y', 
+                        '-i', video_path,
+                        '-vf', drawtext_filter,
+                        '-c:v', 'libx264', 
+                        '-pix_fmt', 'yuv420p', 
+                        '-c:a', 'copy', 
+                        watermarked_path
+                    ]
+                    
+                    logger.info(f"Executing Mobile-Compatible FFmpeg Pipeline...")
+                    subprocess.run(ffmpeg_cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+                    
+                    if os.path.exists(watermarked_path):
+                        filename = wm_filename
+                        logger.info(f"Watermark appended successfully: {filename}")
+                    else:
+                        logger.warning("Watermark engine execution error, fallback to baseline stream.")
+
+                # Ghi nhận cắn trừ lượt tạo thành công
+                licensing_client.commit_generation_success(token, device_id, is_vip_key)
+
+                return {
+                    "status": "success", 
+                    "msg": "Video generated successfully!",
+                    "video_url": f"/static_videos/{filename}"
+                }
         
-        # Trả đường dẫn file video đã dựng hoàn thiện về cho ô trình chiếu VideoPlayer trên WebUI
-        return {
-            "status": "success",
-            "task_id": task_id,
-            "video_url": "/outputs/generated_production_video.mp4" # Thay bằng biến chứa url video thực tế của ông
-        }
-
-    except Exception as pipeline_err:
-        logger.error(f"❌ [CRITICAL PIPELINE FAILURE] - Task {task_id} crashed: {str(pipeline_err)}")
-        return {"status": "failed", "msg": f"Pipeline internal crash: {str(pipeline_err)}"}
+        return JSONResponse(status_code=500, content={"status": "error", "msg": "Render completed but output file not found."})
         
+    except Exception as e:
+        logger.error(f"Execution Error: {str(e)}")
+        return JSONResponse(status_code=500, content={"status": "error", "msg": f"Render Error: {str(e)}"})
     finally:
-        # 7. GIẢI PHÓNG SLOT: Bắt buộc phải giải phóng kênh dù tiến trình render thành công hay thất bại
-        # Chống nghẽn mạch hệ thống, đưa Slot về trạng thái 'idle' chuẩn bị cho lượt tạo kế tiếp
         licensing_client.release_thread_slot(allocated_slot)
-        logger.info(f"♻️ [RESOURCE RELEASED] - Slot {allocated_slot} has been reset to idle state.")
 
-# --- RENDER GIAO DIỆN PREMIUM GLASSMORPHISM TIẾNG ANH ---
+# --- 4. GIAO DIỆN CHÍNH (ĐÃ THAY ĐỔI TOÀN BỘ SANG TIẾNG ANH THEO DÀN Ý CHI TIẾT) ---
 @app.get("/", response_class=HTMLResponse)
 async def index_page():
     voices_options = "".join([f'<option value="{v}">{v}</option>' for v in get_voice_list()])
+    
+    # Sinh khay lựa chọn động cho Clip Duration từ 5s đến 10s (Mặc định 10s)
+    duration_select = "".join([f'<option value="{i}" {"selected" if i==10 else ""}>{i} Seconds</option>' for i in range(5, 11)])
+    
+    # Sinh khay lựa chọn động cho Voice Speed từ 0.5 đến 2.0 bước nhảy 0.1 (Mặc định 1.0)
+    speeds = [round(0.5 + x * 0.1, 1) for x in range(16)]
+    speed_select = "".join([f'<option value="{s}" {"selected" if s==1.0 else ""}>{s}x Speed</option>' for s in speeds])
     html_content = f"""
 <!DOCTYPE html>
 <html lang="en">