Create agent/gemini_agent.py

agent/gemini_agent.py

@@ -0,0 +1,20 @@
+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}"