utils/pokemon.py

import requests
def get_pokemon_image(pokemon_name,extension="png"):
    url = f'https://pokeapi.co/api/v2/pokemon/{pokemon_name.lower()}'
    response = requests.get(url)
    if response.status_code == 200:
        data = response.json()
        image_url = data['sprites']['front_default']
        id=image_url.split("/")[-1]
        id=id.split(".")[0]
        if(extension=="png"):
            return f"https://raw.githubusercontent.com/PokeAPI/sprites/master/sprites/pokemon/other/official-artwork/{id}.png"
        elif(extension=="gif"):
            return f"https://raw.githubusercontent.com/PokeAPI/sprites/master/sprites/pokemon/other/showdown/{id}.gif"
    else:
        return None