from knowledge_base import KnowledgeBase
from datetime import datetime
import random

class ResponseEngine:
    def __init__(self):
        self.kb = KnowledgeBase()

    def format_response(self, user_input: str) -> str:
        answer = self.kb.retrieve_answer(user_input)

        if "don't know" in answer.lower():
            fallback = [
                "I'm not sure about that yet, but I'm learning!",
                "That's a great question. I'll need to look into it.",
                "I don't have that information, but I'll try to find out."
            ]
            return random.choice(fallback)

        timestamp = datetime.now().strftime('%Y-%m-%d %H:%M')
        return f"[{timestamp}] 🤖 Based on what I know: {answer}"

    def close(self):
        self.kb.close()