llm_api/connect.py

from meal_planner_api import utils
import google.generativeai as genai
from meal_planner_api import config


def generate_content(text: str):
    model = connect()
    response = model.generate_content(text)
    return response.text


def connect():
    genai.configure(api_key=config.API_KEY)
    model = genai.GenerativeModel('gemini-pro')
    return model