Spaces:
Build error
Build error
Ilia Tambovtsev commited on
Commit ·
a35a16c
1
Parent(s): 211933f
feat: save vision prompt to presentation
Browse files- src/chains/pipelines.py +7 -1
src/chains/pipelines.py
CHANGED
|
@@ -34,6 +34,7 @@ class PresentationAnalysis(BaseModel):
|
|
| 34 |
"""Container for presentation analysis results"""
|
| 35 |
name: str
|
| 36 |
path: Path
|
|
|
|
| 37 |
metadata: Dict = Field(default_factory=dict)
|
| 38 |
slides: List[SlideAnalysis] = Field(default_factory=list)
|
| 39 |
timestamp: str = Field(
|
|
@@ -155,6 +156,7 @@ class PresentationPipeline(Chain):
|
|
| 155 |
base_path: Base path for storing analysis results
|
| 156 |
"""
|
| 157 |
super().__init__(**kwargs)
|
|
|
|
| 158 |
self._slide_pipeline = SingleSlidePipeline(
|
| 159 |
llm=llm,
|
| 160 |
vision_prompt=vision_prompt,
|
|
@@ -252,7 +254,11 @@ class PresentationPipeline(Chain):
|
|
| 252 |
presentation = (
|
| 253 |
PresentationAnalysis.load(latest_analysis)
|
| 254 |
if latest_analysis
|
| 255 |
-
else PresentationAnalysis(
|
|
|
|
|
|
|
|
|
|
|
|
|
| 256 |
)
|
| 257 |
|
| 258 |
# Get set of already processed pages
|
|
|
|
| 34 |
"""Container for presentation analysis results"""
|
| 35 |
name: str
|
| 36 |
path: Path
|
| 37 |
+
vision_prompt: str
|
| 38 |
metadata: Dict = Field(default_factory=dict)
|
| 39 |
slides: List[SlideAnalysis] = Field(default_factory=list)
|
| 40 |
timestamp: str = Field(
|
|
|
|
| 156 |
base_path: Base path for storing analysis results
|
| 157 |
"""
|
| 158 |
super().__init__(**kwargs)
|
| 159 |
+
self._vision_prompt = vision_prompt
|
| 160 |
self._slide_pipeline = SingleSlidePipeline(
|
| 161 |
llm=llm,
|
| 162 |
vision_prompt=vision_prompt,
|
|
|
|
| 254 |
presentation = (
|
| 255 |
PresentationAnalysis.load(latest_analysis)
|
| 256 |
if latest_analysis
|
| 257 |
+
else PresentationAnalysis(
|
| 258 |
+
name=pdf_path.stem,
|
| 259 |
+
path=pdf_path,
|
| 260 |
+
vision_prompt=self._vision_prompt
|
| 261 |
+
)
|
| 262 |
)
|
| 263 |
|
| 264 |
# Get set of already processed pages
|