import os
import google.generativeai as genai
from .utils.memory_manager import PostgresMemoryManager
from .utils.prompt_manager import PromptManager

class EnhancedAgent:
    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.memory = PostgresMemoryManager()
        self.prompt_manager = PromptManager()

    def generate_response(self, session_id, user_message, context=""):
        history = self.memory.get_history(session_id)
        
        prompt = self.prompt_manager.format_prompt(user_message, context, history)
        
        try:
            response = self.model.generate_content(prompt)
            ai_text = response.text
        except Exception as e:
            ai_text = f"Error generating response: {e}"

        self.memory.add_message(session_id, user_message, ai_text)
        return ai_text