PicImageSearch/demo/code/lenso.py at main · kitUIN/PicImageSearch · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
"""
Warning: The Lenso engine is deprecated as the website now uses Cloudflare turnstile protection,
which prevents this client from working properly.
"""

import asyncio

from demo.code.config import IMAGE_BASE_URL, PROXIES, get_image_path, logger
from PicImageSearch import Lenso, Network
from PicImageSearch.model import LensoResponse
from PicImageSearch.sync import Lenso as LensoSync

url = f"{IMAGE_BASE_URL}/test08.jpg"
file = get_image_path("test08.jpg")


@logger.catch()
async def demo_async() -> None:
    async with Network(proxies=PROXIES) as client:
        lenso = Lenso(client=client)
        resp = await lenso.search(file=file)
        show_result(resp)


@logger.catch()
def demo_sync() -> None:
    lenso = LensoSync(proxies=PROXIES)
    resp = lenso.search(file=file)
    show_result(resp)  # pyright: ignore[reportArgumentType]


def show_result(resp: LensoResponse, search_type: str = "") -> None:
    logger.info(f"Search Type: {search_type}")
    logger.info(f"Search URL: {resp.url}")

    result_lists = {
        "duplicates": resp.duplicates,
        "similar": resp.similar,
        "places": resp.places,
        "related": resp.related,
        "people": resp.people,
    }

    for result_type, items in result_lists.items():
        if not items:
            logger.info(f"--- No {result_type} Results ---")
            continue

        logger.info(f"--- {result_type} Results ---")
        for item in items:
            logger.info(f"  Hash: {item.hash}")
            logger.info(f"  Similarity: {item.similarity}")
            logger.info(f"  Thumbnail URL: {item.thumbnail}")
            logger.info(f"  Size: {item.width}x{item.height}")
            logger.info(f"  URL: {item.url}")
            logger.info(f"  Title: {item.title}")

            if item.url_list:
                logger.info("  URLs:")
                for url_item in item.url_list:
                    logger.info(f"    > Image URL: {url_item.image_url}")
                    logger.info(f"    > Source URL: {url_item.source_url}")
                    logger.info(f"    > Title: {url_item.title}")
                    logger.info(f"    > Lang: {url_item.lang}")
            logger.info("-" * 20)

    logger.info("=" * 50)


if __name__ == "__main__":
    asyncio.run(demo_async())
    # demo_sync()