Update app.py

app.py

@@ -13,46 +13,37 @@ class UnifiedTranslator:
         self.model = pipeline("translation", model=model_name)
         self.phrase_mapping = phrase_mapping
 
-    def is_name(self, word):
-        # 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 and strip extra spaces
-        text_clean = text.strip()
-
+        # Normalize text to lowercase and strip extra spaces
+        text_lower = text.lower().strip()
+        
         # Debugging output
-        print(f"Input text: {text_clean}")
+        print(f"Input text: {text_lower}")
 
         # 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_clean)
+                match = pattern_regex.fullmatch(text_lower)
                 if match:
                     print(f"Match found: {match.group(0)}")
-                    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)
+                    # Replace the placeholder with the actual value if needed
+                    if '{name}' in pattern:
+                        return translation.format(name=match.group(1))
                     else:
-                        # If it's not a name, replace the placeholder with the matched word directly
-                        return translation.replace('{name}', matched_word)
+                        return translation
             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_clean}")
-            translation = self.model(text_clean)[0]
+            print(f"Fallback to model translation for text: {text}")
+            translation = self.model(text)[0]
             return translation['translation_text']
         except Exception as e:
             print(f"Model translation error: {e}")
@@ -74,4 +65,4 @@ iface = gr.Interface(
 )
 
 # Launch the interface
-iface.launch(share=True)
+iface.launch(share=True)