# Install:
# pip install transformers torch pillow duckduckgo-search

from transformers import pipeline
from PIL import Image
from duckduckgo_search import DDGS

print("😊 Starting FaceMatch AI...")

# AI that describes images
vision_ai = pipeline(
    "image-to-text",
    model="Salesforce/blip-image-captioning-base"
)

def search_face_image(photo_path):
    print("\n😊 Hey! Let's look at your image.")

    try:
        image = Image.open(photo_path).convert("RGB")

        print("🔎 Analyzing the image...")
        description = vision_ai(image)[0]["generated_text"]

        print(f"🧠 Image details: {description}")
        print("🌐 Searching for related public webpages...")

        with DDGS() as search:
            results = list(search.text(
                description,
                max_results=10
            ))

        if results:
            print("\n😊 Yay! I found some relevant search results!")

            for i, result in enumerate(results, 1):
                print(f"\n📄 Result {i}: {result['title']}")
                print(f"🔗 Link: {result['href']}")
                print(f"📝 {result['body']}")
        else:
            print("🙂 No results this time. Let's try another image.")

    except Exception as error:
        print(f"😕 Something went wrong: {error}")

if __name__ == "__main__":
    photo = input("📷 Enter the path to your image: ")
    search_face_image(photo)