style: formatting (#278)

Author: timerring

requirements.txt

@@ -8,7 +8,7 @@ pysrt==1.1.2
 six==1.16.0
 tiktoken==0.4.0
 torch==1.11.0
-triton==3.1.0
+# triton==3.1.0
 zhconv==1.4.3
 Pillow>=5.2.0
 PyYAML>=6.0.2

src/autoslice/__init__.py

@@ -2,7 +2,8 @@
 
 import sys
 import os
+
 sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
 from .auto_slice_video.autosv import slice_video_by_danmaku
 
-__all__ = ["slice_video_by_danmaku"]
+__all__ = ["slice_video_by_danmaku"]

src/autoslice/auto_slice_video

@@ -3,21 +3,26 @@
 import subprocess
 from src.log.logger import scan_log
 
+
 # https://stackoverflow.com/questions/64849478/cant-insert-stream-metadata-into-mp4
 def inject_metadata(video_path, generate_title, output_path):
     """Slice the video using ffmpeg."""
     command = [
-        'ffmpeg',
-        '-i', video_path,
-        '-metadata:g', f'generate={generate_title}',
-        '-c:v', 'copy',
-        '-c:a', 'copy',
-        output_path
+        "ffmpeg",
+        "-i",
+        video_path,
+        "-metadata:g",
+        f"generate={generate_title}",
+        "-c:v",
+        "copy",
+        "-c:a",
+        "copy",
+        output_path,
     ]
     try:
         result = subprocess.run(command, check=True, capture_output=True, text=True)
         scan_log.debug(f"FFmpeg output: {result.stdout}")
         if result.stderr:
             scan_log.debug(f"FFmpeg debug: {result.stderr}")
     except subprocess.CalledProcessError as e:
-        scan_log.error(f"Error: {e.stderr}")
+        scan_log.error(f"Error: {e.stderr}")

src/autoslice/inject_metadata.py

@@ -1,27 +1,28 @@
 from google import genai
-from google.genai import types 
+from google.genai import types
 from src.log.logger import scan_log
 from src.config import GEMINI_API_KEY, SLICE_PROMPT
 
+
 def gemini_generate_title(video_path, artist):
 
     client = genai.Client(api_key=GEMINI_API_KEY)
 
     # Only for videos of size <20Mb
-    video_bytes = open(video_path, 'rb').read()
+    video_bytes = open(video_path, "rb").read()
 
     response = client.models.generate_content(
-        model='models/gemini-2.0-flash',
+        model="models/gemini-2.0-flash",
         contents=types.Content(
             parts=[
                 types.Part(text=SLICE_PROMPT),
                 types.Part(
-                    inline_data=types.Blob(data=video_bytes, mime_type='video/mp4')
-                )
+                    inline_data=types.Blob(data=video_bytes, mime_type="video/mp4")
+                ),
             ]
-        )
+        ),
     )
     scan_log.info("使用 Gemini-2.0-flash 生成切片标题")
     scan_log.info(f"Prompt: {SLICE_PROMPT}")
     scan_log.info(f"生成的切片标题为: {response.text}")
-    return response.text
+    return response.text

src/autoslice/mllm_sdk/gemini_new_sdk.py

@@ -6,6 +6,7 @@
 # the new gemini sdk has the conflicts pydantic version in project, so we use the old one
 # https://github.com/google-gemini/deprecated-generative-ai-python
 
+
 def gemini_generate_title(video_path, artist):
 
     genai.configure(api_key=GEMINI_API_KEY)
@@ -27,11 +28,12 @@ def gemini_generate_title(video_path, artist):
     # Set the model to Gemini Flash.
     model = genai.GenerativeModel(model_name="models/gemini-2.0-flash")
 
-    response = model.generate_content([prompt, video_file],
-                                    request_options={"timeout": 600})
+    response = model.generate_content(
+        [prompt, video_file], request_options={"timeout": 600}
+    )
     # delete the video file
     genai.delete_file(video_file.name)
     scan_log.info("使用 Gemini-2.0-flash 生成切片标题")
     scan_log.info(f"Prompt: {SLICE_PROMPT}")
     scan_log.info(f"生成的切片标题为: {response.text}")
-    return response.text
+    return response.text

src/autoslice/mllm_sdk/gemini_old_sdk.py

@@ -4,10 +4,12 @@
 import os
 import base64
 
+
 def encode_video(video_path):
     with open(video_path, "rb") as video_file:
         return base64.b64encode(video_file.read()).decode("utf-8")
 
+
 def qwen_generate_title(video_path, artist):
     client = OpenAI(
         api_key=QWEN_API_KEY,
@@ -20,7 +22,8 @@ def qwen_generate_title(video_path, artist):
         messages=[
             {
                 "role": "system",
-                "content": [{"type":"text","text": "你是一个视频切片员"}]},
+                "content": [{"type": "text", "text": "你是一个视频切片员"}],
+            },
             {
                 "role": "user",
                 "content": [
@@ -30,7 +33,7 @@ def qwen_generate_title(video_path, artist):
                     },
                     {"type": "text", "text": SLICE_PROMPT},
                 ],
-            }
+            },
         ],
     )
     scan_log.info("使用 Qwen-2.5-72B-Instruct 生成切片标题")

src/autoslice/mllm_sdk/qwen_sdk.py

@@ -5,32 +5,25 @@
 from zhipuai import ZhipuAI
 from src.log.logger import scan_log
 
+
 def zhipu_glm_4v_plus_generate_title(video_path, artist):
-    with open(video_path, 'rb') as video_file:
-        video_base = base64.b64encode(video_file.read()).decode('utf-8')
+    with open(video_path, "rb") as video_file:
+        video_base = base64.b64encode(video_file.read()).decode("utf-8")
 
     client = ZhipuAI(api_key=ZHIPU_API_KEY)
     response = client.chat.completions.create(
         model="glm-4v-plus-0111",
         messages=[
-        {
-            "role": "user",
-            "content": [
-            {
-                "type": "video_url",
-                "video_url": {
-                    "url" : video_base
-                }
-            },
             {
-                "type": "text",
-                "text": SLICE_PROMPT
+                "role": "user",
+                "content": [
+                    {"type": "video_url", "video_url": {"url": video_base}},
+                    {"type": "text", "text": SLICE_PROMPT},
+                ],
             }
-            ]
-        }
-        ]
+        ],
     )
     scan_log.info("使用 Zhipu-glm-4v-plus 生成切片标题")
     scan_log.info(f"Prompt: {SLICE_PROMPT}")
     scan_log.info(f"生成的切片标题为: {response.choices[0].message.content}")
-    return response.choices[0].message.content.replace("《", "").replace("》", "")
+    return response.choices[0].message.content.replace("《", "").replace("》", "")

src/autoslice/mllm_sdk/zhipu_sdk.py

@@ -2,30 +2,38 @@
 from src.log.logger import scan_log
 from src.config import MLLM_MODEL
 
+
 def title_generator(model_type):
     """Decorator to select title generation function based on model type
     Args:
         model_type: str, type of model to use
     Returns:
         function: wrapped title generation function
     """
+
     def decorator(func):
         def wrapper(video_path, artist):
             if model_type == "zhipu":
                 from .mllm_sdk.zhipu_sdk import zhipu_glm_4v_plus_generate_title
+
                 return zhipu_glm_4v_plus_generate_title(video_path, artist)
             elif model_type == "gemini":
                 from .mllm_sdk.gemini_old_sdk import gemini_generate_title
+
                 return gemini_generate_title(video_path, artist)
             elif model_type == "qwen":
                 from .mllm_sdk.qwen_sdk import qwen_generate_title
+
                 return qwen_generate_title(video_path, artist)
             else:
                 scan_log.error(f"Unsupported model type: {model_type}")
                 return None
+
         return wrapper
+
     return decorator
 
+
 @title_generator(MLLM_MODEL)
 def generate_title(video_path, artist):
     """Generate title for video
@@ -35,4 +43,4 @@ def generate_title(video_path, artist):
     Returns:
         str: generated title
     """
-    pass  # The actual implementation is handled by the decorator
+    pass  # The actual implementation is handled by the decorator

src/autoslice/title_generator.py

@@ -2,4 +2,5 @@
 
 import sys
 import os
-sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
+
+sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))