free-short-video 5.2.1 → 5.3.1

package/bin/cli.js

@@ -21,6 +21,27 @@ const os = require('os');
 
 const PKG_ROOT = path.resolve(__dirname, '..');
 
+// 终端超链接（ANSI OSC 8）：TTY 下输出可点击链接，非 TTY（管道/重定向）降级为纯文本
+const link = (url, text) =>
+  process.stdout.isTTY ? `\x1b]8;;${url}\x1b\\${text}\x1b]8;;\x1b\\` : text;
+
+// 启动图案（ASCII art）：TTY 下亮青色显示，非 TTY 纯文本
+const FSV_LOGO = String.raw`   _____ ______     __
+  |  ___/ ___\ \   / /
+  | |_  \___ \\ \ / /
+  |  _|  ___) |\ V /
+  |_|   |____/  \_/`.split('\n');
+function printLogo() {
+  if (process.stdout.isTTY) {
+    console.log('\x1b[1;36m');
+    FSV_LOGO.forEach((l) => console.log(l));
+    process.stdout.write('\x1b[0m');
+  } else {
+    console.log('');
+    FSV_LOGO.forEach((l) => console.log(l));
+  }
+}
+
 function printHelp() {
   console.log(`free-short-video — free AI short-video generator
 
@@ -37,6 +58,9 @@ Options:
 Environment:
   AGNES_API_KEY  Your Agnes API key (can also be set later in the Web UI)
 
+Website:
+  ${link('https://video.lichuanyang.top', 'https://video.lichuanyang.top')}   (official site & no-install online demo)
+
 First run creates a local venv and installs Python dependencies automatically.
 `);
 }
@@ -96,7 +120,10 @@ function venvBin(name) {
 
 // ── 1. python check ───────────────────────────────────────────────────────
 console.log('================================================');
-console.log('  free-short-video');
+printLogo();
+console.log('   free-short-video — Free AI short-video generator');
+console.log('');
+console.log(`   🌐 官网: ${link('https://video.lichuanyang.top', 'https://video.lichuanyang.top')}  |  ⚡ 在线体验: ${link('https://video.lichuanyang.top/demo', 'https://video.lichuanyang.top/demo')}`);
 console.log('================================================');
 console.log('');
 

package/core/api/agnes_image.py

@@ -17,6 +17,9 @@ from utils.image import download_image
 
 logger = logging.getLogger(__name__)
 
+# 图像生成读超时基数（秒）：timeout = 基数 * (attempt + 1)，首次 120s 逐步放宽
+_READ_TIMEOUT_BASE_SECONDS = 120
+
 
 class ImageOutput:
     def __init__(self, fmt: str, ext: str, data: str):
@@ -115,7 +118,7 @@ class AgnesImageAPI:
                 # 全局限速：在发起 HTTP 请求前获取令牌
                 await asyncio.to_thread(get_rate_limiter().acquire)
                 # 动态超时：第一次 120s，后续逐步增加（图像生成较慢，放宽读超时）
-                read_timeout = 120 * (attempt + 1)
+                read_timeout = _READ_TIMEOUT_BASE_SECONDS * (attempt + 1)
                 resp = await asyncio.to_thread(
                     requests.post,
                     f"{get_agnes_base_url()}/images/generations",

package/core/api/agnes_video.py

@@ -26,6 +26,9 @@ DURATION_PRESETS = {
     20: (409, 24),   # capped at 409 (API max for 720p); actual ~17s
 }
 
+# 图片上传 429 退避间隔基数（秒）：delay = 基数 * (attempt + 1)
+_UPLOAD_RETRY_BASE_DELAY_SECONDS = 30
+
 
 class VideoOutput:
     def __init__(self, fmt: str, ext: str, data: str):
@@ -144,7 +147,7 @@ class AgnesVideoAPI:
                     timeout=(30, 120),
                 )
                 if resp.status_code == 429:
-                    delay = 30 * (attempt + 1)
+                    delay = _UPLOAD_RETRY_BASE_DELAY_SECONDS * (attempt + 1)
                     logger.warning(f"[AgnesVideo] Image upload 429, retry in {delay}s...")
                     await asyncio.sleep(delay)
                     continue

package/core/artifacts.py

@@ -12,7 +12,11 @@ core/artifacts.py — 中间产物注册表与级联删除计划
 
 from __future__ import annotations
 
+import json
+import logging
 import os
+import shutil
+import time
 from dataclasses import dataclass, field
 from typing import Any, Optional
 
@@ -572,3 +576,82 @@ def apply_cascade_plan(state: BaseTaskState, plan: CascadePlan) -> dict:
     update_kwargs["status"] = StepStatus.PENDING
 
     return update_kwargs
+
+
+# ═══════════════════════════════════════════════════════════════
+# 僵尸任务磁盘清理（v5.0 Batch 5 / 5.1）
+# ═══════════════════════════════════════════════════════════════
+
+_DEFAULT_PROTECT_STATUSES = {StepStatus.RUNNING, StepStatus.QUEUED, StepStatus.PENDING}
+
+
+def sweep_stale_tasks(age_days: int = 7,
+                      protect_statuses: Optional[set] = None) -> dict:
+    """扫描并清理僵尸任务目录（状态文件超龄且任务非活跃）。
+
+    清理条件（全部满足才删除）：
+    1. 工作区中存在 ``task_state.json`` 且可解析出 ``status`` 字段；
+    2. ``status`` 不在保护集合中；
+    3. ``task_state.json`` 的修改时间距今超过 ``age_days`` 天。
+
+    默认保护 ``RUNNING``（运行中）/ ``QUEUED``（排队中）/ ``PENDING``（断点续传
+    候选），保证活跃任务与可恢复任务永不误删；调用方可传 ``protect_statuses``
+    显式覆盖（如仅保护活跃状态，放开 PENDING）。
+
+    删除策略：整目录 ``shutil.rmtree``（等价于按级联计划删除全部产物 + 状态文件），
+    删除前校验 realpath 必须位于工作区内，防符号链接逃逸。
+
+    Args:
+        age_days: 状态文件超龄阈值（天）
+        protect_statuses: 保护的状态集合；None 时使用默认保护集合
+
+    Returns:
+        {"swept": [目录名], "protected": [目录名], "errors": [描述]}
+    """
+    working_dir = get_working_dir()
+    protect = protect_statuses if protect_statuses is not None else _DEFAULT_PROTECT_STATUSES
+    swept: list[str] = []
+    protected: list[str] = []
+    errors: list[str] = []
+
+    if not os.path.isdir(working_dir):
+        return {"swept": swept, "protected": protected, "errors": errors}
+
+    real_working = os.path.realpath(working_dir)
+    cutoff = time.time() - age_days * 86400
+
+    for name in sorted(os.listdir(working_dir)):
+        task_dir = os.path.join(working_dir, name)
+        if not os.path.isdir(task_dir):
+            continue
+        task_file = os.path.join(task_dir, "task_state.json")
+        if not os.path.isfile(task_file):
+            continue  # 非任务目录（如 uploads/ 或普通文件目录）
+        try:
+            with open(task_file, "r", encoding="utf-8") as f:
+                status_str = json.load(f).get("status")
+        except (OSError, json.JSONDecodeError):
+            errors.append(f"{name}: task_state.json 无法解析")
+            continue
+        if status_str in {s.value for s in protect}:
+            protected.append(name)
+            continue
+        try:
+            mtime = os.path.getmtime(task_file)
+        except OSError:
+            errors.append(f"{name}: 无法读取 task_state.json 修改时间")
+            continue
+        if mtime > cutoff:
+            protected.append(name)  # 未超龄，保留
+            continue
+        # 路径穿越防护：任务目录 realpath 必须位于工作区内
+        if not os.path.realpath(task_dir).startswith(real_working + os.sep):
+            errors.append(f"{name}: 任务目录逃逸工作区，拒绝删除")
+            continue
+        try:
+            shutil.rmtree(task_dir)
+            swept.append(name)
+        except OSError as e:
+            errors.append(f"{name}: 删除失败 ({e})")
+
+    return {"swept": swept, "protected": protected, "errors": errors}

package/core/audio/subtitle/__init__.py

@@ -0,0 +1,35 @@
+"""core.audio.subtitle — SRT 字幕生成 + moviepy 叠加（v5.0 Batch 4 / 4.3 拆分后的包入口）
+
+SubtitleGenerator 由两个职责 mixin（SubtitleSrtMixin / SubtitleRenderMixin）
+组合而成；对外保留全部原符号（subtitle.py 兼容 re-export）。"""
+
+import logging
+
+from .generator import SubtitleSrtMixin
+from .renderer import SubtitleRenderMixin
+
+logger = logging.getLogger(__name__)
+
+
+class SubtitleGenerator(
+    SubtitleSrtMixin,
+    SubtitleRenderMixin,
+):
+
+    """字幕生成器：cues → SRT + moviepy 叠加。"""
+
+
+from . import generator as srt_generator
+from . import renderer as subtitle_renderer
+
+# 运行时注入：mixin 方法内显式 `SubtitleGenerator.xxx` 自引用
+# （原 @staticmethod 风格）在模块级延迟解析到组合类。
+srt_generator.SubtitleGenerator = SubtitleGenerator
+subtitle_renderer.SubtitleGenerator = SubtitleGenerator
+
+__all__ = [
+    "SubtitleGenerator",
+    "SubtitleSrtMixin",
+    "SubtitleRenderMixin",
+]
+