import ollama
from src.model.vision_model import ChartVisionModel

class ChartAnalyzer:
    def __init__(self):
        self.vision_model = ChartVisionModel()
        self.client = ollama.Client()
    
    def analyze_full(self, chart_image, pattern_images, indicator_charts, historical_data):
        vision_analysis = self.vision_model.analyze_chart(chart_image)
        
        system_prompt = """Analyze this technical chart with all patterns and indicators:
        1. Pattern Analysis (300+ patterns)
        2. Indicator Analysis
        3. Current Chart Analysis
        4. Movement Predictions
        Provide detailed analysis for each component."""
        
        analysis = self.client.generate(
            model="llama3.2-vision:latest",
            prompt=system_prompt,
            images=[chart_image] + pattern_images
        )
        
        return {
            'patterns': self._analyze_patterns(pattern_images),
            'indicators': self._analyze_indicators(indicator_charts),
            'predictions': self._generate_predictions(analysis)
        }