Update app.py

app.py

@@ -14,10 +14,8 @@ def preprocess_capitalization(text: str) -> str:
     processed_words = []
     
     for word in words:
-        # Check if the word is an acronym (all uppercase letters)
         if re.match(r"^[A-Z]+$", word):
             processed_words.append(word)  # Leave acronyms unchanged
-        # Check if the word has mixed capitalization (e.g., "HEllo")
         elif re.search(r"[A-Z]", word) and re.search(r"[a-z]", word):
             processed_words.append(word[0].upper() + word[1:].lower())  # Correct capitalization
         else:
@@ -25,7 +23,7 @@ def preprocess_capitalization(text: str) -> str:
     
     return " ".join(processed_words)
 
-def preprocess_text(text: str):
+def preprocess_text(text: str, is_spell_corrected: bool = False):
     """Process text and return corrections with position information."""
     result = {
         "spell_suggestions": [],
@@ -44,33 +42,29 @@ def preprocess_text(text: str):
         })
         text = capitalized_text  # Update text for further processing
 
-    # Transformer spell check - only for words that look misspelled
-    spell_checked = spell_checker(text, max_length=512)[0]['generated_text']
-    if spell_checked != text:
-        spell_words = spell_checked.split()
-        text_words = text.split()
-        
-        # Only suggest spell corrections for words that are actually different
-        for orig, corrected in zip(text_words, spell_words):
-            if orig.lower() != corrected.lower():  # Compare case-insensitive
-                result["spell_suggestions"].append({
-                    "original": orig,
-                    "corrected": corrected,
-                    "type": "spell"
-                })
+    # Skip spell checking if this is a spell-corrected input
+    if not is_spell_corrected:
+        spell_checked = spell_checker(text, max_length=512)[0]['generated_text']
+        if spell_checked != text:
+            spell_words = spell_checked.split()
+            text_words = text.split()
+            
+            for orig, corrected in zip(text_words, spell_words):
+                if orig.lower() != corrected.lower():
+                    result["spell_suggestions"].append({
+                        "original": orig,
+                        "corrected": corrected,
+                        "type": "spell"
+                    })
 
     # Add fluency/style suggestions (other suggestions)
-    # Only add if the word isn't already in spell suggestions
     spell_originals = {s["original"] for s in result["spell_suggestions"]}
     words = text.split()
     
     for word in words:
-        # Skip words that have spell suggestions or already have punctuation
         if word in spell_originals or word.endswith(('.', '?', '!')):
             continue
-            
-        # Add style suggestions (like adding punctuation)
-        if not word.endswith('!'):  # Only suggest adding exclamation if it doesn't already have one
+        if not word.endswith('!'):
             result["other_suggestions"].append({
                 "original": word,
                 "corrected": word + "!",
@@ -89,7 +83,6 @@ def translate_text(text: str):
     client = Client("Frenchizer/space_21")
     try:
         translation = client.predict(text)
-        # Return an empty suggestions object since we're just translating
         result = {
             "spell_suggestions": [],
             "other_suggestions": [],
@@ -100,11 +93,10 @@ def translate_text(text: str):
     except Exception as e:
         return f"Error: {str(e)}", {}
 
-def preprocess_and_forward(text: str):
+def preprocess_and_forward(text: str, is_spell_corrected: bool = False):
     """Process text and forward to translation service."""
-    original_text, preprocessing_result = preprocess_text(text)
+    original_text, preprocessing_result = preprocess_text(text, is_spell_corrected)
     
-    # Forward original text to translation service
     client = Client("Frenchizer/space_21")
     try:
         translation = client.predict(original_text)
@@ -121,9 +113,10 @@ def translate_only(text: str):
 with gr.Blocks() as demo:
     with gr.Tab("Main"):
         input_text = gr.Textbox(label="Input Text")
+        is_spell_corrected = gr.Checkbox(label="Spell Corrected", value=False, visible=False)  # Hidden flag
         output_text = gr.Textbox(label="Output Text")
         preprocess_button = gr.Button("Process and Translate")
-        preprocess_button.click(fn=preprocess_and_forward, inputs=input_text, outputs=output_text)
+        preprocess_button.click(fn=preprocess_and_forward, inputs=[input_text, is_spell_corrected], outputs=output_text)
     with gr.Tab("Translation Only"):
         translate_input = gr.Textbox(label="Input Text")
         translate_output = gr.Textbox(label="Output Text")