1580158
/
Auto-Photoshop-StableDiffusion-Plugin
mirror of https://github.com/AbdullahAlfaraj/Auto-Photoshop-StableDiffusion-Plugin.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

recode image search to use AsyncDDGS

pull/378/head
deedy5 2023-10-01 09:46:03 +03:00 committed by Abdullah Alfaraj
parent 7aeed21c5f
commit 0a2470b7b7
1 changed files with 13 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
server/python_server/search.py
View File

@ -1,7 +1,7 @@
from itertools import islice import asyncio
try: try:
from duckduckgo_search import DDGS from duckduckgo_search import AsyncDDGS
except ImportError: except ImportError:
raise ImportError( raise ImportError(
"duckduckgo_search is required to image search. Please install it with `pip install --upgrade duckduckgo_search`." "duckduckgo_search is required to image search. Please install it with `pip install --upgrade duckduckgo_search`."
@ -9,17 +9,18 @@ except ImportError:
async def imageSearch(keywords="cute cats"): async def imageSearch(keywords="cute cats"):
with DDGS() as ddgs: async with AsyncDDGS() as ddgs:
return [x for x in islice(ddgs.images(keywords,safesearch='off'), 50)] return [
x async for x in ddgs.images(keywords, safesearch="off", max_results=50)
]
async def main():
result = await imageSearch()
print("result: ", result)
# result = await imageSearch2()
# print(result)
if __name__ == "__main__": if __name__ == "__main__":
async def main():
result = await imageSearch()
print("result: ",result)
# result = await imageSearch2()
# print(result)
import asyncio
asyncio.run(main()) asyncio.run(main())