Update models.py

models.py

@@ -2,20 +2,20 @@ from openai import OpenAI
 import os
 from config import OPENAI_API_KEY, OPENAI_DEFAULT_MODEL
 
-# Initialize OpenAI client
+# Initialize OpenAI
 client = OpenAI(api_key=OPENAI_API_KEY)
 
-
-def chat_with_openai(system_message, user_message, model=OPENAI_DEFAULT_MODEL, temperature=0.3):
+def chat_with_openai(system_prompt, user_prompt, model=OPENAI_DEFAULT_MODEL, temperature=0.3):
     """
-    Chat with OpenAI's model using a system message and user message.
+    Orchestrates a conversation with OpenAI's Chat API, 
+    combining system + user messages for a final answer.
     """
     response = client.chat.completions.create(
         model=model,
         messages=[
-            {"role": "system", "content": system_message},
-            {"role": "user", "content": user_message},
+            {"role": "system", "content": system_prompt},
+            {"role": "user", "content": user_prompt}
         ],
         temperature=temperature,
     )
-    return response.choices[0].message.content
+    return response.choices[0].message.content.strip()