Delete agents/agent_3_difference_analyzer_enhanced.py

agents/agent_3_difference_analyzer_enhanced.py

@@ -1,385 +0,0 @@
-"""
-Agent 3: Enhanced Difference Analyzer
-Detects visual differences including typography, spacing, components, and layout
-Uses HF vision model + CSS analysis + pixel comparison
-"""
-
-import os
-import sys
-from typing import Dict, Any, List
-from pathlib import Path
-
-sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
-
-from state_schema import WorkflowState, VisualDifference
-
-
-class EnhancedDifferenceAnalyzer:
-    """Enhanced analyzer for detecting visual differences"""
-    
-    def __init__(self, hf_token: str = None):
-        """Initialize analyzer with HF token"""
-        self.hf_token = hf_token or os.getenv('HUGGINGFACE_API_KEY')
-        self.differences: List[VisualDifference] = []
-        self.detected_categories = {}
-    
-    def analyze_differences(self, state: WorkflowState) -> WorkflowState:
-        """
-        Comprehensive difference analysis
-        
-        Args:
-            state: Current workflow state with screenshots
-        
-        Returns:
-            Updated state with detected differences
-        """
-        print("\n🔍 Agent 3: Enhanced Difference Analysis...")
-        
-        try:
-            self.differences = []
-            self.detected_categories = {}
-            
-            # Analyze each viewport
-            for viewport_name in ["desktop", "mobile"]:
-                figma_screenshots = state.get("figma_screenshots", {})
-                website_screenshots = state.get("website_screenshots", {})
-                
-                if viewport_name not in figma_screenshots or viewport_name not in website_screenshots:
-                    continue
-                
-                print(f"\n  📊 Analyzing {viewport_name.upper()} viewport...")
-                
-                figma_path = figma_screenshots[viewport_name]
-                website_path = website_screenshots[viewport_name]
-                
-                # Run comprehensive analysis
-                self._analyze_layout_structure(figma_path, website_path, viewport_name)
-                self._analyze_typography(figma_path, website_path, viewport_name)
-                self._analyze_colors(figma_path, website_path, viewport_name)
-                self._analyze_spacing(figma_path, website_path, viewport_name)
-                self._analyze_components(figma_path, website_path, viewport_name)
-                self._analyze_buttons(figma_path, website_path, viewport_name)
-                self._analyze_visual_hierarchy(figma_path, website_path, viewport_name)
-            
-            # Calculate similarity score
-            similarity_score = self._calculate_similarity_score()
-            
-            # Update state
-            state["visual_differences"] = [d.to_dict() if hasattr(d, "to_dict") else d for d in self.differences]
-            state["similarity_score"] = similarity_score
-            state["status"] = "analysis_complete"
-            
-            # Print summary
-            self._print_summary()
-            
-            return state
-        
-        except Exception as e:
-            print(f"  ❌ Analysis failed: {str(e)}")
-            import traceback
-            traceback.print_exc()
-            state["status"] = "analysis_failed"
-            state["error_message"] = f"Enhanced Analysis Error: {str(e)}"
-            return state
-    
-    def _analyze_layout_structure(self, figma_path: str, website_path: str, viewport: str):
-        """Analyze layout and structural differences"""
-        print(f"    📐 Checking layout & structure...")
-        
-        # Simulate detection of layout issues
-        layout_issues = [
-            {
-                "name": "Header height difference",
-                "category": "Layout & Structure",
-                "description": "Header height differs between design and development",
-                "severity": "High",
-                "location": {"x": 100, "y": 50}
-            },
-            {
-                "name": "Container width differs",
-                "category": "Layout & Structure",
-                "description": "Main container width is different",
-                "severity": "High",
-                "location": {"x": 400, "y": 200}
-            }
-        ]
-        
-        for issue in layout_issues:
-            if viewport == "desktop":  # Adjust per viewport
-                diff = VisualDifference(
-                    issue_id=f"layout-{len(self.differences)}",
-                    title=issue["name"],
-                    category=issue["category"],
-                    description=issue["description"],
-                    severity=issue["severity"],
-                    viewport=viewport,
-                    location=issue["location"],
-                    design_value="Design",
-                    website_value="Website",
-                    detection_method="HF Vision + Screenshot Analysis"
-                )
-                self.differences.append(diff)
-                self._track_category(issue["category"], issue["severity"])
-    
-    def _analyze_typography(self, figma_path: str, website_path: str, viewport: str):
-        """Analyze typography differences"""
-        print(f"    🔤 Checking typography...")
-        
-        typography_issues = [
-            {
-                "name": "Checkout heading font differs",
-                "category": "Typography",
-                "description": "Font family, size, and letter spacing differ",
-                "severity": "High",
-                "location": {"x": 150, "y": 100}
-            },
-            {
-                "name": "Contact info font weight differs",
-                "category": "Typography",
-                "description": "Font weight changed to bold in development",
-                "severity": "High",
-                "location": {"x": 200, "y": 250}
-            }
-        ]
-        
-        for issue in typography_issues:
-            if viewport == "desktop":
-                diff = VisualDifference(
-                    issue_id=f"typography-{len(self.differences)}",
-                    title=issue["name"],
-                    category=issue["category"],
-                    description=issue["description"],
-                    severity=issue["severity"],
-                    viewport=viewport,
-                    location=issue["location"],
-                    design_value="Design",
-                    website_value="Website",
-                    detection_method="CSS Extraction + HF Analysis"
-                )
-                self.differences.append(diff)
-                self._track_category(issue["category"], issue["severity"])
-    
-    def _analyze_colors(self, figma_path: str, website_path: str, viewport: str):
-        """Analyze color differences"""
-        print(f"    🎨 Checking colors...")
-        
-        # Color analysis would go here
-        pass
-    
-    def _analyze_spacing(self, figma_path: str, website_path: str, viewport: str):
-        """Analyze spacing and padding differences"""
-        print(f"    📏 Checking spacing...")
-        
-        spacing_issues = [
-            {
-                "name": "Padding differs (left, right)",
-                "category": "Spacing & Sizing",
-                "description": "Horizontal padding is different",
-                "severity": "Medium",
-                "location": {"x": 300, "y": 300}
-            },
-            {
-                "name": "Component spacing differs",
-                "category": "Spacing & Sizing",
-                "description": "Gap between components is different",
-                "severity": "Medium",
-                "location": {"x": 400, "y": 400}
-            }
-        ]
-        
-        for issue in spacing_issues:
-            if viewport == "desktop":
-                diff = VisualDifference(
-                    issue_id=f"spacing-{len(self.differences)}",
-                    title=issue["name"],
-                    category=issue["category"],
-                    description=issue["description"],
-                    severity=issue["severity"],
-                    viewport=viewport,
-                    location=issue["location"],
-                    design_value="Design",
-                    website_value="Website",
-                    detection_method="Screenshot Pixel Analysis"
-                )
-                self.differences.append(diff)
-                self._track_category(issue["category"], issue["severity"])
-    
-    def _analyze_components(self, figma_path: str, website_path: str, viewport: str):
-        """Analyze missing or misplaced components"""
-        print(f"    🧩 Checking components...")
-        
-        component_issues = [
-            {
-                "name": "Login link missing",
-                "category": "Components & Elements",
-                "description": "Login link component is missing in development",
-                "severity": "High",
-                "location": {"x": 450, "y": 50}
-            },
-            {
-                "name": "Payment component not visible",
-                "category": "Components & Elements",
-                "description": "Payment component is hidden or not rendered",
-                "severity": "High",
-                "location": {"x": 500, "y": 300}
-            },
-            {
-                "name": "Payment methods design missing",
-                "category": "Components & Elements",
-                "description": "Payment methods section is missing",
-                "severity": "High",
-                "location": {"x": 300, "y": 350}
-            },
-            {
-                "name": "Icons missing",
-                "category": "Components & Elements",
-                "description": "Various icons are not displayed",
-                "severity": "High",
-                "location": {"x": 250, "y": 400}
-            }
-        ]
-        
-        for issue in component_issues:
-            if viewport == "desktop":
-                diff = VisualDifference(
-                    issue_id=f"component-{len(self.differences)}",
-                    title=issue["name"],
-                    category=issue["category"],
-                    description=issue["description"],
-                    severity=issue["severity"],
-                    viewport=viewport,
-                    location=issue["location"],
-                    design_value="Design",
-                    website_value="Website",
-                    detection_method="HF Vision Model"
-                )
-                self.differences.append(diff)
-                self._track_category(issue["category"], issue["severity"])
-    
-    def _analyze_buttons(self, figma_path: str, website_path: str, viewport: str):
-        """Analyze button and interactive element differences"""
-        print(f"    🔘 Checking buttons...")
-        
-        button_issues = [
-            {
-                "name": "Button size, height, color differs",
-                "category": "Buttons & Interactive",
-                "description": "Button has no elevation/shadow and different styling",
-                "severity": "High",
-                "location": {"x": 350, "y": 500}
-            }
-        ]
-        
-        for issue in button_issues:
-            if viewport == "desktop":
-                diff = VisualDifference(
-                    issue_id=f"button-{len(self.differences)}",
-                    title=issue["name"],
-                    category=issue["category"],
-                    description=issue["description"],
-                    severity=issue["severity"],
-                    viewport=viewport,
-                    location=issue["location"],
-                    design_value="Design",
-                    website_value="Website",
-                    detection_method="CSS + Visual Analysis"
-                )
-                self.differences.append(diff)
-                self._track_category(issue["category"], issue["severity"])
-    
-    def _analyze_visual_hierarchy(self, figma_path: str, website_path: str, viewport: str):
-        """Analyze visual hierarchy and consistency"""
-        print(f"    🏗️  Checking visual hierarchy...")
-        
-        hierarchy_issues = [
-            {
-                "name": "Image size is different",
-                "category": "Components & Elements",
-                "description": "Product images have different dimensions",
-                "severity": "Medium",
-                "location": {"x": 600, "y": 250}
-            },
-            {
-                "name": "Checkout placement difference",
-                "category": "Components & Elements",
-                "description": "Checkout heading is positioned differently",
-                "severity": "High",
-                "location": {"x": 200, "y": 80}
-            }
-        ]
-        
-        for issue in hierarchy_issues:
-            if viewport == "desktop":
-                diff = VisualDifference(
-                    issue_id=f"layout-{len(self.differences)}",
-                    title=issue["name"],
-                    category=issue["category"],
-                    description=issue["description"],
-                    severity=issue["severity"],
-                    viewport=viewport,
-                    location=issue["location"],
-                    design_value="Design",
-                    website_value="Website",
-                    detection_method="HF Vision + Screenshot Analysis"
-                )
-                self.differences.append(diff)
-                self._track_category(issue["category"], issue["severity"])
-    
-    def _track_category(self, category: str, severity: str):
-        """Track detected categories and severity"""
-        if category not in self.detected_categories:
-            self.detected_categories[category] = {"High": 0, "Medium": 0, "Low": 0}
-        self.detected_categories[category][severity] += 1
-    
-    def _calculate_similarity_score(self) -> float:
-        """Calculate overall similarity score"""
-        if not self.differences:
-            return 100.0
-        
-        # Weight by severity
-        high_count = len([d for d in self.differences if d.severity == "High"])
-        medium_count = len([d for d in self.differences if d.severity == "Medium"])
-        low_count = len([d for d in self.differences if d.severity == "Low"])
-        
-        # Score calculation: each high = -10, medium = -5, low = -2
-        score = 100.0 - (high_count * 10 + medium_count * 5 + low_count * 2)
-        return max(0, score)
-    
-    def _print_summary(self):
-        """Print analysis summary"""
-        print(f"\n  📊 Analysis Summary:")
-        print(f"    Total Differences: {len(self.differences)}")
-        print(f"    High Severity: {len([d for d in self.differences if d.severity == 'High'])}")
-        print(f"    Medium Severity: {len([d for d in self.differences if d.severity == 'Medium'])}")
-        print(f"    Low Severity: {len([d for d in self.differences if d.severity == 'Low'])}")
-        print(f"    Similarity Score: {self._calculate_similarity_score():.1f}/100")
-        
-        print(f"\n  📂 Categories Detected:")
-        for category, counts in self.detected_categories.items():
-            total = sum(counts.values())
-            if total > 0:
-                print(f"    • {category}: {total} issues")
-
-
-def agent_3_node(state: Dict[str, Any]) -> Dict[str, Any]:
-    """
-    LangGraph node for Agent 3 (Enhanced Difference Analyzer)
-    
-    Args:
-        state: Current workflow state
-    
-    Returns:
-        Updated state with detected differences
-    """
-    # Convert dict to WorkflowState if needed
-    if isinstance(state, dict):
-        workflow_state = WorkflowState(**state)
-    else:
-        workflow_state = state
-    
-    # Create analyzer and analyze differences
-    analyzer = EnhancedDifferenceAnalyzer()
-    updated_state = analyzer.analyze_differences(workflow_state)
-    
-    # Convert back to dict for LangGraph
-    return updated_state.__dict__