import os
import google.generativeai as genai

class BasicAgent:
    def __init__(self):
        api_key = os.getenv("GOOGLE_API_KEY")
        if not api_key:
            raise ValueError("GOOGLE_API_KEY environment variable is missing. Please ensure it is set in Hugging Face Secrets.")
        genai.configure(api_key=api_key)
        self.model = genai.GenerativeModel('gemini-2.5-flash')
        self.sessions = {}

    def generate_response(self, session_id, user_message):
        if session_id not in self.sessions:
            self.sessions[session_id] = []
            
        history = self.sessions[session_id]
        
        prompt = "Conversation History:\n"
        for msg in history:
            prompt += f"User: {msg['user']}\nAI: {msg['ai']}\n"
        prompt += f"User: {user_message}\nAI:"
        
        try:
            response = self.model.generate_content(prompt)
            ai_text = response.text
        except Exception as e:
            ai_text = f"Error generating response: {e}"

        self.sessions[session_id].append({"user": user_message, "ai": ai_text})
        return ai_text