Update agent/gemini_agent.py

agent/gemini_agent.py

@@ -1,20 +1,10 @@
-import os
-import google.generativeai as genai
+import os, google.generativeai as genai
+from config.settings import settings
 
-def get_gemini_api_key():
-    return os.getenv("GEMINI_API_KEY")
+genai.configure(api_key=settings.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)
+def chat_with_gemini(user, prompt):
     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}"
+    resp = chat.send_message(prompt)
+    return resp.text