Spaces:
No application file
No application file
| import aiohttp | |
| import bot_config | |
| import uuid | |
| async def get_suggestions(query): | |
| if not bot_config.SUGGESTIONS_API_KEY: | |
| return None | |
| suggestions = [] | |
| headers = { | |
| 'Accept-Language': 'en-US,en;q=1', | |
| 'User-Agent': 'Mozilla/5.0 (X11; Linux x86_64; rv:109.0) Gecko/20100101 Firefox/111.0 BingSapphire/24.1.410310303' | |
| } | |
| async with aiohttp.ClientSession(headers=headers) as session: | |
| async with session.get(f'https://www.bingapis.com/api/v7/suggestions?appid={bot_config.SUGGESTIONS_API_KEY}&etmkt=en-US&q={query}&lang=en-US,en;q=0.9') as response: | |
| if response.status != 200: | |
| return None | |
| js = await response.json() | |
| for suggestionGroups in js['suggestionGroups']: | |
| for searchSuggestion in suggestionGroups['searchSuggestions']: | |
| if 'ghostText' in searchSuggestion.keys(): | |
| suggestions.append( | |
| {'id': str(uuid.uuid4()), 'query': searchSuggestion['ghostText']}) | |
| else: | |
| suggestion = searchSuggestion['query'] | |
| query_split = query.split() | |
| if suggestion.startswith(query_split[-1]) and len(query_split) > 1: | |
| suggestion = ' '.join( | |
| query_split[:-1]) + ' ' + suggestion | |
| else: | |
| suggestions.append( | |
| {'id': str(uuid.uuid4()), 'query': suggestion}) | |
| return suggestions | |