test_engine.py

from ai_engine import generate

history = []

while True:
    prompt = input("\nYou: ")
    response = generate(prompt, history)

    print(f"AI: {response}")
    
    history.append(f"User: {prompt}")
    history.append(f"AI: {response}")

    print("\nHistory so far:")
    for line in history:
        print(line)