agent.py

# chatbot/agent.py

import os
import google.generativeai as genai
from dotenv import load_dotenv

# Load env file (works locally, ignored on Hugging Face if no .env present)
load_dotenv()

# Configure Gemini API
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")

def generate_response(conversation):
    """
    Generate a response from Gemini.
    conversation: list of (role, message) tuples
    Only 'user' role supported.
    """
    messages = [{"role": "user", "parts": msg} for role, msg in conversation if role == "user"]
    response = model.generate_content(messages)
    return response.text.strip()