
import sys
import os

# Add src to path
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))

from src.core.grammar_assistant import GrammarAssistant

def test_punctuation():
    assistant = GrammarAssistant()
    assistant.load_model()
    
    samples = [
        # User's example (verbatim)
        "If the voice recognition doesn't recognize that I like stopped Or something would that would it also correct that",
        
        # Generic run-on
        "hello how are you doing today i am doing fine thanks for asking",
        
        # Missing commas/periods
        "well i think its valid however we should probably check the logs first"
    ]
    
    print("\nStarting Punctuation Tests:\n")
    
    for sample in samples:
        print(f"Original: {sample}")
        corrected = assistant.correct(sample)
        print(f"Corrected: {corrected}")
        print("-" * 20)

if __name__ == "__main__":
    test_punctuation()
