app/utils/handle_api.py

import os
import getpass
from dotenv import load_dotenv

def get_api(api_name="OPENAI_API_KEY"):
    load_dotenv()
    api_key = os.getenv(api_name)
    if api_key == None:
        api_key = getpass.getpass(f"Enter API key for {api_name}: ")
    if api_key == None:
        raise ValueError(f"API key is required. Please set it in the environment variable {api_name} or provide it directly.")
    return api_key