feat: Enhance StyleGuard with VLM-based style extraction for specific style_prefix

2026-01-07 00:16:57 +08:00
parent a3ab12e87c
commit 297f3ccda4
2 changed files with 104 additions and 15 deletions
--- a/api/routers/quality.py
+++ b/api/routers/quality.py
@@ -254,8 +254,13 @@ async def extract_style(
            return style_schema
        
        from pixelle_video.services.quality.style_guard import StyleGuard
+        from api.dependencies import get_pixelle_video
        
-        style_guard = StyleGuard()
+        # Get LLM service for VLM-based style extraction
+        pixelle_video = await get_pixelle_video()
+        llm_service = pixelle_video.llm if pixelle_video else None
+        
+        style_guard = StyleGuard(llm_service=llm_service)
        anchor = await style_guard.extract_style_anchor(actual_path)
        
        style_schema = StyleAnchorSchema(