Update openai_client.py

openai_client.py

@@ -1,12 +1,26 @@
 import os
-from openai import OpenAI
+import google.generativeai as genai
 
-client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
+# 🔐 Load API key
+genai.configure(api_key=os.environ["GEMINI_API_KEY"])
+
+# 🌟 Use Gemini Pro
+model = genai.GenerativeModel("gemini-pro")
 
 def ask_gpt(messages):
-    response = client.chat.completions.create(
-        model="gpt-3.5-turbo",
-        messages=messages,
-        temperature=0.6
-    )
-    return response.choices[0].message.content
+    """
+    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)
+
+        response = model.generate_content(prompt)
+        return response.text.strip()
+    except Exception as e:
+        return f"⚠️ Gemini Error: {str(e)}"