Initial commit of akmon project

video_splitter/processor.py

@@ -0,0 +1,58 @@
+import ffmpeg
+import os
+import sys
+
+class VideoProcessor:
+    def __init__(self):
+        pass
+
+    def get_video_info(self, input_path: str):
+        try:
+            probe = ffmpeg.probe(input_path)
+            video_stream = next((stream for stream in probe['streams'] if stream['codec_type'] == 'video'), None)
+            return video_stream
+        except ffmpeg.Error as e:
+            print(f"Error probing video: {e.stderr}", file=sys.stderr)
+            return None
+
+    def split_video(self, input_path: str, start_time: str, end_time: str, output_path: str):
+        """
+        使用 FFmpeg 切割视频
+        Args:
+            input_path: 输入视频路径
+            start_time: 开始时间 (HH:MM:SS)
+            end_time: 结束时间 (HH:MM:SS)
+            output_path: 输出文件路径
+        """
+        print(f"[*] 正在切割视频: {start_time} - {end_time} -> {output_path}")
+        
+        try:
+            (
+                ffmpeg
+                .input(input_path, ss=start_time, to=end_time)
+                .output(output_path, c='copy') # Stream copy for speed and quality
+                .overwrite_output()
+                .run(capture_stdout=True, capture_stderr=True)
+            )
+            print(f"[+] 切割成功: {output_path}")
+            return True
+        except ffmpeg.Error as e:
+            print(f"[-] 切割失败: {e.stderr.decode('utf8')}", file=sys.stderr)
+            return False
+
+    def extract_audio(self, input_path: str, output_path: str):
+        """
+        提取音频用于 ASR
+        """
+        try:
+            (
+                ffmpeg
+                .input(input_path)
+                .output(output_path, ac=1, ar=16000) # Mono, 16k for ASR
+                .overwrite_output()
+                .run(quiet=True)
+            )
+            return True
+        except ffmpeg.Error as e:
+            print(f"[-] 音频提取失败: {e.stderr.decode('utf8')}")
+            return False