Spaces:
Sleeping
Sleeping
| import gradio as gr | |
| from difflib import SequenceMatcher | |
| import string | |
| def load_dictionary(path=r'Dictionary.txt'): | |
| with open(path, 'r') as file: | |
| words = [line.strip() for line in file if line.strip()]# Strips whitespace and ignores empty lines | |
| return words | |
| def preprocess_input(input_text): | |
| input_text = input_text.lower() | |
| abbreviations_dict = { | |
| "can't": "cannot", "won't": "will not", "'re": " are", "'s": " is", | |
| "'d": " would", "'ll": " will", "'ve": " have", "'m": " am", | |
| "I'm": "I am", "i'm": "i am", "it's": "it is", "he's": "he is", | |
| "she's": "she is", "we're": "we are", "they're": "they are", | |
| "there's": "there is", "who's": "who is", "that's": "that is", | |
| "what's": "what is", "let's": "let us", "i've": "I have", | |
| "we've": "we have", "they've": "they have", "should've": "should have", | |
| "could've": "could have", "would've": "would have", "you've": "you have", | |
| "he'd": "he would", "she'd": "she would", "we'd": "we would", | |
| "they'd": "they would", "i'd": "I would" | |
| } # Dictionary to replace abbreviations with its expanded form | |
| words = input_text.split() | |
| for i , word in enumerate(words): | |
| if word in abbreviations_dict.keys(): | |
| words[i] = abbreviations_dict[word] | |
| input_text = ' '.join(words) | |
| # remove the punctuation | |
| input_text = ''.join([char for char in input_text if char not in string.punctuation]) | |
| # string.punctuation --> holds all the punctuation | |
| return input_text.split() | |
| def is_in_dict(dict,word): | |
| first , last = 0 , len(dict) - 1 | |
| while first <= last : | |
| mid = (first+last) // 2 | |
| if dict[mid] == word : | |
| return True # word found | |
| else : | |
| if word > dict[mid] : | |
| first = mid + 1 | |
| else : | |
| last = mid - 1 | |
| return False # word is not found in the dictionary --> misspelled | |
| def get_misspelled_words(input_text, dict): | |
| misspelled_words = [word for word in input_text if not is_in_dict(dict, word)] | |
| return misspelled_words | |
| def higlight_misspelled(input_text,misspelled_words): # yasterday, i'm played fotball | |
| for i , word in enumerate(input_text): | |
| if word in misspelled_words: | |
| input_text[i] = f"\033[4;31m{word}\033[0m" # Red text using ANSI escape codes | |
| highlighted_text = ' '.join(input_text) | |
| return highlighted_text # returns the text after being highlighted | |
| def selection_sort(suggested_words_list): | |
| n = len(suggested_words_list) | |
| for i in range(n) : | |
| max_value = suggested_words_list[i][1] #[i] --> the first element in the list ,[1] the second element in the tuple (Ratio) | |
| max_position = i | |
| for j in range (i+1 , n) : | |
| if suggested_words_list[j][1] > max_value : | |
| max_value = suggested_words_list[j][1] | |
| max_position = j | |
| suggested_words_list[i],suggested_words_list[max_position] = suggested_words_list[max_position],suggested_words_list[i] # swaping | |
| def get_suggestions(misspelled_words, dict): | |
| suggestions = {} | |
| similarity = SequenceMatcher() | |
| for word in misspelled_words: | |
| matches = [] | |
| similarity.set_seq2(word) | |
| for item in dict : | |
| similarity.set_seq1(item) | |
| if similarity.ratio() >= 0.70: | |
| matches.append((item, similarity.ratio())) | |
| selection_sort(matches) | |
| suggestions[word] = [item for item, _ in matches[:2]] | |
| return suggestions | |
| # the function that prints the misspelled words and their suggestions | |
| def print_suggestions (suggestions , misspelled_words): | |
| if not misspelled_words : # if the list is empty | |
| print('No misspelled words detected') | |
| else: | |
| print ("\n\033[1;32mMisspelling and Suggestions\033[0m") # prints the header in green color | |
| for word in misspelled_words : | |
| if not suggestions[word] :# the word has no suggestions | |
| print(f"\033[38;5;231m{word}:\033[0m \033[38;5;45mNo suggestions available.\033[0m") | |
| else : | |
| suggested = ", ".join(suggestions[word]) # Joins the 2 suggestions into a string separated by a comma | |
| print(f"\033[38;5;231m{word}:\033[0m \033[38;5;45m{suggested}.\033[0m") | |
| def print_suggested_input(suggestions, original_input_text, misspelled_words): | |
| preprocessed_words = preprocess_input(original_input_text) # Preprocessed words for comparison | |
| for i, word in enumerate(preprocessed_words): | |
| if word in misspelled_words: | |
| preprocessed_words[i] = suggestions[word][0] # Replace in the original text with the first value | |
| suggested_input = ' '.join(preprocessed_words) | |
| return suggested_input | |
| # def spell_check(original_input_text): | |
| # dict = load_dictionary() | |
| # preprocessed_text = preprocess_input(original_input_text) | |
| # misspelled_words = get_misspelled_words(preprocessed_text, dict) | |
| # suggestions = get_suggestions(misspelled_words, dict) | |
| # Suggested_input = print_suggested_input(suggestions, original_input_text, misspelled_words) | |
| # print('Your input is: ', higlight_misspelled(preprocessed_text, misspelled_words)) | |
| # print_suggestions(suggestions, misspelled_words) | |
| # if misspelled_words: # because if the input is right, then dont print Did You Mean | |
| # print('\nSuggested Text:', f'"{Suggested_input}"') | |
| def spell_check(original_input_text): | |
| dict = load_dictionary() | |
| preprocessed_text = preprocess_input(original_input_text) | |
| misspelled_words = get_misspelled_words(preprocessed_text, dict) | |
| suggestions = get_suggestions(misspelled_words, dict) | |
| Suggested_input = print_suggested_input(suggestions, original_input_text, misspelled_words) | |
| if misspelled_words: # because if the input is right, then dont print Did You Mean | |
| return 'Corrected Text: ' + f' "{Suggested_input}"' | |
| else : | |
| return 'No misspelled words detected' | |
| interface = gr.Interface(fn= spell_check, | |
| inputs= "text", | |
| outputs='text', | |
| title= "Spell Checker ", | |
| description="Enter a text to check its spelling") | |
| interface.launch() |