Initial clean project import

2026-06-19 18:41:41 +08:00
commit a13b804c7a
1306 changed files with 220568 additions and 0 deletions
@@ -0,0 +1,71 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""
+测试完整字幕和封面生成
+"""
+
+import os
+import json
+import sys
+
+# 配置信息
+CONFIG = {
+    "videoPath": r"C:\Users\Administrator\Desktop\视频生成\对比结果\121212.mp4",
+    "scriptText": "今天天气真好！\n明天下雨。\n我们去公园玩吧。\n这个电影太好看了。\n质量不好，我很失望。",
+    "subtitleStyle": "emotion-mixed",  # 使用情感混合模式
+    "coverStyle": "default",
+    "outputDir": r"C:\Users\Administrator\AppData\Roaming\aigcpanel\data\temp\test_full_generation",
+    "useFunasr": False,
+    "coverOnly": False
+}
+
+def main():
+    print("=" * 60)
+    print("测试完整字幕和封面生成")
+    print("=" * 60)
+
+    # 确保输出目录存在
+    os.makedirs(CONFIG["outputDir"], exist_ok=True)
+
+    # 保存配置文件
+    config_path = os.path.join(CONFIG["outputDir"], "config.json")
+    with open(config_path, 'w', encoding='utf-8') as f:
+        json.dump(CONFIG, f, ensure_ascii=False, indent=2)
+
+    print(f"\n配置文件已保存: {config_path}")
+    print("\n开始生成...\n")
+
+    # 调用字幕生成器
+    from subtitle_cover_generator_simple import SubtitleCoverGenerator
+
+    generator = SubtitleCoverGenerator(CONFIG)
+    result = generator.generate_all()
+
+    print("\n" + "=" * 60)
+    print("生成结果:")
+    print("=" * 60)
+    print(json.dumps(result, ensure_ascii=False, indent=2))
+
+    if result['success']:
+        print("\n✓ 生成成功!")
+        print(f"\n封面路径: {result.get('coverPath')}")
+        print(f"视频路径: {result.get('videoPath')}")
+        print(f"\nNLP分析结果:")
+        print(f"  标题: {result['nlpAnalysis'].get('title')}")
+        print(f"  关键词: {result['nlpAnalysis'].get('keywords')}")
+        print(f"  情感: {result['nlpAnalysis'].get('sentiment')} (分数: {result['nlpAnalysis'].get('sentiment_score', 0):.4f})")
+        print(f"\n字幕数量: {len(result.get('subtitles', []))}")
+
+        # 打印每个字幕的情感和样式
+        print("\n字幕详情:")
+        for i, subtitle in enumerate(result.get('subtitles', []), 1):
+            print(f"  {i}. {subtitle['text']}")
+            print(f"     情感: {subtitle['sentiment']} (分数: {subtitle['sentiment_score']:.4f}) -> 样式: {subtitle['style']}")
+
+        return 0
+    else:
+        print(f"\n✗ 生成失败: {result.get('error')}")
+        return 1
+
+if __name__ == '__main__':
+    sys.exit(main())