service/gemini_service.py

def prompt_gemini(prompt: str):
    """
    :param prompt:
    :return:
    """
    import os
    gemini_api_key: str = os.environ.get("GEMINI_API_KEY")
    if not gemini_api_key:
        print("\033[91mYou need an API key to download Gemini models.\033[0m")

    return prompt