Update openai_client.py

openai_client.py

@@ -1,24 +1,26 @@
 import os
 import google.generativeai as genai
 
-# 🔐 Load API key
+# 🔐 Authenticate
 genai.configure(api_key=os.environ["GEMINI_API_KEY"])
 
-# 🌟 Use Gemini Pro
-model = genai.GenerativeModel("gemini-pro")
+# ✅ Recommended way to use Gemini
+model = genai.GenerativeModel(model_name="models/gemini-pro")
 
 def ask_gpt(messages):
-    """
-    messages: [
-        {"role": "user", "content": "hello"},
-        {"role": "assistant", "content": "hi there!"}
-    ]
-    """
     try:
-        formatted = [
-            m["content"] for m in messages if m["role"] == "user"
-        ]
-        prompt = "\n\n".join(formatted)
+        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)
 
         response = model.generate_content(prompt)
         return response.text.strip()