fix: resolve imports

src/__init__.py

@@ -3,7 +3,8 @@ from src.config import (
     Config,
     ModelConfig,
     Navigator,
-    MultilineWrapper
+    MultilineWrapper,
+    query_and_display,
 )
 
 from src.chains import (

src/chains/chains.py

@@ -17,7 +17,7 @@ from io import BytesIO
 from PIL import Image
 from src.chains.chain_funcs import get_param_or_default
 
-from src.config import Navigator
+from src.config.navigator import Navigator
 from src.processing import page2image, image2base64
 
 logger = logging.getLogger(__name__)

src/chains/pipelines.py

@@ -17,7 +17,7 @@ from src.chains.chains import (
     VisionAnalysisChain
 )
 
-from src.config import Navigator
+from src.config.navigator import Navigator
 
 
 logger = logging.getLogger(__name__)

src/config/navigator.py

@@ -1,6 +1,9 @@
 from pathlib import Path
 from dataclasses import dataclass
 from typing import List, Optional, Union
+import logging
+
+logger = logging.getLogger(__name__)
 
 @dataclass
 class Navigator:
@@ -78,7 +81,7 @@ class Navigator:
         if extension is not None:
             results = [path for path in results if path.name.endswith(extension)]
         if len(results) > 1:
-            print(f"Found {len(results)} matches for {substr}")
+            logger.info(f"Found {len(results)} matches for {substr}")
 
         if not results:
             return None

src/testing_utils/__init__.py

@@ -1,2 +1 @@
 from src.testing_utils.echo_llm import EchoLLM
-from src.testing_utils.display import query_and_display

src/testing_utils/display.py

@@ -1,73 +0,0 @@
-from typing import Optional, Union
-from pathlib import Path
-import matplotlib.pyplot as plt
-
-from src.config.multiline_wrapper import MultilineWrapper
-from src.config import Navigator
-from src.chains import SingleSlidePipeline
-from langchain_core.language_models.base import BaseLanguageModel
-
-
-def query_and_display(
-    path: Union[str, Path],
-    page_num: int,
-    llm: BaseLanguageModel,
-    vision_prompt: Optional[str] = None,
-    wrap_width: Optional[int] = None,
-    display_image: bool = True
-) -> dict:
-    """Query vision model and display results
-
-    Args:
-        path: PDF file path or substring of filename
-        page_num: Zero-based page number
-        llm: Language model with vision capabilities
-        vision_prompt: Optional custom prompt to use
-        wrap_width: Width to wrap text output
-        display_image: Whether to display image with matplotlib
-
-    Returns:
-        Dictionary with model outputs
-    """
-    # Find file if path is substring
-    nav = Navigator()
-    if isinstance(path, str):
-        pdf_path = nav.find_file_by_substr(path)
-        if pdf_path is None:
-            raise ValueError(f"No PDF found matching '{path}'")
-    else:
-        pdf_path = Path(path)
-
-    # Create slide processing pipeline
-    pipeline = SingleSlidePipeline(
-        llm=llm,
-        vision_prompt=vision_prompt if vision_prompt else "Describe this slide in detail"
-    )
-
-    # Process slide
-    result = pipeline.invoke({
-        "pdf_path": pdf_path,
-        "page_num": page_num
-    })
-
-    # Display results
-    if display_image:
-        fig, ax = plt.subplots(figsize=(12, 12))
-        # Convert page to image again to display
-        chain_result = (
-            pipeline._chain.invoke({
-                "pdf_path": pdf_path,
-                "page_num": page_num
-            })
-        )
-        ax.imshow(chain_result["image"])
-        ax.axis("off")
-        plt.show()
-
-    # Format and print text
-    text_wrapper = MultilineWrapper(width=wrap_width)
-    slide = result["slide_analysis"]
-    print(f"Prompt: {slide.vision_prompt}\n")
-    print(text_wrapper.fill(slide.content))
-
-    return result