from core.memory import Memory
from core.router import route
from core.module_writer import ModuleWriter

class Brain:
    def __init__(self):
        self.memory = Memory()
        self.writer = ModuleWriter()

    def think(self, text):
        context = self.memory.retrieve(text)
        output = route(text, context)
        # Optional: auto-generate module if keyword present
        if "create module" in text.lower():
            name = text.split()[-1]
            template = f"def {name}():\n    return 'This is auto-generated module {name}'\n"
            return self.writer.create_module(name, template)
        
        if "improve modules" in text.lower():
            updated = []
            for m in self.writer.list_modules():
                path = os.path.join(self.writer.dir, m)
                with open(path, "a") as f:
                    f.write("\n# [auto-enhanced]\n")
                updated.append(m)
            return f"Modules improved: {', '.join(updated)}"
        
        return output