Update app.py

app.py

@@ -27,6 +27,7 @@ common_phrases = {
     "Good luck": "موفق باشید",
     "Have a good trip": "سفر خوبی داشته باشید",
     "What is your name?": "اسم شما چیست؟",
+    "My name is Sara": "اسم من سارا است",
     "Where are you from?": "اهل کجا هستید؟",
     "I am from Iran": "من اهل ایران هستم",
     "Do you speak English?": "آیا انگلیسی صحبت می‌کنید؟",
@@ -43,7 +44,7 @@ common_phrases = {
     "Where is the hospital?": "بیمارستان کجاست؟",
 }
 
-# Translation function
+# Improved transliteration function (Farsi to Cyrillic)
 def transliterate_farsi_to_cyrillic(farsi_text):
     transliteration_map = {
         "ا": "А", "ب": "Б", "پ": "П", "ت": "Т", "ث": "С", "ج": "Ҷ",
@@ -83,6 +84,21 @@ def transliterate_farsi_to_cyrillic(farsi_text):
 
     return cyrillic_text
 
+# Translation function
+def translate_to_cyrillic_farsi(text):
+    # First, check if the phrase is in predefined common phrases
+    if text in common_phrases:
+        farsi_text = common_phrases[text]
+    else:
+        # Otherwise, use the translation model
+        tokenizer.src_lang = "en"
+        encoded_text = tokenizer(text, return_tensors="pt")
+        generated_tokens = model.generate(**encoded_text, forced_bos_token_id=tokenizer.get_lang_id("fa"))
+        farsi_text = tokenizer.decode(generated_tokens[0], skip_special_tokens=True)
+    
+    # Transliterate Farsi to Cyrillic
+    cyrillic_text = transliterate_farsi_to_cyrillic(farsi_text)
+    return farsi_text, cyrillic_text
 
 # Gradio Interface
 interface = gr.Interface(
@@ -97,4 +113,5 @@ interface = gr.Interface(
 )
 
 # Launch the app
-interface.launch()
+if __name__ == "__main__":
+    interface.launch()