# image_searcher.py
import requests
import random
from typing import List, Dict, Optional

class ImageSearcher:
    def __init__(self, pexels_api_key: str):
        self.pexels_api_key = pexels_api_key
        self.headers = {
            'Authorization': self.pexels_api_key,
            'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36'
        }

    def search_pexels(self, query: str, num_images: int = 10) -> List[Dict]:
        try:
            url = "https://api.pexels.com/v1/search"
            params = {
                'query': query,
                'per_page': min(num_images, 80),
                'page': 1
            }
            response = requests.get(url, params=params, headers=self.headers, timeout=10)
            response.raise_for_status()
            data = response.json()
            images = []
            for item in data.get('photos', []):
                images.append({
                    'url': item['src']['medium'],
                    'thumbnail': item['src']['small'],
                    'title': item.get('alt', query),
                    'author': item['photographer'],
                    'source': 'pexels'
                })
            print(f"Found {len(images)} images from Pexels for query: {query}")
            return images
        except Exception as e:
            print(f"Pexels search failed for query {query}: {e}")
            return []

    def search_lorem_picsum(self, query: str, width: int = 400, height: int = 300, num_images: int = 10) -> List[Dict]:
        images = []
        title_base = f"Pakistan {query}" if query else "Pakistan Travel"
        for i in range(num_images):
            seed = random.randint(1, 1000)
            images.append({
                'url': f"https://picsum.photos/{width}/{height}?random={seed}",
                'thumbnail': f"https://picsum.photos/200/150?random={seed}",
                'title': f"{title_base} View {i+1}",
                'author': 'Lorem Picsum',
                'source': 'lorem_picsum'
            })
        print(f"Generated {len(images)} placeholder images for query: {query}")
        return images

    def search_all_sources(self, queries: List[str], num_images: int = 6) -> List[Dict]:
        all_images = []
        for query in queries:
            if query:
                full_query = f"{query} Pakistan"
                pexels_images = self.search_pexels(full_query, num_images)
                all_images.extend(pexels_images)
                if len(all_images) >= num_images: break

        if len(all_images) < num_images:
            for query in queries:
                if query and not any(f"{query} Pakistan" in q for q in [img['title'] for img in all_images]):
                    pexels_images = self.search_pexels(query, num_images - len(all_images))
                    all_images.extend(pexels_images)
                    if len(all_images) >= num_images: break

        if len(all_images) < num_images:
            print(f"Insufficient results, generating placeholders.")
            placeholder_images = self.search_lorem_picsum(queries[0] if queries else '', num_images=num_images - len(all_images))
            all_images.extend(placeholder_images)

        return all_images[:num_images]

def create_image_queries(travel_data: Dict, destination: Optional[str]) -> List[str]:
    queries = []
    location_type = travel_data.get("location_type", "").lower() if travel_data.get("location_type") != "Other" else ""
    province = travel_data.get("province", "").lower().replace("-", " ") if travel_data.get("province") else ""
    preferences = travel_data.get("preferences", [])
    preference_map = {
        "Cultural Experience": "culture", "Food and Dining": "food",
        "Shopping": "market", "Nature": "nature", "History": "historical"
    }
    pref_terms = [preference_map.get(pref, pref.lower()) for pref in preferences if pref in preference_map]

    if destination and destination != "Suggested Destination": queries.append(destination)
    if destination and destination != "Suggested Destination" and location_type: queries.append(f"{destination} {location_type}")
    if destination and destination != "Suggested Destination" and pref_terms: queries.append(f"{destination} {' '.join(pref_terms)}")
    if destination and destination != "Suggested Destination" and location_type and pref_terms: queries.append(f"{destination} {location_type} {' '.join(pref_terms)}")
    if province and location_type: queries.append(f"{province} {location_type}")
    if province and pref_terms: queries.append(f"{province} {' '.join(pref_terms)}")
    if province: queries.append(province)
    if location_type and pref_terms: queries.append(f"{location_type} {' '.join(pref_terms)}")
    if location_type: queries.append(location_type)

    queries = list(dict.fromkeys([q for q in queries if q]))
    if not queries: queries.append("Pakistan travel")
    print(f"Created image search queries: {queries}")
    return queries

def extract_destination(plan_text: str, travel_data: Dict) -> Optional[str]:
    lines = plan_text.split('\n')
    for line in lines:
        if line.strip().startswith('- Destination:'):
            parts = line.split(':', 1)
            if len(parts) > 1:
                destination = parts[1].strip()
                print(f"Extracted destination from plan: {destination}")
                return destination

    if travel_data.get("destination"): return travel_data["destination"]
    if travel_data.get("additional_preferences"): return " ".join(travel_data["additional_preferences"].split()[:3])
    if travel_data.get("province"): return travel_data["province"]
    return None