import os
import google.generativeai as genai

def get_gemini_api_key():
    return os.getenv("GEMINI_API_KEY")

def run_gemini_chat(prompt, context=None):
    api_key = get_gemini_api_key()
    if not api_key:
        return "Error: Gemini API key not set. Please add it to your environment variables."
    genai.configure(api_key=api_key)
    model = genai.GenerativeModel("gemini-pro")
    chat = model.start_chat()
    if context:
        prompt = f"{context}\n\n{prompt}"
    try:
        response = chat.send_message(prompt)
        return response.text
    except Exception as e:
        return f"Gemini Error: {e}"