"""
Entry point script for analyzing images with the forensic agent.
"""

import argparse
import sys
from pathlib import Path

# Add src to path
sys.path.insert(0, str(Path(__file__).parent.parent))

from src.agents import ForensicAgent


def main():
    parser = argparse.ArgumentParser(
        description="Analyze an image using the forensic agent"
    )
    parser.add_argument(
        "--image",
        type=str,
        required=True,
        help="Path to the image file to analyze"
    )
    parser.add_argument(
        "--query",
        type=str,
        default=None,
        help="Optional specific question about the image"
    )
    parser.add_argument(
        "--model",
        type=str,
        default="gpt-5.1",
        help="LLM model to use (default: gpt-5.1)"
    )
    parser.add_argument(
        "--temperature",
        type=float,
        default=0.2,
        help="LLM temperature (default: 0.2)"
    )
    
    args = parser.parse_args()
    
    # Initialize agent
    print(f"Initializing forensic agent with model: {args.model}...")
    agent = ForensicAgent(
        llm_model=args.model,
        temperature=args.temperature
    )
    
    # Analyze image
    print(f"\nAnalyzing image: {args.image}")
    if args.query:
        print(f"Query: {args.query}")
    print("-" * 60)
    
    try:
        result = agent.analyze(args.image, args.query)
        
        print("\n" + "=" * 60)
        print("ANALYSIS RESULTS")
        print("=" * 60)
        print(f"\nConclusion:\n{result['conclusion']}")
        
        if result['tool_usage']:
            print(f"\nTools used: {', '.join(result['tool_usage'])}")
        else:
            print("\nNo tools were used.")
        
        print("\n" + "=" * 60)
        
    except Exception as e:
        print(f"\nError: {e}", file=sys.stderr)
        sys.exit(1)


if __name__ == "__main__":
    main()