model/utils.py

import os
import google.generativeai as genai
from model.prompts import chatbot_prompt
from settings.base import setup_logger

logger = setup_logger()

def create_model(api_key, profile):
    """
    Create a GenerativeModel instance using the provided API key.
    """
    generation_config = {
    "temperature": 0.8,
    "top_p": 0.95,
    "top_k": 64,
    "max_output_tokens": 8192,
    "response_mime_type": "text/plain",
    }

    try:
        genai.configure(api_key=api_key)
        model = genai.GenerativeModel(
                    model_name="gemini-1.5-pro",
                    generation_config=generation_config,
                    system_instruction= chatbot_prompt(profile),
                    )
        logger.info("Model created successfully")
        return model
    except Exception as e:
        logger.error(f"Error creating model: {e}")
        return None


def generate_content(model, message):
    """
    Send a message to the chat session and get the response.
    """
    try:
        # msg = f"Query:{message}/n/n Every time you provide something please confirm from the user if the identified details are correct. If they are correct, suggest the next healthy meal with the detailed recipe. If any details are incorrect, please ask for additional information."
        response = model.generate_content(message)
        logger.info(f"Generated content for message")
        return response.text
    except Exception as e:
        logger.error(f"Error generating content for message: {e}")
        return f"Error generating content: {e}"