Update agent.py

agent.py

@@ -4,11 +4,15 @@ import os
 import google.generativeai as genai
 from dotenv import load_dotenv
 
-# Load environment variables from .env
+# Load env file (works locally, ignored on Hugging Face if no .env present)
 load_dotenv()
 
 # Configure Gemini API
-genai.configure(api_key=os.getenv("AGENT_API_KEY"))
+api_key = os.getenv("AGENT_API_KEY")
+if not api_key:
+    raise ValueError("❌ No API key found! Please set AGENT_API_KEY in .env or Hugging Face Secrets.")
+
+genai.configure(api_key=api_key)
 
 # Initialize Gemini model
 model = genai.GenerativeModel("models/gemini-1.5-flash-latest")
@@ -16,13 +20,9 @@ model = genai.GenerativeModel("models/gemini-1.5-flash-latest")
 def generate_response(conversation):
     """
     Generate a response from Gemini.
-    conversation: list of (role, message) tuples.
-    Only 'user' role is supported.
+    conversation: list of (role, message) tuples
+    Only 'user' role supported.
     """
-    # Convert conversation to Gemini-compatible format
     messages = [{"role": "user", "parts": msg} for role, msg in conversation if role == "user"]
-
-    # Call Gemini directly (no system/model roles, no start_chat)
     response = model.generate_content(messages)
-
     return response.text.strip()