Update openai_client.py

openai_client.py

@@ -1,27 +1,16 @@
 import os
 import google.generativeai as genai
 
-# 🔐 Authenticate
 genai.configure(api_key=os.environ["GEMINI_API_KEY"])
 
-# ✅ Recommended way to use Gemini
-model = genai.GenerativeModel(model_name="models/gemini-pro")
+# ✅ Use the correct model name for v1
+model = genai.GenerativeModel("gemini-pro")  # no "models/" prefix needed
 
 def ask_gpt(messages):
     try:
-        prompt_parts = []
-
-        for m in messages:
-            role = m["role"]
-            content = m["content"]
-
-            if role == "user":
-                prompt_parts.append(f"User: {content}")
-            elif role == "assistant":
-                prompt_parts.append(f"Assistant: {content}")
-
-        prompt = "\n".join(prompt_parts)
-
+        prompt = "\n".join(
+            [f"{m['role'].capitalize()}: {m['content']}" for m in messages]
+        )
         response = model.generate_content(prompt)
         return response.text.strip()
     except Exception as e: