Update app.py

app.py

@@ -28,15 +28,24 @@ languages_list = [("Gujarati", "gu_IN"), ('Hindi',"hi_IN") , ("Bengali","bn_IN")
                   ("Marathi","mr_IN"), ("Tamil","ta_IN"), ("Telugu","te_IN")]
 
 
+def intitalize_lang(language):
+    return language
 
+def english_to_indian(sentence):
+    lang = intitalize_lang()
+    translation_tokenizer.src_lang = "en_xx"
+    encoded_hi = translation_tokenizer(sentence, return_tensors="pt")
+    generated_tokens = translation_model.generate(**encoded_hi, forced_bos_token_id=translation_tokenizer.lang_code_to_id[lang] )
+    return (translation_tokenizer.batch_decode(generated_tokens, skip_special_tokens=True))
+
+
+def indian_to_english(sentence):
+    lang = intitalize_lang()
+    translation_tokenizer.src_lang = lang
+    encoded_hi = translation_tokenizer(sentence, return_tensors="pt")
+    generated_tokens = translation_model.generate(**encoded_hi, forced_bos_token_id=translation_tokenizer.lang_code_to_id["en_XX"] )
+    return (translation_tokenizer.batch_decode(generated_tokens, skip_special_tokens=True))
 
-def intitalize_lang(language):
-    # translation_tokenizer.src_lang = "en_xx"
-    # encoded_hi = translation_tokenizer(sentence, return_tensors="pt")
-    # generated_tokens = translation_model.generate(**encoded_hi, forced_bos_token_id=translation_tokenizer.lang_code_to_id[language] )
-    # x = translation_tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)
-    print(language)
-    # return x
 
 
 llm_model = "mistralai/Mistral-7B-Instruct-v0.2"
@@ -182,7 +191,7 @@ def conversation(qa_chain, message, history):
     # Generate response using QA chain
     response = qa_chain({"question": message, "chat_history": formatted_chat_history})
     response_answer = response["answer"]
-    print(english_to_indian(response["answer"][:100]))
+    print(english_to_indian(response["answer"][:500]))
     if response_answer.find("Helpful Answer:") != -1:
         response_answer = response_answer.split("Helpful Answer:")[-1]
     response_sources = response["source_documents"]
@@ -212,18 +221,6 @@ def upload_file(file_obj):
     return list_file_path
 
 
-def english_to_indian(sentence):
-    translation_tokenizer.src_lang = "en_xx"
-    encoded_hi = translation_tokenizer(sentence, return_tensors="pt")
-    generated_tokens = translation_model.generate(**encoded_hi, forced_bos_token_id=translation_tokenizer.lang_code_to_id[lang_btn] )
-    return (translation_tokenizer.batch_decode(generated_tokens, skip_special_tokens=True))
-
-
-def indian_to_english(sentence):
-    translation_tokenizer.src_lang = lang_btn
-    encoded_hi = translation_tokenizer(sentence, return_tensors="pt")
-    generated_tokens = translation_model.generate(**encoded_hi, forced_bos_token_id=translation_tokenizer.lang_code_to_id["en_XX"] )
-    return (translation_tokenizer.batch_decode(generated_tokens, skip_special_tokens=True))
 
 
 def demo():
@@ -278,7 +275,7 @@ def demo():
         with gr.Row():
             lang_btn = gr.Dropdown(languages_list, label="Languages", value = languages_list[1],
                                 type="value", info="Choose your language",interactive = True)
-            gr.Dropdown.select(english_to_indian, inputs=lang_btn)
+            lang_btn.select(intitalize_lang, inputs = lang_btn)
         
         chatbot = gr.Chatbot(height=300)