import os
from google import genai

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