add upgrade model version

.env

@@ -1,2 +1,3 @@
 GEMINI_API_KEY = AIzaSyDIPbH7zKoeTS5aKMQuMjzBBiVlWadcmr8
-MONGODB_URI = mongodb+srv://admin:[email protected]/?retryWrites=true&w=majority&appName=Cluster0
+MONGODB_URI = mongodb+srv://admin:[email protected]/?retryWrites=true&w=majority&appName=Cluster0
+MODEL_VERSION = gemini-2.0-flash-lite

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.5-pro")
+        model = genai.GenerativeModel(os.getenv("MODEL_VERSION"))
 
         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

@@ -7,8 +7,10 @@ import json
 import time
 from google.api_core.exceptions import ResourceExhausted
 import re
+from dotenv import load_dotenv
+load_dotenv()  # Load environment variables from .env file
 
-genai.configure(api_key="AIzaSyDInJcxzqBvsh1avs4Zkxb4ZGBooNzOyEM")
+genai.configure(os.getenv("GEMINI_API_KEY"))  # Configure the Gemini API key
 
 
 def unzip_office_file(pptx_file: io.BytesIO):
@@ -34,7 +36,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.5-pro')  # hoặc 'gemini-1.5-flash'
+            model = genai.GenerativeModel(os.getenv("MODEL_VERSION"))  # hoặc 'gemini-1.5-flash'
 
             system_prompt = f"""You are a translation engine.
 Translate the following text accurately to {target_lang}.
@@ -141,7 +143,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.5-pro')
+            model = genai.GenerativeModel(os.getenv("MODEL_VERSION"))
             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.5-pro')
+    model = genai.GenerativeModel(os.getenv("MODEL_VERSION"))
     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,