Update app.py

app.py

@@ -14,45 +14,45 @@ class UnifiedTranslator:
         self.phrase_mapping = phrase_mapping
 
     def is_name(self, word):
-        # A simple heuristic: a word is considered a name if it starts with an uppercase letter
+        # A heuristic: a word is considered a name if it starts with an uppercase letter
+        # and is not one of the common lowercase words
         return word[0].isupper()
 
     def translate(self, text):
-        # Normalize text to lowercase and strip extra spaces
-        text_lower = text.lower().strip()
-        
+        # Normalize text and strip extra spaces
+        text_clean = text.strip()
+
         # Debugging output
-        print(f"Input text: {text_lower}")
+        print(f"Input text: {text_clean}")
 
         # Check if the text matches any pattern in the phrase_mapping (English to Swahili)
         for pattern, translation in self.phrase_mapping.items():
             try:
                 # Use regex to match the pattern with placeholders
                 pattern_regex = re.compile(
-                    re.escape(pattern).replace(r"\{name\}", r"([\w'-]+)").strip(), 
+                    re.escape(pattern).replace(r"\{name\}", r"([\w'-]+)").strip(),
                     re.IGNORECASE
                 )
                 print(f"Checking pattern: {pattern}")
-                match = pattern_regex.fullmatch(text_lower)
+                match = pattern_regex.fullmatch(text_clean)
                 if match:
                     print(f"Match found: {match.group(0)}")
-                    # Replace the placeholder with the actual value if it's a name
-                    if '{name}' in pattern:
-                        name = match.group(1)
-                        if self.is_name(name):
-                            return translation.format(name=name)
-                        else:
-                            # If the matched word isn't a name, use it directly in the translation
-                            return translation.replace('{name}', name)
+                    matched_word = match.group(1)
+                    
+                    # Check if the matched word is a name
+                    if '{name}' in pattern and self.is_name(matched_word):
+                        # If it's a name, replace the placeholder with the actual value
+                        return translation.format(name=matched_word)
                     else:
-                        return translation
+                        # If it's not a name, replace the placeholder with the matched word directly
+                        return translation.replace('{name}', matched_word)
             except re.error as e:
                 print(f"Regex error with pattern {pattern}: {e}")
         
         # Fallback to model translation if no pattern matches
         try:
-            print(f"Fallback to model translation for text: {text}")
-            translation = self.model(text)[0]
+            print(f"Fallback to model translation for text: {text_clean}")
+            translation = self.model(text_clean)[0]
             return translation['translation_text']
         except Exception as e:
             print(f"Model translation error: {e}")