Update app.py

app.py

@@ -13,6 +13,10 @@ class UnifiedTranslator:
         self.model = pipeline("translation", model=model_name)
         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
+        return word[0].isupper()
+
     def translate(self, text):
         # Normalize text to lowercase and strip extra spaces
         text_lower = text.lower().strip()
@@ -32,9 +36,14 @@ class UnifiedTranslator:
                 match = pattern_regex.fullmatch(text_lower)
                 if match:
                     print(f"Match found: {match.group(0)}")
-                    # Replace the placeholder with the actual value if needed
+                    # Replace the placeholder with the actual value if it's a name
                     if '{name}' in pattern:
-                        return translation.format(name=match.group(1))
+                        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)
                     else:
                         return translation
             except re.error as e: