ai_services.py

from openai import OpenAI

from dotenv import load_dotenv
import os

if os.getenv("OPENAI_API_KEY") is None:
    load_dotenv()

openAIclient = OpenAI()

def get_suggestion_text(closestReadBookData, targetBookData):    
    closestReadTitle = closestReadBookData['title']
    closestReadAuthor = closestReadBookData['authors']

    targetBookTitle = targetBookData['title']
    targetBookAuthor = targetBookData['authors']

    messageContent = "In 2 sentences max, please cheerfully explain why I might enjoy " + targetBookTitle + " by " + targetBookAuthor + " if I liked " + closestReadTitle + " by " + closestReadAuthor + "."

    response = openAIclient.chat.completions.create(
        model="gpt-4o-mini",
        messages=[
            {"role": "system", "content": "You are an experienced librarian."},
            {"role": "user", "content": messageContent}
        ]
    )

    chatCompletionMessage = response.choices[0].message    

    # return f"Because you liked {closestReadTitle} by {closestReadAuthor}, we think you might like {targetBookTitle} by {targetBookAuthor}."
    return chatCompletionMessage.content