Update app.py

app.py

@@ -9,7 +9,7 @@ import json
 
 class LTRC_Translation_API():
     def __init__(self, url = 'https://ssmt.iiit.ac.in/onemt', src_lang = 'en', tgt_lang = 'te'):
-        self.lang_map = {'te': 'tel', 'en': 'eng', 'ta': 'tam', 'ml': 'mal', 'mr': 'mar', 'kn': 'kan', 'hi': 'hin', 'en': 'eng'}
+        self.lang_map = {'te': 'tel', 'en': 'eng', 'ta': 'tam', 'ml': 'mal', 'mr': 'mar', 'kn': 'kan', 'hi': 'hin'}
         self.url = url
 
         self.headers = {
@@ -17,10 +17,11 @@ class LTRC_Translation_API():
             'Accept': 'application/json'
         }
 
-        lang = self.lang_map.get(tgt_lang, 'te')
+        tgt_lang = self.lang_map.get(tgt_lang, 'te')
+        src_lang = self.lang_map.get(src_lang, 'en')
 
         self.src_lang = src_lang
-        self.tgt_lang = lang
+        self.tgt_lang = tgt_lang
 
     def translate(self, text):
         try:
@@ -127,7 +128,7 @@ class LTRC_Translation_API():
 # 		return " ".join(decoded_output)
 
 
-def get_prediction(text, lang_id, translate = False):
+def get_prediction(text, src_lang_id, tgt_lang_id, translate = False):
     # if len(sys.argv)<3:
     # 	print("Usage: python app.py <text_file_path> <lang_id>")
     # 	print("Text file should contain the article news")
@@ -165,7 +166,7 @@ def get_prediction(text, lang_id, translate = False):
     # return "Headline: " + headline + "\nSummary: " + summary
     # return [text, summary, headline]
 
-    translator = LTRC_Translation_API(tgt_lang = lang_id)
+    translator = LTRC_Translation_API(src_lang = src_lang_id, tgt_lang = tgt_lang_id)
     text = translator.translate(text)
     
     return text