o X hE @sdddlZddlZddlmZddlmZddlmZddlm Z ddl m Z e GdddZ dS) N)Groq) AsyncSession) Prediction) encode_image) load_dotenvc@sFeZdZdefddZdededefddZd ed ed efd d ZdS)ImageComparatordbcCs||_ttdd|_dS)N GROQ_API_KEY)api_key)rrosgetenvclient)selfrrK/var/www/html/construction_image-detection-poc/src/components/comparison.py__init__ szImageComparator.__init__base64_1base64_2returnc sz?tj|jjjjddddddddd d d |id dd dd d d |id dddgdgdddIdH}|jdjjWSt yY}z dt |WYd}~Sd}~ww)Nz-meta-llama/llama-4-maverick-17b-128e-instructsystemzYou are a professional construction site analyst with expertise in identifying and assessing construction materials, equipment, and safety conditions.)rolecontentusertextaGCompare these two construction site images and provide ONLY: 1. A concise bulleted list of changes between Image 1 and Image 2, focusing on: - New materials/equipment that have appeared - Materials/equipment that have been removed - Materials/equipment that have been relocated - Changes in quantity of materials - Progress indicators (e.g., "scaffolding extended", "bricks used") - Any new safety concerns 2. A single summary sentence about the overall progress made between the two images. Focus on these construction elements: - Wooden logs - Concrete Mixer - Rebar - Scaffolding - Doors - Piles of Bricks - Metallic Shuttering Keep your response brief, factual, and structured for easy conversion to JSON format. Do not include any introductory text, analysis methodology, or conclusions beyond what is requested.)typer image_urlurlzdata:image/jpeg;base64,)rrzImage 1 (Earlier timestamp)zImage 2 (Later timestamp)g333333?i)modelmessages temperaturemax_completion_tokensrzError during comparison: ) asyncio to_threadr chat completionscreatechoicesmessager Exceptionstr)rrrresponseerrrgenerate_comparisons,  %z#ImageComparator.generate_comparison prediction_id before_path after_pathc szHt|IdH}t|IdH}|||IdH}|jt|IdH}|r?d|i|_d|_|jIdH|j|IdHnt d||jdWSt y^}z t dt |d}~ww)Ncomparison_report processingzPrediction not found)r0statusz#Failed to run the text comparison: ) rr,rgetr text_resultsr2commitrefreshr(r)) rr-r.r/rrcomparison_text predictionr+rrrrun_comparison;s& zImageComparator.run_comparisonN) __name__ __module__ __qualname__rrr)r,intr9rrrrr s+r) r r!groqrsqlalchemy.ext.asynciorsrc.models.predictionrsrc.utils.files_utilsrdotenvrrrrrrs