From a878d2459ba60bfad3656c84228da2feb008077d Mon Sep 17 00:00:00 2001
From: yichuan-w <yichuan_wang@berkeley.edu>
Date: Wed, 17 Dec 2025 09:02:48 +0000
Subject: [PATCH] Format code style in leann_multi_vector.py for better
 readability
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude Sonnet 4.5 <noreply@anthropic.com>
---
 .../leann_multi_vector.py                     | 30 +++++++++++--------
 1 file changed, 17 insertions(+), 13 deletions(-)

diff --git a/apps/multimodal/vision-based-pdf-multi-vector/leann_multi_vector.py b/apps/multimodal/vision-based-pdf-multi-vector/leann_multi_vector.py
index efac526..7168ac2 100644
--- a/apps/multimodal/vision-based-pdf-multi-vector/leann_multi_vector.py
+++ b/apps/multimodal/vision-based-pdf-multi-vector/leann_multi_vector.py
@@ -97,21 +97,23 @@ def _natural_sort_key(name: str) -> int:
     return int(m.group()) if m else 0
 
 
-def _load_images_from_dir(pages_dir: str, recursive: bool = False) -> tuple[list[str], list[Image.Image]]:
+def _load_images_from_dir(
+    pages_dir: str, recursive: bool = False
+) -> tuple[list[str], list[Image.Image]]:
     """
     Load images from a directory.
-    
+
     Args:
         pages_dir: Directory path containing images
         recursive: If True, recursively search subdirectories (default: False)
-    
+
     Returns:
         Tuple of (filepaths, images)
     """
-    
+
     # Supported image extensions
     extensions = ("*.png", "*.jpg", "*.jpeg", "*.PNG", "*.JPG", "*.JPEG", "*.webp", "*.WEBP")
-    
+
     if recursive:
         # Recursive search
         filepaths = []
@@ -124,31 +126,33 @@ def _load_images_from_dir(pages_dir: str, recursive: bool = False) -> tuple[list
         for ext in extensions:
             pattern = os.path.join(pages_dir, ext)
             filepaths.extend(glob.glob(pattern))
-    
+
     # Sort files naturally
     filepaths = sorted(filepaths, key=lambda x: _natural_sort_key(os.path.basename(x)))
-    
+
     # Load images with error handling
     images = []
     valid_filepaths = []
     failed_count = 0
-    
+
     for filepath in filepaths:
         try:
             img = Image.open(filepath)
             # Convert to RGB if necessary (handles RGBA, P, etc.)
-            if img.mode != 'RGB':
-                img = img.convert('RGB')
+            if img.mode != "RGB":
+                img = img.convert("RGB")
             images.append(img)
             valid_filepaths.append(filepath)
         except Exception as e:
             failed_count += 1
             print(f"Warning: Failed to load image {filepath}: {e}")
             continue
-    
+
     if failed_count > 0:
-        print(f"Warning: Failed to load {failed_count} image(s) out of {len(filepaths)} total files")
-    
+        print(
+            f"Warning: Failed to load {failed_count} image(s) out of {len(filepaths)} total files"
+        )
+
     return valid_filepaths, images