File size: 1,451 Bytes
734b5b4 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 | from PIL import Image
from typing import Dict, Any
from .base_agent import LLMAgent
class ImageProcessingAgent(LLMAgent):
name = "image_processing_agent"
description = "Enhances and processes images within PDF documents - handles resolution, contrast, noise reduction"
def build_prompt(self, pil_image: Image.Image) -> str:
width, height = pil_image.size
return f"""Analyze this image from a PDF ({width}x{height} pixels).
Assess:
1. Image quality - is it clear, blurry, or noisy?
2. Contrast - good or poor?
3. Resolution - adequate for the page size?
4. Any distortion or artifacts?
5. Are there multiple images or a single hero image?
6. Should this image be enhanced or is it already optimal?
Recommend processing steps if needed."""
def execute(self, pil_image: Image.Image, reasoning: str) -> Dict[str, Any]:
return {
"processed": True,
"image_count": 1,
"quality": self._assess_quality(reasoning),
"needs_enhancement": "blur" in reasoning.lower()
or "noise" in reasoning.lower(),
"reasoning": reasoning,
}
def _assess_quality(self, reasoning: str) -> str:
reasoning_lower = reasoning.lower()
if "blur" in reasoning_lower or "noise" in reasoning_lower:
return "poor"
if "clear" in reasoning_lower or "sharp" in reasoning_lower:
return "high"
return "medium"
|