import os
import google.generativeai as 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")

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)}"