models.py

from openai import OpenAI
from config import OPENAI_API_KEY

# Initialize OpenAI client
client = OpenAI(api_key=OPENAI_API_KEY)

def query_openai_text(system_message, user_message, model="gpt-4o", temperature=0.3):
    """
    Query OpenAI's Chat API for a response to medical questions or tasks.
    """
    try:
        response = client.chat.completions.create(
            model=model,
            messages=[
                {"role": "system", "content": system_message},
                {"role": "user", "content": user_message},
            ],
            temperature=temperature,
        )
        return response.choices[0].message.content.strip()
    except Exception as e:
        return f"Error querying OpenAI API: {str(e)}"