Add Ctrl-C handling to gracefully exit training early

- Wrap trainer.train() in try-catch to handle KeyboardInterrupt
- Allow users to stop training and proceed to text generation demo
- Add clear instruction about Ctrl-C functionality for better UX
- Ensure text generation always runs regardless of training completion

Author: dariocazzani

examples/simple.py

@@ -141,7 +141,13 @@ def main():
         )
 
         print("\nStarting training...")
-        trainer.train(data=data_source, tokenizer=tokenizer)
+        print("(Press Ctrl-C to stop training early and see text generation)")
+
+        try:
+            trainer.train(data=data_source, tokenizer=tokenizer)
+            print("\nTraining completed successfully!")
+        except KeyboardInterrupt:
+            print("\n\nTraining interrupted by user. Moving to text generation with current model state...")
 
         # Step 6: Simple text generation demo
         print("\nTesting text generation:")
@@ -162,7 +168,7 @@ def main():
                 result = tokenizer.decode(generated[0].tolist())
                 print(f"Generated: {result}")
 
-        print("\nTraining complete! All temporary files automatically cleaned up.")
+        print("\nAll temporary files automatically cleaned up.")
         print("Run the script again to start fresh.")