Update models.py

models.py

@@ -1,19 +1,22 @@
 from openai import OpenAI
-from config import OPENAI_API_KEY, OPENAI_DEFAULT_MODEL
+from config import OPENAI_API_KEY
 
 # Initialize OpenAI client
 client = OpenAI(api_key=OPENAI_API_KEY)
 
-def chat_with_openai(system_message, user_message, model=OPENAI_DEFAULT_MODEL, temperature=0.3):
+def query_openai_text(system_message, user_message, model="gpt-4o", temperature=0.3):
     """
-    Use OpenAI's Chat API to generate responses based on system and user messages.
+    Query OpenAI's Chat API for a response to medical questions or tasks.
     """
-    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()
+    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)}"