Update gemini_client.py

gemini_client.py

@@ -1,18 +1,20 @@
 import os
-import google.generativeai as genai
+from google import genai
 
-# Configure Gemini client
-genai.configure(api_key=os.environ["GEMINI_API_KEY"])
-
-# ✅ Use your custom Gemini model
-model = genai.GenerativeModel("gemini-2.0-flash-thinking-exp")
+client = genai.Client(api_key=os.environ["GEMINI_API_KEY"])
 
 def ask_gpt(messages):
     try:
         prompt = "\n".join(
             [f"{m['role'].capitalize()}: {m['content']}" for m in messages]
         )
-        response = model.generate_content(prompt)
-        return response.text.strip()
+
+        response = client.models.generate_content(
+            model="gemini-2.5-flash",
+            contents=prompt,
+        )
+
+        return (response.text or "").strip()
+
     except Exception as e:
-        return f"⚠️ Gemini Error: {str(e)}"
+        return f"⚠️ Gemini Error: {str(e)}"