Skip to content

Commit 669ab76

Browse files
lmazuellmazuel
committed
Add ImageSearch sample
1 parent f6d722b commit 669ab76

File tree

1 file changed

+201
-0
lines changed

1 file changed

+201
-0
lines changed

‎samples/search/image_search_samples.py‎

Lines changed: 201 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,201 @@
1+
from azure.cognitiveservices.search.imagesearch import ImageSearchAPI
2+
from azure.cognitiveservices.search.imagesearch.models import ImageType, ImageAspect, ImageInsightModule
3+
from msrest.authentication import CognitiveServicesCredentials
4+
5+
SUBSCRIPTION_KEY_ENV_NAME = "IMAGESEARCH_SUBSCRIPTION_KEY"
6+
7+
def image_search(subscription_key):
8+
"""ImageSearch.
9+
10+
This will search images for (canadian rockies) then verify number of results and print out first image result, pivot suggestion, and query expansion.
11+
"""
12+
client = ImageSearchAPI(CognitiveServicesCredentials(subscription_key))
13+
14+
try:
15+
image_results = client.images.search(query="canadian rockies")
16+
print("Search images for query \"canadian rockies\"")
17+
18+
# Image results
19+
if image_results.value:
20+
first_image_result = image_results.value[0]
21+
print("Image result count: {}".format(len(image_results.value)))
22+
print("First image insights token: {}".format(first_image_result.image_insights_token))
23+
print("First image thumbnail url: {}".format(first_image_result.thumbnail_url))
24+
print("First image content url: {}".format(first_image_result.content_url))
25+
else:
26+
print("Couldn't find image results!")
27+
28+
print("Image result total estimated matches: {}".format(image_results.total_estimated_matches))
29+
print("Image result next offset: {}".format(image_results.next_offset))
30+
31+
# Pivot suggestions
32+
if image_results.pivot_suggestions:
33+
first_pivot = image_results.pivot_suggestions[0]
34+
print("Pivot suggestion count: {}".format(len(image_results.pivot_suggestions)))
35+
print("First pivot: {}".format(first_pivot.pivot))
36+
37+
if first_pivot.suggestions:
38+
first_suggestion = first_pivot.suggestions[0]
39+
print("Suggestion count: {}".format(len(first_pivot.suggestions)))
40+
print("First suggestion text: {}".format(first_suggestion.text))
41+
print("First suggestion web search url: {}".format(first_suggestion.web_search_url))
42+
else:
43+
print("Couldn't find suggestions!")
44+
else:
45+
print("Couldn't find pivot suggestions!")
46+
47+
# Query expansions
48+
if image_results.query_expansions:
49+
first_query_expansion = image_results.query_expansions[0]
50+
print("Query expansion count: {}".format(len(image_results.query_expansions)))
51+
print("First query expansion text: {}".format(first_query_expansion.text))
52+
print("First query expansion search link: {}".format(first_query_expansion.search_link))
53+
else:
54+
print("Couldn't find image results!")
55+
56+
except Exception as err:
57+
print("Encountered exception. {}".format(err))
58+
59+
60+
def image_search_with_filters(subscription_key):
61+
"""ImageSearchWithFilters.
62+
63+
This will search images for (studio ghibli), filtered for animated gifs and wide aspect, then verify number of results and print out insightsToken, thumbnail url and web url of first result.
64+
"""
65+
client = ImageSearchAPI(CognitiveServicesCredentials(subscription_key))
66+
67+
try:
68+
image_results = client.images.search(
69+
query="studio ghibli",
70+
image_type=ImageType.animated_gif, # Could be the str "AnimatedGif"
71+
aspect=ImageAspect.wide # Could be the str "Wide"
72+
)
73+
print("Search images for \"studio ghibli\" results that are animated gifs and wide aspect")
74+
75+
if image_results.value:
76+
first_image_result = image_results.value[0]
77+
print("Image result count: {}".format(len(image_results.value)))
78+
print("First image insights token: {}".format(first_image_result.image_insights_token))
79+
print("First image thumbnail url: {}".format(first_image_result.thumbnail_url))
80+
print("First image web search url: {}".format(first_image_result.web_search_url))
81+
else:
82+
print("Couldn't find image results!")
83+
84+
except Exception as err:
85+
print("Encountered exception. {}".format(err))
86+
87+
88+
def image_trending(subscription_key):
89+
"""ImageTrending.
90+
91+
This will search for trending images then verify categories and tiles.
92+
"""
93+
client = ImageSearchAPI(CognitiveServicesCredentials(subscription_key))
94+
95+
try:
96+
trending_result = client.images.trending()
97+
print("Search trending images")
98+
99+
# Categorires
100+
if trending_result.categories:
101+
first_category = trending_result.categories[0]
102+
print("Category count: {}".format(len(trending_result.categories)))
103+
print("First category title: {}".format(first_category.title))
104+
if first_category.tiles:
105+
first_tile = first_category.tiles[0]
106+
print("Subcategory tile count: {}".format(len(first_category.tiles)))
107+
print("First tile text: {}".format(first_tile.query.text))
108+
print("First tile url: {}".format(first_tile.query.web_search_url))
109+
else:
110+
print("Couldn't find subcategory tiles!")
111+
else:
112+
print("Couldn't find categories!")
113+
114+
except Exception as err:
115+
print("Encountered exception. {}".format(err))
116+
117+
118+
def image_detail(subscription_key):
119+
"""ImageDetail.
120+
121+
This will search images for (degas) and then search for image details of the first image.
122+
"""
123+
client = ImageSearchAPI(CognitiveServicesCredentials(subscription_key))
124+
125+
try:
126+
image_results = client.images.search(query="degas")
127+
print("Search images for \"degas\"")
128+
129+
first_image = image_results.value[0]
130+
131+
image_detail = client.images.details(
132+
query="degas",
133+
insights_token=first_image.image_insights_token,
134+
modules=[
135+
ImageInsightModule.all # Could be the str "all"
136+
],
137+
138+
)
139+
print("Search detail for image insights token: {}".format(first_image.image_insights_token))
140+
print("Expected image insights token: {}".format(image_detail.image_insights_token))
141+
142+
# Best representative query
143+
if image_detail.best_representative_query:
144+
print("Best representative query text: {}".format(image_detail.best_representative_query.text))
145+
print("Best representative query web search url: {}".format(image_detail.best_representative_query.web_search_url))
146+
else:
147+
print("Couldn't find best representative query!")
148+
149+
# Caption
150+
if image_detail.image_caption:
151+
print("Image caption: {}".format(image_detail.image_caption.caption))
152+
print("Image caption data source url: {}".format(image_detail.image_caption.data_source_url))
153+
else:
154+
print("Couldn't find image caption!")
155+
156+
# Pages including the image
157+
if image_detail.pages_including.value:
158+
first_page = image_detail.pages_including.value[0]
159+
print("Pages including cound: {}".format(len(image_detail.pages_including.value)))
160+
print("First page content url: {}".format(first_page.content_url))
161+
print("First page name: {}".format(first_page.name))
162+
print("First page date published: {}".format(first_page.date_published))
163+
else:
164+
print("Couldn't find any pages including this image!")
165+
166+
# Related searched
167+
if image_detail.related_searches.value:
168+
first_related_search = image_detail.related_searches.value[0]
169+
print("Related searches count: {}".format(len(image_detail.related_searches.value)))
170+
print("First related search text: {}".format(first_related_search.text))
171+
print("First related search web search url: {}".format(first_related_search.web_search_url))
172+
else:
173+
print("Couldn't find any related searches!")
174+
175+
# Visually similar images
176+
if image_detail.visually_similar_images.value:
177+
first_visually_similar_images = image_detail.visually_similar_images.value[0]
178+
print("Visually similar images count: {}".format(len(image_detail.visually_similar_images.value)))
179+
print("First visually similar image name: {}".format(first_visually_similar_images.name))
180+
print("First visually similar image content url: {}".format(first_visually_similar_images.content_url))
181+
print("First visually similar image content size: {}".format(first_visually_similar_images.content_size))
182+
else:
183+
print("Couldn't find any visually similar images!")
184+
185+
# Image tags:
186+
if image_detail.image_tags.value:
187+
first_image_tag = image_detail.image_tags.value[0]
188+
print("Image tags count: {}".format(len(image_detail.image_tags.value)))
189+
print("First tag name: {}".format(first_image_tag.name))
190+
else:
191+
print("Couldn't find any image tags!")
192+
193+
except Exception as err:
194+
print("Encountered exception. {}".format(err))
195+
196+
197+
if __name__ == "__main__":
198+
import sys, os.path
199+
sys.path.append(os.path.abspath(os.path.join(__file__, "..", "..")))
200+
from tools import execute_samples
201+
execute_samples(globals(), SUBSCRIPTION_KEY_ENV_NAME)

0 commit comments

Comments
 (0)