add choose lang

excel/excel_translate.py

@@ -7,7 +7,6 @@ import io
 import pandas as pd
 import pymongo
 import gridfs
-from io import BytesIO
 import tempfile
 import os
 

pages/upload.py

@@ -25,6 +25,7 @@ st.title("Upload PPTX to MongoDB")
 # st.set_option("server.fileUploader.allowMediaFiles", True)
 uploaded_file = st.file_uploader("Chọn file")
 file_name_input = st.text_input("Tên file để lưu (không cần .pptx)", value="")
+target_lang = st.selectbox("Chọn ngôn ngữ dịch", ["english", "vietnamese"])
 
 final_pptx_id = None  # Biến lưu ID file sau khi xử lý
 
@@ -39,7 +40,7 @@ if uploaded_file is not None:
 
             xml_file_id = ppt_to_xml_mongodb(file_id)
             text_dict = extract_text_from_xml(file_id=xml_file_id)
-            translated_dict = translate_text_dict(text_dict, source_lang="VietNamese", target_lang="English", gemini_api=api_key)
+            translated_dict = translate_text_dict(text_dict, target_lang=target_lang, gemini_api=api_key)
 
             final_xml_id = update_xml_with_translated_text_mongodb(xml_file_id, translated_dict)
             st.write(f"Final XML ID: {final_xml_id}")

translate/translator.py

@@ -6,8 +6,7 @@ import google.generativeai as genai
 def translate_text_dict(text_dict: Dict[str, List[str]], source_lang: str, target_lang: str = "vi", gemini_api: str = "") -> Dict[str, List[str]]:
     def translate_batch(batch_dict: Dict[str, List[str]]) -> Dict[str, List[str]]:
         """Translates a single batch of text."""
-        prompt = f"""The following python dictionary contains pieces of text that form a whole document: {json.dumps(batch_dict)}
-        The text is in {source_lang}, with a chance of there being phrases in other languages as well.
+        prompt = f"""The following python dictionary contains pieces of text that form a whole document: {json.dumps(batch_dict)}.
 
         Read through the entire dictionary, then translate the texts into {target_lang} so that the meaning is as close to the intended context as possible.