from aggpt14 import AgGPT14
from training_data import corpus

if __name__ == "__main__":
    model = AgGPT14(corpus, order=4, seed=None)
    
    print("Chat with AgGPT14 (type 'quit' to exit)")
    print("-" * 40)
    
    while True:
        prompt = input("You: ")
        
        if prompt.lower() in ['quit', 'exit', 'q']:
            print("Goodbye!")
            break
            
        response = model.ask(prompt, max_tokens=999999, temperature=0.5, top_k=5, text_world_model=False)
        print(f"AI:  {response}\n")