refactor: change project structure

src/__init__.py

@@ -0,0 +1,7 @@
+from processing import pdf2image, image2base64
+from config import (
+    Config,
+    ModelConfig,
+    Navigator,
+    MultilineWrapper
+)

src/chains/__init__.py

@@ -0,0 +1,13 @@
+from pipelines import (
+    SingleSlidePipeline,
+    PresentationPipeline,
+    PresentationAnalysis,
+    SlideAnalysis
+)
+
+from chains import (
+    LoadPageChain,
+    Page2ImageChain,
+    ImageEncodeChain,
+    VisionAnalysisChain
+)

src/{pdf_utils → chains}/chains.py

@@ -15,10 +15,10 @@ import fitz
 
 from io import BytesIO
 from PIL import Image
-from src.pdf_utils.chain_funcs import get_param_or_default
+from src.chains.chain_funcs import get_param_or_default
 
 from src.config import Navigator
-from src.pdf_utils.pdf2image import page2image
+from src.processing import page2image, image2base64
 
 logger = logging.getLogger(__name__)
 
@@ -171,14 +171,7 @@ class ImageEncodeChain(Chain):
             Dictionary with base64 encoded image string
         """
         image: Image.Image = inputs["image"]
-
-        # Save image to bytes buffer
-        buffer = BytesIO()
-        image.save(buffer, format="PNG")
-
-        # Encode to base64
-        encoded = base64.b64encode(buffer.getvalue()).decode("utf-8")
-
+        encoded = image2base64(image)
         return dict(image_encoded=encoded)
 
 

src/{pdf_utils → chains}/pipelines.py

@@ -10,7 +10,7 @@ import fitz
 from langchain_openai.chat_models import ChatOpenAI
 from langchain.chains.base import Chain
 
-from src.pdf_utils.chains import (
+from src.chains.chains import (
     LoadPageChain,
     Page2ImageChain,
     ImageEncodeChain,

src/{prompts → chains/prompts}/vit_describe_image_ru_detailed.txt

@@ -22,7 +22,7 @@
 - Формулы (математические выражения)
 - Код (язык программирования, листинги)
 - Цитаты
-- Ссылки (URLs, референсы)
+- Ссылки (URLs, референсы, источники)
 - Аббревиатуры и специальные термины
 
 Стилистика:
@@ -32,13 +32,13 @@
 - Расположение текста (центрирование, колонки)
 </examples>
 
-Не ограничивайтесь этим списком. Если на слайде есть другие важные текстовые элементы, опишите их тоже. Если каких-то элементов нет - пропустите их. 
+Не ограничивайтесь этим списком. Если на слайде есть другие важные текстовые элементы, опишите их тоже. Если каких-то элементов нет - пропустите их.
 Не пишите об отсутствующих элементах!
 
 Не суммаризируйте основной текст. Передавайте его полностью.
 
 # Визуальное содержание
-Опишите все визуальные элементы. Вот примеры визуальных элементов, на которые нужно обращать внимание: 
+Опишите все визуальные элементы. Вот примеры визуальных элементов, на которые нужно обращать внимание:
 
 <examples>
 Данные и графики:
@@ -70,6 +70,6 @@
 - Форматирование ячеек
 </examples>
 
-Не ограничивайтесь этим списком. Если на слайде есть другие важные визуальные элементы, опишите их тоже. Если каких-то элементов нет - пропустите их. 
+Не ограничивайтесь этим списком. Если на слайде есть другие важные визуальные элементы, опишите их тоже. Если каких-то элементов нет - пропустите их.
 Не пишите об отсутствующих элементах!
 </description>

src/config/navigator.py

@@ -18,7 +18,8 @@ class Navigator:
 
         # src paths
         self.src = self.root / "src"
-        self.prompts = self.src / "prompts"
+        self.chains = self.src / "chains"
+        self.prompts = self.chains / "prompts"
 
         # Create directories if they don't exist
         for path in [self.interim, self.processed]:

src/processing/__init__.py

@@ -0,0 +1,2 @@
+from src.processing.image_utlis import image2base64
+from src.processing.pdf_utils import pdf2image