From bb89d4bc410d93f54a7a7779288c213de018fd0b Mon Sep 17 00:00:00 2001
From: "kyj@bowong.ai" <kyj@bowong.ai>
Date: Wed, 26 Mar 2025 15:48:59 +0800
Subject: [PATCH] =?UTF-8?q?FIX=20=E4=BF=AE=E5=A4=8D=E7=B2=BE=E7=A1=AE?=
 =?UTF-8?q?=E8=A3=81=E5=89=AA=E8=A7=86=E9=A2=91=E8=8A=82=E7=82=B9?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 nodes/videocut.py | 321 +++++++++++++++++++++++-----------------------
 1 file changed, 164 insertions(+), 157 deletions(-)

diff --git a/nodes/videocut.py b/nodes/videocut.py
index c2d173e..99a0157 100644
--- a/nodes/videocut.py
+++ b/nodes/videocut.py
@@ -6,7 +6,6 @@ import uuid
 from datetime import datetime
 
 import ffmpy
-import torchaudio
 import torchvision.io
 
 video_extensions = ['webm', 'mp4', 'mkv', 'gif', 'mov']
@@ -36,88 +35,92 @@ class VideoCut:
     CATEGORY = "不忘科技-自定义节点🚩"
 
     def cut(self, video_path, start, end):
-        # 原文件名
-        origin_fname = ".".join(video_path.split(os.sep)[-1].split(".")[:-1])
-        # 新文件名 复制改名适配ffmpeg
-        uid = uuid.uuid1()
-        temp_fname = os.sep.join(
-            [
-                *video_path.split(os.sep)[:-1],
-                "%s.%s" % (str(uid), video_path.split(".")[-1]),
-            ]
-        )
         try:
-            shutil.copy(video_path, temp_fname)
-        except:
-            return ("请检查输入文件权限",)
-        video_path = temp_fname
-        # 组装输出文件名
-        output_name = ".".join(
-            [
-                *video_path.split(os.sep)[-1].split(".")[:-2],
-                video_path.split(os.sep)[-1].split(".")[-2]
-                + "_output_%s" % datetime.now().strftime("%Y%m%d_%H%M%S"),
-                video_path.split(os.sep)[-1].split(".")[-1],
-            ]
-        )
-        output = (
-            os.sep.join([*video_path.split(os.sep)[:-1], output_name])
-            .replace(
-                os.sep.join(["ComfyUI", "input"]), os.sep.join(["ComfyUI", "output"])
-            )
-            .replace(" ", "")
-        )
-        # 调用ffmpeg
-        ff = ffmpy.FFmpeg(
-            inputs={video_path: None},
-            outputs={
-                output: [
-                    "-ss",
-                    start,
-                    "-to",
-                    end,
-                    "-c:v",
-                    "libx264",
-                    "-c:a",
-                    "libmp3lame",
-                    "-reset_timestamps",
-                    "1",
-                    "-sc_threshold",
-                    "0",
-                    "-g",
-                    "1"
-                    "-force_key_frames",
-                    "expr:gte(t, n_forced * 1)",
-                    "-v",
-                    "-8"
+            # 原文件名
+            origin_fname = ".".join(video_path.split(os.sep)[-1].split(".")[:-1])
+            # 新文件名 复制改名适配ffmpeg
+            uid = uuid.uuid1()
+            temp_fname = os.sep.join(
+                [
+                    *video_path.split(os.sep)[:-1],
+                    "%s.%s" % (str(uid), video_path.split(".")[-1]),
                 ]
-            },
-        )
-        print(ff.cmd)
-        ff.run()
-        # uuid填充改回原文件名
-        try:
-            os.remove(temp_fname)
-        except:
-            pass
-        try:
-            files = glob.glob(output.replace("%03d", "*"))
-            for file in files:
-                shutil.move(file, file.replace(str(uid), origin_fname))
-            files = glob.glob(
-                output.replace(str(uid), origin_fname).replace("%03d", "*")
             )
+            try:
+                shutil.copy(video_path, temp_fname)
+            except:
+                return ("请检查输入文件权限",)
+            video_path = temp_fname
+            # 组装输出文件名
+            output_name = ".".join(
+                [
+                    *video_path.split(os.sep)[-1].split(".")[:-2],
+                    video_path.split(os.sep)[-1].split(".")[-2]
+                    + "_output_%s" % datetime.now().strftime("%Y%m%d_%H%M%S"),
+                    video_path.split(os.sep)[-1].split(".")[-1],
+                ]
+            )
+            output = (
+                os.sep.join([*video_path.split(os.sep)[:-1], output_name])
+                .replace(
+                    os.sep.join(["ComfyUI", "input"]), os.sep.join(["ComfyUI", "output"])
+                )
+                .replace(" ", "")
+            )
+            # 调用ffmpeg
+            ff = ffmpy.FFmpeg(
+                inputs={video_path: None},
+                outputs={
+                    output: [
+                        "-ss",
+                        start,
+                        "-to",
+                        end,
+                        "-c:v",
+                        "libx264",
+                        "-c:a",
+                        "libmp3lame",
+                        "-reset_timestamps",
+                        "1",
+                        "-sc_threshold",
+                        "0",
+                        "-g",
+                        "1"
+                        "-force_key_frames",
+                        "expr:gte(t, n_forced * 1)",
+                        "-v",
+                        "-8"
+                    ]
+                },
+            )
+            print(ff.cmd)
+            ff.run()
+            # uuid填充改回原文件名
+            try:
+                os.remove(temp_fname)
+            except:
+                pass
+            try:
+                files = glob.glob(output.replace("%03d", "*"))
+                for file in files:
+                    shutil.move(file, file.replace(str(uid), origin_fname))
+                files = glob.glob(
+                    output.replace(str(uid), origin_fname).replace("%03d", "*")
+                )
+            except:
+                files = glob.glob(output.replace("%03d", "*"))
+                traceback.print_exc()
+            video, audio, info = torchvision.io.read_video(files[0])
+            video.mul_(255)
+            audio.unsqueeze_(0)
+            try:
+                os.remove(files[0])
+            except:
+                pass
+            return (video, {"waveform":audio,"sample_rate":info["audio_fps"]},)
         except:
-            files = glob.glob(output.replace("%03d", "*"))
             traceback.print_exc()
-        video, audio, info = torchvision.io.read_video(files[0])
-        video.mul_(255)
-        audio.unsqueeze_(0)
-        try:
-            os.remove(files[0])
-        except:
-            pass
-        return (video, {"waveform":audio,"sample_rate":info["audio_fps"]},)
+            raise Exception("Cut Failed")
 
 class VideoCutByFramePoint:
     """FFMPEG视频剪辑-帧位"""
@@ -144,87 +147,91 @@ class VideoCutByFramePoint:
     CATEGORY = "不忘科技-自定义节点🚩"
 
     def cut(self, video_path, start_point, duration, fps, force_match_fps):
-        # 原文件名
-        origin_fname = ".".join(video_path.split(os.sep)[-1].split(".")[:-1])
-        # 新文件名 复制改名适配ffmpeg
-        uid = uuid.uuid1()
-        temp_fname = os.sep.join(
-            [
-                *video_path.split(os.sep)[:-1],
-                "%s.%s" % (str(uid), video_path.split(".")[-1]),
-            ]
-        )
         try:
-            shutil.copy(video_path, temp_fname)
-        except:
-            return ("请检查输入文件权限",)
-        video_path = temp_fname
-        # 组装输出文件名
-        output_name = ".".join(
-            [
-                *video_path.split(os.sep)[-1].split(".")[:-2],
-                video_path.split(os.sep)[-1].split(".")[-2]
-                + "_output_%s" % datetime.now().strftime("%Y%m%d_%H%M%S"),
-                video_path.split(os.sep)[-1].split(".")[-1],
-            ]
-        )
-        output = (
-            os.sep.join([*video_path.split(os.sep)[:-1], output_name])
-            .replace(
-                os.sep.join(["ComfyUI", "input"]), os.sep.join(["ComfyUI", "output"])
-            )
-            .replace(" ", "")
-        )
-        # 调用ffmpeg
-        ff = ffmpy.FFmpeg(
-            inputs={video_path: None},
-            outputs={
-                output: [
-                    "-ss",
-                    "%.3f" % (start_point/fps),
-                    "-t",
-                    "%.3f" % (duration/fps),
-                    "-c:v",
-                    "libx264",
-                    "-c:a",
-                    "libmp3lame",
-                    "-reset_timestamps",
-                    "1",
-                    "-sc_threshold",
-                    "0",
-                    "-g",
-                    "1",
-                    "-force_key_frames",
-                    "expr:gte(t, n_forced * 1)",
-                    "-r" if force_match_fps else "",
-                    "%d" % fps if force_match_fps else "",
-                    "-v",
-                    "-8"
+            # 原文件名
+            origin_fname = ".".join(video_path.split(os.sep)[-1].split(".")[:-1])
+            # 新文件名 复制改名适配ffmpeg
+            uid = uuid.uuid1()
+            temp_fname = os.sep.join(
+                [
+                    *video_path.split(os.sep)[:-1],
+                    "%s.%s" % (str(uid), video_path.split(".")[-1]),
                 ]
-            },
-        )
-        print(ff.cmd)
-        ff.run()
-        # uuid填充改回原文件名
-        try:
-            os.remove(temp_fname)
-        except:
-            pass
-        try:
-            files = glob.glob(output.replace("%03d", "*"))
-            for file in files:
-                shutil.move(file, file.replace(str(uid), origin_fname))
-            files = glob.glob(
-                output.replace(str(uid), origin_fname).replace("%03d", "*")
             )
+            try:
+                shutil.copy(video_path, temp_fname)
+            except:
+                return ("请检查输入文件权限",)
+            video_path = temp_fname
+            # 组装输出文件名
+            output_name = ".".join(
+                [
+                    *video_path.split(os.sep)[-1].split(".")[:-2],
+                    video_path.split(os.sep)[-1].split(".")[-2]
+                    + "_output_%s" % datetime.now().strftime("%Y%m%d_%H%M%S"),
+                    video_path.split(os.sep)[-1].split(".")[-1],
+                ]
+            )
+            output = (
+                os.sep.join([*video_path.split(os.sep)[:-1], output_name])
+                .replace(
+                    os.sep.join(["ComfyUI", "input"]), os.sep.join(["ComfyUI", "output"])
+                )
+                .replace(" ", "")
+            )
+            # 调用ffmpeg
+            ff = ffmpy.FFmpeg(
+                inputs={video_path: None},
+                outputs={
+                    output: [
+                        "-ss",
+                        "%.3f" % (start_point/fps),
+                        "-t",
+                        "%.3f" % (duration/fps),
+                        "-c:v",
+                        "libx264",
+                        "-c:a",
+                        "libmp3lame",
+                        "-reset_timestamps",
+                        "1",
+                        "-sc_threshold",
+                        "0",
+                        "-g",
+                        "1",
+                        "-force_key_frames",
+                        "expr:gte(t, n_forced * 1)",
+                        "-r" if force_match_fps else "",
+                        "%d" % fps if force_match_fps else "",
+                        "-v",
+                        "-8"
+                    ]
+                },
+            )
+            print(ff.cmd)
+            ff.run()
+            # uuid填充改回原文件名
+            try:
+                os.remove(temp_fname)
+            except:
+                pass
+            try:
+                files = glob.glob(output.replace("%03d", "*"))
+                for file in files:
+                    shutil.move(file, file.replace(str(uid), origin_fname))
+                files = glob.glob(
+                    output.replace(str(uid), origin_fname).replace("%03d", "*")
+                )
+            except:
+                files = glob.glob(output.replace("%03d", "*"))
+                traceback.print_exc()
+            video, audio, info = torchvision.io.read_video(files[0])
+            video.mul_(255)
+            audio.unsqueeze_(0)
+            try:
+                os.remove(files[0])
+            except:
+                pass
+            return (video, {"waveform":audio,"sample_rate":info["audio_fps"]},)
         except:
-            files = glob.glob(output.replace("%03d", "*"))
             traceback.print_exc()
-        video, audio, info = torchvision.io.read_video(files[0])
-        video.mul_(255)
-        audio.unsqueeze_(0)
-        try:
-            os.remove(files[0])
-        except:
-            pass
-        return (video, {"waveform":audio,"sample_rate":info["audio_fps"]},)
+            raise Exception("Cut Failed")