6 |
29 | )
30 | }
31 |
32 |
--------------------------------------------------------------------------------
/backend/start.sh:
--------------------------------------------------------------------------------
1 | #!/bin/bash
2 |
3 | # 检查 Python 环境
4 | if ! command -v python3 &> /dev/null; then
5 | echo "❌ 错误: 未找到 Python3，请先安装 Python 3.8+"
6 | exit 1
7 | fi
8 |
9 | # 检查 Python 版本
10 | PYTHON_VERSION=$(python3 --version 2>&1 | awk '{print $2}')
11 | echo "✓ Python 版本: $PYTHON_VERSION"
12 |
13 | # 检查虚拟环境
14 | if [ ! -d "venv" ]; then
15 | echo "📦 创建虚拟环境..."
16 | python3 -m venv venv
17 | if [ $? -ne 0 ]; then
18 | echo "❌ 虚拟环境创建失败"
19 | exit 1
20 | fi
21 | echo "✓ 虚拟环境创建成功"
22 | else
23 | echo "✓ 虚拟环境已存在"
24 | fi
25 |
26 | # 激活虚拟环境
27 | echo "🔧 激活虚拟环境..."
28 | source venv/bin/activate
29 |
30 | # 升级 pip
31 | echo "⬆️ 升级 pip..."
32 | pip install --upgrade pip -q
33 |
34 | # 安装依赖
35 | echo "📥 安装依赖包..."
36 | pip install -r requirements.txt
37 | if [ $? -ne 0 ]; then
38 | echo "❌ 依赖安装失败"
39 | exit 1
40 | fi
41 | echo "✓ 依赖安装完成"
42 |
43 | # 检查 .env 文件
44 | if [ ! -f ".env" ]; then
45 | echo "⚠️ 警告: 未找到 .env 文件"
46 | echo " 请复制 .env.example 为 .env 并配置环境变量:"
47 | echo " cp .env.example .env"
48 | echo " 然后编辑 .env 文件，填入你的 OPENAI_API_KEY"
49 | echo ""
50 | read -p "是否继续启动? (y/n) " -n 1 -r
51 | echo
52 | if [[ ! $REPLY =~ ^[Yy]$ ]]; then
53 | exit 1
54 | fi
55 | else
56 | echo "✓ 环境变量文件已配置"
57 | fi
58 |
59 | # 启动服务
60 | echo ""
61 | echo "🚀 启动后端服务..."
62 | echo " 访问地址: http://localhost:8483"
63 | echo " API 文档: http://localhost:8483/docs"
64 | echo ""
65 | python main.py
66 |
67 |
--------------------------------------------------------------------------------
/backend/start.bat:
--------------------------------------------------------------------------------
1 | @echo off
2 | chcp 65001 >nul
3 |
4 | REM 检查 Python 环境
5 | python --version >nul 2>&1
6 | if errorlevel 1 (
7 | echo ❌ 错误: 未找到 Python，请先安装 Python 3.8+
8 | pause
9 | exit /b 1
10 | )
11 |
12 | python --version
13 | echo ✓ Python 环境检查通过
14 | echo.
15 |
16 | REM 检查虚拟环境
17 | if not exist "venv" (
18 | echo 📦 创建虚拟环境...
19 | python -m venv venv
20 | if errorlevel 1 (
21 | echo ❌ 虚拟环境创建失败
22 | pause
23 | exit /b 1
24 | )
25 | echo ✓ 虚拟环境创建成功
26 | ) else (
27 | echo ✓ 虚拟环境已存在
28 | )
29 |
30 | REM 激活虚拟环境
31 | echo 🔧 激活虚拟环境...
32 | call venv\Scripts\activate.bat
33 |
34 | REM 升级 pip
35 | echo ⬆️ 升级 pip...
36 | python -m pip install --upgrade pip -q
37 |
38 | REM 安装依赖
39 | echo 📥 安装依赖包...
40 | pip install -r requirements.txt
41 | if errorlevel 1 (
42 | echo ❌ 依赖安装失败
43 | pause
44 | exit /b 1
45 | )
46 | echo ✓ 依赖安装完成
47 | echo.
48 |
49 | REM 检查 .env 文件
50 | if not exist ".env" (
51 | echo ⚠️ 警告: 未找到 .env 文件
52 | echo 请复制 .env.example 为 .env 并配置环境变量:
53 | echo copy .env.example .env
54 | echo 然后编辑 .env 文件，填入你的 OPENAI_API_KEY
55 | echo.
56 | set /p continue="是否继续启动? (y/n): "
57 | if /i not "%continue%"=="y" (
58 | exit /b 1
59 | )
60 | ) else (
61 | echo ✓ 环境变量文件已配置
62 | )
63 |
64 | REM 启动服务
65 | echo.
66 | echo 🚀 启动后端服务...
67 | echo 访问地址: http://localhost:8483
68 | echo API 文档: http://localhost:8483/docs
69 | echo.
70 | python main.py
71 |
72 | pause
73 |
74 |
--------------------------------------------------------------------------------
/backend/app/utils/video_helper.py:
--------------------------------------------------------------------------------
1 | import subprocess
2 | import os
3 | import uuid
4 | from pathlib import Path
5 | from typing import Optional
6 | from app.utils.logger import get_logger
7 | from app.utils.ffmpeg_helper import get_ffmpeg_path
8 |
9 | logger = get_logger(__name__)
10 |
11 |
12 | def generate_screenshot(video_path: str, output_dir: str, timestamp: int, index: int) -> str:
13 | """
14 | 使用 ffmpeg 生成截图，返回生成图片路径
15 |
16 | :param video_path: 视频文件路径
17 | :param output_dir: 输出目录
18 | :param timestamp: 时间戳（秒）
19 | :param index: 截图索引
20 | :return: 生成的截图文件路径
21 | """
22 | output_dir = Path(output_dir)
23 | output_dir.mkdir(parents=True, exist_ok=True)
24 |
25 | filename = f"screenshot_{index:03d}_{uuid.uuid4().hex[:8]}.jpg"
26 | output_path = output_dir / filename
27 |
28 | ffmpeg_path = get_ffmpeg_path()
29 | command = [
30 | ffmpeg_path,
31 | "-ss", str(timestamp),
32 | "-i", str(video_path),
33 | "-frames:v", "1",
34 | "-q:v", "2", # 高质量
35 | "-y", # 覆盖已存在文件
36 | str(output_path)
37 | ]
38 |
39 | try:
40 | result = subprocess.run(
41 | command,
42 | capture_output=True,
43 | text=True,
44 | check=True
45 | )
46 | logger.info(f"截图生成成功: {output_path} (时间戳: {timestamp}秒)")
47 | return str(output_path)
48 | except subprocess.CalledProcessError as e:
49 | logger.error(f"生成截图失败: {e.stderr}")
50 | raise Exception(f"生成截图失败: {e.stderr}")
51 |
52 |
53 |
54 |
--------------------------------------------------------------------------------
/frontend/src/store/taskStore.ts:
--------------------------------------------------------------------------------
1 | import { create } from 'zustand'
2 |
3 | export interface Task {
4 | id: string
5 | filename: string
6 | status: 'pending' | 'processing' | 'transcribing' | 'summarizing' | 'completed' | 'failed'
7 | markdown?: string
8 | createdAt?: string
9 | }
10 |
11 | interface TaskStore {
12 | tasks: Task[]
13 | currentTaskId: string | null
14 | addTask: (task: Task) => void
15 | updateTask: (id: string, updates: Partial
7 |
28 |
8 |
11 |
12 | 上传视频/音频
9 |上传视频或音频文件，系统将自动处理并生成笔记
10 |
13 |
15 |
16 | {/* 使用提示 */}
17 |
18 |
27 | 💡 使用提示
19 |-
20 |
- • 支持 MP4, AVI, MOV, MKV, MP3, WAV 等格式 21 |
- • 文件大小限制：最大 500MB 22 |
- • 上传后系统会自动处理：提取音频 → 转写文字 → 生成笔记 23 |
- • 处理完成后可以在首页查看任务列表和笔记预览 24 |
- • 可以选择是否生成截图标记（仅视频文件） 25 |
19 |
23 | )
24 | }
25 |
26 | if (currentTask?.status === 'failed') {
27 | return (
28 |
20 |
22 | 请选择一个任务查看笔记
21 |
29 |
33 | )
34 | }
35 |
36 | if (!markdown && currentTask?.status !== 'completed') {
37 | return (
38 |
30 |
32 | 任务处理失败，请重试
31 |
39 |
43 | )
44 | }
45 |
46 | return (
47 |
40 |
42 | 正在生成笔记，请稍候...
41 |
48 |
61 | )
62 | }
63 |
64 |
--------------------------------------------------------------------------------
/frontend/src/App.tsx:
--------------------------------------------------------------------------------
1 | import { useState } from 'react'
2 | import Sidebar from './components/Sidebar'
3 | import MainContent from './components/MainContent'
4 | import CurrentModelDisplay from './components/CurrentModelDisplay'
5 | import { Toaster } from 'react-hot-toast'
6 |
7 | type MenuItem = 'home' | 'upload' | 'model' | 'settings' | 'download'
8 |
9 | function App() {
10 | const [activeMenu, setActiveMenu] = useState
49 |
54 |
55 | 笔记预览
50 | {currentTask && ( 51 |{currentTask.filename}
52 | )} 53 |
56 |
57 | {markdown}
58 |
59 |
60 |