Update openai_client.py

openai_client.py

@@ -1,19 +1,18 @@
 import os
 import google.generativeai as genai
 
+# Configure Gemini client
+genai.configure(api_key=os.environ["GEMINI_API_KEY"])
 
-# Initialize Gemini client using the correct environment variable
-api_key = os.getenv("GEMINI_API_KEY")
-if not api_key:
-    raise ValueError("Missing GEMINI_API_KEY environment variable")
+# ✅ Use your custom Gemini model
+model = genai.GenerativeModel("gemini-2.0-flash-thinking-exp")
 
-client = genai.Client(api_key=api_key)
-
-# Default model (same as used in the full system)
-model_name = "gemini-2.0-flash-thinking-exp"
-
-def ask_gpt(messages: list[dict]) -> str:
-    prompt_parts = "\n".join([msg["content"] for msg in messages])
-    chat = client.chats.create(model=model_name)
-    resp = chat.send_message(prompt_parts)
-    return resp.text
+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()
+    except Exception as e:
+        return f"⚠️ Gemini Error: {str(e)}"