84f10092e9
- Removed M2M100 Grammar Correction model completely to reduce bloat/complexity. - Implemented 'Style Prompting' in Settings -> AI Engine to handle punctuation natively via Whisper. - Added Style Presets: Standard (Default), Casual, and Custom. - Optimized Build: Bootstrapper no longer requires transformers/sentencepiece. - Fixed 'torch' NameError in Low VRAM mode. - Fixed Bootstrapper missing dependency detection. - Updated UI to reflect removed features. - Included compiled v1.0.2 Executable in dist/.
35 lines
949 B
Python
35 lines
949 B
Python
|
|
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()
|