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