Update app.py

app.py

@@ -14,8 +14,10 @@ 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:
@@ -23,7 +25,7 @@ def preprocess_capitalization(text: str) -> str:
     
     return " ".join(processed_words)
 
-def preprocess_text(text: str, is_spell_corrected: bool = False):
+def preprocess_text(text: str):
     """Process text and return corrections with position information."""
     result = {
         "spell_suggestions": [],
@@ -42,22 +44,20 @@ def preprocess_text(text: str, is_spell_corrected: bool = False):
         })
         text = capitalized_text  # Update text for further processing
 
-    # Transformer spell check - only for words that look misspelled, skipped if spell corrected
-    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()
-            
-            # 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"
-                    })
-            text = spell_checked  # Update text after spell correction
+    # 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"
+                })
 
     # Add fluency/style suggestions (other suggestions)
     # Only add if the word isn't already in spell suggestions
@@ -69,8 +69,8 @@ def preprocess_text(text: str, is_spell_corrected: bool = False):
         if word in spell_originals or word.endswith(('.', '?', '!')):
             continue
             
-        # Add style suggestion (like adding punctuation)
-        if not word.endswith('!'):
+        # Add style suggestions (like adding punctuation)
+        if not word.endswith('!'):  # Only suggest adding exclamation if it doesn't already have one
             result["other_suggestions"].append({
                 "original": word,
                 "corrected": word + "!",
@@ -89,6 +89,7 @@ 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": [],
@@ -99,10 +100,11 @@ def translate_text(text: str):
     except Exception as e:
         return f"Error: {str(e)}", {}
 
-def preprocess_and_forward(text: str, is_spell_corrected: bool = False):
+def preprocess_and_forward(text: str):
     """Process text and forward to translation service."""
-    original_text, preprocessing_result = preprocess_text(text, is_spell_corrected)
+    original_text, preprocessing_result = preprocess_text(text)
     
+    # Forward original text to translation service
     client = Client("Frenchizer/space_21")
     try:
         translation = client.predict(original_text)
@@ -119,10 +121,9 @@ 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, is_spell_corrected], outputs=output_text)
+        preprocess_button.click(fn=preprocess_and_forward, inputs=input_text, outputs=output_text)
     with gr.Tab("Translation Only"):
         translate_input = gr.Textbox(label="Input Text")
         translate_output = gr.Textbox(label="Output Text")