add upgrade model version

pages/upload.py

@@ -79,7 +79,7 @@ with st.container():
 
     with col2:
         st.markdown('<p style="font-size:16px; font-weight:bold; margin-bottom:4px;">🌐 Ngôn ngữ muốn dịch sang</p>', unsafe_allow_html=True)
-        target_lang = st.selectbox("  ", ["chinese", "english", "vietnamese", "japanese"], key="target_lang")
+        target_lang = st.selectbox("  ", ["vietnamese", "chinese", "english", "japanese"], key="target_lang")
 
 # Xử lý file trực tiếp
 def process_file(file, file_type):

translate/translator.py

@@ -21,7 +21,7 @@ def translate_text_dict(text_dict: Dict[str, List[str]], source_lang:  str = "vi
         Return the translated texts formatted like the original dictionary. Do NOT say anthing else. Return it as a JSON block."""
 
         genai.configure(api_key=os.getenv("GEMINI_API_KEY"))
-        model = genai.GenerativeModel("gemini-2.0-flash")
+        model = genai.GenerativeModel("gemini-2.5-pro")
 
         response = model.generate_content(prompt) # Use a model appropriate for your needs and API key.  gemini-2.0-flash doesn't exist.  1.5-pro is a good general-purpose model.
 

utils/utils.py

@@ -34,7 +34,7 @@ def translate_single_text(text: str, source_lang: str = 'English', target_lang:
     retries = 0
     while retries <= max_retries:
         try:
-            model = genai.GenerativeModel('gemini-2.0-flash')  # hoặc 'gemini-1.5-flash'
+            model = genai.GenerativeModel('gemini-2.5-pro')  # hoặc 'gemini-1.5-flash'
 
             system_prompt = f"""You are a translation engine.
 Translate the following text accurately to {target_lang}.
@@ -141,7 +141,7 @@ def translate_text(text_dict, source_lang='English', target_lang="Vietnamese", m
     retry_count = 0
     while retry_count < max_retries:
         try:
-            model = genai.GenerativeModel('gemini-2.0-flash')
+            model = genai.GenerativeModel('gemini-2.5-pro')
             full_prompt = f"{system_prompt.strip()}\n\n{user_prompt.strip()}"
 
             response = model.generate_content(

word/word_helper.py

@@ -36,7 +36,7 @@ def batch_translate(texts, source_lang = 'English', target_lang="Vietnamese"):
     json_data = json.dumps({i: t for i, t in enumerate(texts)})
     user_prompt = f"Target language: {target_lang}. JSON file: {json_data}" 
   
-    model = genai.GenerativeModel('gemini-2.0-flash')
+    model = genai.GenerativeModel('gemini-2.5-pro')
     response = model.generate_content(contents = system_prompt.strip() + "\n" + user_prompt.strip(), generation_config={
             'temperature': 0.3,  # Adjust temperature for desired creativity
             'top_p': 1,
@@ -89,7 +89,7 @@ def batch_translate_loop(batch, source_lang, target_lang):
     print(len(batch), len(translated_batch))
     return translated_batch
 
-def get_batches(texts, limit = 1000):
+def get_batches(texts, limit = 2000):
     batches = []
     batch = []
     word_count = 0
@@ -108,7 +108,7 @@ def get_batches(texts, limit = 1000):
 
 def full_translate(texts, source_lang = 'English', target_lang="Vietnamese"):
     full_translated_texts = []
-    batches = get_batches(texts, limit = 1000)
+    batches = get_batches(texts, limit = 2000)
     word_count = 0
     global time_spent_sleeping