feat: add webp conversion to balatrobench cli

Author: S1M0N38

src/balatrollm/balatrobench_cli.py

@@ -49,6 +49,11 @@ def main() -> None:
         action="store_true",
         help="Convert PNG screenshots to AVIF format after analysis",
     )
+    parser.add_argument(
+        "--webp",
+        action="store_true",
+        help="Convert PNG screenshots to WebP format after analysis",
+    )
 
     args = parser.parse_args()
 
@@ -80,6 +85,11 @@ def main() -> None:
             print("Converting PNG screenshots to AVIF format...")
             analyzer.convert_pngs_to_avif(args.output_dir)
 
+        # Convert PNGs to WebP if requested
+        if args.webp:
+            print("Converting PNG screenshots to WebP format...")
+            analyzer.convert_pngs_to_webp(args.output_dir)
+
         # Generate manifest.json in the base benchmark directory
         manifest_base_dir = args.output_dir.parent
         analyzer.generate_manifest(manifest_base_dir, __version__)

src/balatrollm/benchmark.py

@@ -402,6 +402,50 @@ def convert_pngs_to_avif(self, directory: Path) -> None:
         except Exception as e:
             print(f"Warning: cavif conversion error in {directory}: {e}")
 
+    def _convert_single_png_to_webp(self, png_file: Path) -> None:
+        """Convert a single PNG file to WebP."""
+        try:
+            webp_file = png_file.with_suffix(".webp")
+            subprocess.run(
+                [
+                    "cwebp",
+                    "-q",
+                    "80",
+                    "-quiet",
+                    str(png_file),
+                    "-o",
+                    str(webp_file),
+                ],
+                capture_output=True,
+                text=True,
+                check=True,
+            )
+            png_file.unlink()
+        except subprocess.CalledProcessError as e:
+            print(f"Warning: cwebp conversion failed for {png_file}: {e.stderr}")
+        except OSError as e:
+            print(f"Warning: Could not remove {png_file}: {e}")
+
+    def convert_pngs_to_webp(self, directory: Path) -> None:
+        """Convert all PNG files in directory to WebP using cwebp with parallelization."""
+        try:
+            png_files = list(directory.rglob("screenshot.png"))
+            if not png_files:
+                return
+
+            with ThreadPoolExecutor(max_workers=8) as executor:
+                list(
+                    tqdm(
+                        executor.map(self._convert_single_png_to_webp, png_files),
+                        total=len(png_files),
+                        desc="Converting to WebP",
+                    )
+                )
+        except FileNotFoundError:
+            print("Warning: cwebp not found, keeping PNG format")
+        except Exception as e:
+            print(f"Warning: cwebp conversion error in {directory}: {e}")
+
     def extract_request_content(self, requests_file: Path) -> dict[str, str]:
         """Extract request content from requests.jsonl by custom_id."""
         content_by_id = {}