chore: fix linting and code formatting

app.py

@@ -1,6 +1,5 @@
 import streamlit as st
 import weave
-from dotenv import load_dotenv
 
 from medrag_multi_modal.assistant import (
     FigureAnnotatorFromPageImage,

medrag_multi_modal/document_loader/image_loader/base_img_loader.py

@@ -5,8 +5,8 @@ from typing import Dict, List, Optional
 
 import jsonlines
 import rich
-
 import wandb
+
 from medrag_multi_modal.document_loader.text_loader.base_text_loader import (
     BaseTextLoader,
 )

medrag_multi_modal/retrieval/bm25s_retrieval.py

@@ -3,11 +3,10 @@ from glob import glob
 from typing import Optional
 
 import bm25s
+import wandb
 import weave
 from Stemmer import Stemmer
 
-import wandb
-
 LANGUAGE_DICT = {
     "english": "en",
     "french": "fr",

medrag_multi_modal/retrieval/colpali_retrieval.py

@@ -6,9 +6,8 @@ import weave
 if TYPE_CHECKING:
     from byaldi import RAGMultiModalModel
 
-from PIL import Image
-
 import wandb
+from PIL import Image
 
 from ..utils import get_wandb_artifact
 

medrag_multi_modal/retrieval/common.py

@@ -3,7 +3,6 @@ from enum import Enum
 import safetensors
 import safetensors.torch
 import torch
-
 import wandb
 
 

medrag_multi_modal/utils.py

@@ -3,9 +3,8 @@ import io
 
 import jsonlines
 import torch
-from PIL import Image
-
 import wandb
+from PIL import Image
 
 
 def get_wandb_artifact(