83 lines
2.3 KiB
Python
83 lines
2.3 KiB
Python
# -*- coding: utf-8 -*-
|
|
import io
|
|
import json
|
|
import os
|
|
import sys
|
|
import clip
|
|
import getgrame
|
|
import Push_back_Prompt
|
|
import public_tools
|
|
import shotSplit
|
|
|
|
sys.stdout = io.TextIOWrapper(sys.stdout.buffer, encoding="utf-8")
|
|
|
|
# sys.argv = ["C:\\Users\\27698\\Desktop\\LAITool\\resources\\scripts\\Lai.exe","-c","D:/父爱如山倒/scripts/output_crop_00001.json"]
|
|
print(sys.argv)
|
|
|
|
if len(sys.argv) < 2:
|
|
print("Params: <runtime-config.json>")
|
|
exit(0)
|
|
|
|
if getattr(sys, "frozen", False):
|
|
cript_directory = os.path.dirname(sys.executable)
|
|
elif __file__:
|
|
cript_directory = os.path.dirname(__file__)
|
|
|
|
|
|
def set_ffmpeg_env():
|
|
# 根据你的ffmpeg路径替换这个
|
|
ffmpeg_path = os.path.join(
|
|
cript_directory, "../package/ffmpeg-2023-12-07-git-f89cff96d0-full_build/bin"
|
|
)
|
|
|
|
if sys.platform == "win32":
|
|
ffmpeg_path = ffmpeg_path.replace("/", "\\")
|
|
|
|
# 检查环境变量中是否已经设置
|
|
if "PATH" in os.environ:
|
|
path_values = os.environ["PATH"]
|
|
if ffmpeg_path not in path_values:
|
|
os.environ["PATH"] += f";{ffmpeg_path}"
|
|
else:
|
|
os.environ["PATH"] = ffmpeg_path
|
|
|
|
|
|
set_ffmpeg_env()
|
|
|
|
# 执行剪辑的方法
|
|
if sys.argv[1] == "-c":
|
|
clip = clip.Clip(cript_directory, sys.argv[2])
|
|
clip.MergeVideosAndClip()
|
|
pass
|
|
# 获取字体
|
|
elif sys.argv[1] == "-f":
|
|
# 获取本地已安装的字幕。然后返回
|
|
public_tools = public_tools.PublicTools()
|
|
font_list = public_tools.get_installed_fonts()
|
|
font_obj_list = []
|
|
for font_name in font_list:
|
|
obj = {"label": font_name, "value": font_name}
|
|
font_obj_list.append(obj)
|
|
|
|
with open(sys.argv[2], "r", encoding="utf-8") as file:
|
|
data = json.load(file)
|
|
data["font_name_list"] = font_obj_list
|
|
with open(sys.argv[2], "w", encoding="utf-8") as file:
|
|
json.dump(data, file, ensure_ascii=False, indent=4)
|
|
|
|
# 反推提示词
|
|
elif sys.argv[1] == "-p":
|
|
Push_back_Prompt.init(sys.argv[2], sys.argv[3], sys.argv[4])
|
|
pass
|
|
|
|
# 剪映抽帧
|
|
elif sys.argv[1] == "-k":
|
|
# print("")
|
|
getgrame.init(sys.argv[2], sys.argv[3], sys.argv[4])
|
|
pass
|
|
|
|
# 智能分镜。字幕识别
|
|
elif sys.argv[1] == "-a":
|
|
print("开始算法分镜：" + sys.argv[2] + " -- 输出文件夹：" + sys.argv[3])
|
|
shotSplit.init(sys.argv[2], sys.argv[3], sys.argv[4], sys.argv[5])
|