Update app.py

app.py

@@ -21,11 +21,18 @@ LANG_TARGET = 'vi'
 OUTPUT_SUFFIX = "_Dich_CN"
 BATCH_SIZE = 100
 SLEEP_BETWEEN_BATCHES = 0.1
-GEMINI_MODEL = "gemini-1.5-flash-latest"
 MAX_RETRIES = 5
+
+# List of available Gemini models for the user to choose from
+GEMINI_MODELS = [
+    "gemini-flash-latest",  # Fast, cost-effective, good for most tasks
+    "gemini-2.5-pro",    # Highest quality, larger context window
+    "gemini-flash-lite-latest",           # Previous generation pro model
+] #
+
 # -----------------
 
-# --- HELPER FUNCTIONS (from your original script) ---
+# --- HELPER FUNCTIONS (Unchanged) ---
 
 def safe_sheet_title(base: str) -> str:
     cleaned = re.sub(r'[:\\/?*\[\]]', '_', base)
@@ -40,7 +47,6 @@ def excel_quote_sheet(name: str) -> str:
 def _gemini_translate_call(model, texts: List[str], *, src: str, tgt: str, max_retries: int = MAX_RETRIES) -> List[str]:
     if not texts:
         return []
-
     system_prompt = (
         "You are a professional translator. "
         f"Translate each input string from {src} to {tgt}. "
@@ -55,7 +61,6 @@ def _gemini_translate_call(model, texts: List[str], *, src: str, tgt: str, max_r
         + json.dumps(payload, ensure_ascii=False)
         + "\n\nOUTPUT: JSON array of strings only."
     )
-
     for attempt in range(1, max_retries + 1):
         try:
             resp = model.generate_content(user_msg)
@@ -85,27 +90,32 @@ def update_formula_references(sheet, sheet_map):
                     formula = pattern.sub(f"{new_q}!", formula)
                 cell.value = formula
 
-# --- CORRECTED MAIN PROCESSING FUNCTION FOR GRADIO ---
-def process_translation(api_key: str, input_file, progress=gr.Progress(track_tqdm=True)):
+# --- MAIN PROCESSING FUNCTION FOR GRADIO ---
+
+# <-- MODIFIED: Added 'model_name' as an argument
+def process_translation(api_key: str, model_name: str, input_file, progress=gr.Progress(track_tqdm=True)):
     """
     Main function to be called by the Gradio interface.
-    It takes the API key and uploaded file, performs translation, and returns the output file.
+    It takes the API key, selected model, and uploaded file, performs translation, and returns the output file.
     """
     if not api_key or not api_key.strip():
         raise gr.Error("Gemini API Key is missing. Please paste your key.")
     if input_file is None:
         raise gr.Error("No Excel file uploaded. Please upload a .xlsx file.")
+    if not model_name:
+        raise gr.Error("No Gemini model selected. Please choose a model from the dropdown.")
 
     logs = []
     
     try:
-        # 1. Configure Gemini with the user-provided key
+        # 1. Configure Gemini with the user-provided key and selected model
         genai.configure(api_key=api_key.strip())
-        model = genai.GenerativeModel(GEMINI_MODEL)
-        logs.append(f"✅ Initialized Gemini model: {GEMINI_MODEL}")
+        # <-- MODIFIED: Use the 'model_name' argument instead of a fixed constant
+        model = genai.GenerativeModel(model_name)
+        logs.append(f"✅ Initialized Gemini model: {model_name}")
         yield "\n".join(logs), None # Update logs, no file yet
 
-        # 2. Load the workbook from the temporary path of the uploaded file
+        # The rest of the function remains the same as the corrected version from before
         wb = load_workbook(input_file.name)
         original_sheet_names = [name for name in wb.sheetnames if OUTPUT_SUFFIX not in name]
         logs.append(f"✅ Loaded Excel file. Found {len(original_sheet_names)} sheet(s) to translate.")
@@ -116,10 +126,8 @@ def process_translation(api_key: str, input_file, progress=gr.Progress(track_tqd
         # 3. Step 1: Translate and create new sheets
         for sheet_name in progress.tqdm(original_sheet_names, desc="Translating Sheets"):
             source_sheet = wb[sheet_name]
-            # FIX: Was using the undefined 'source_sheet_name' here
             new_sheet_name_raw = f"{sheet_name}{OUTPUT_SUFFIX}"
             new_sheet_name = safe_sheet_title(new_sheet_name_raw)
-            # FIX: Was using 'source_sheet_name' as the key
             sheet_name_map[sheet_name] = new_sheet_name
 
             if new_sheet_name in wb.sheetnames:
@@ -141,7 +149,6 @@ def process_translation(api_key: str, input_file, progress=gr.Progress(track_tqd
                         cells_to_translate.append({'text': val, 'row': cell.row, 'col': cell.column})
             
             total_cells = len(cells_to_translate)
-            # FIX: Was using 'source_sheet_name' in the log message
             logs.append(f"--- Translating Sheet: {sheet_name} ({total_cells} cells) -> {new_sheet_name} ---")
             yield "\n".join(logs), None
             
@@ -176,14 +183,12 @@ def process_translation(api_key: str, input_file, progress=gr.Progress(track_tqd
                         target_cell.value = translated
                 
                 processed_count = min(idx + len(chunk), total_cells)
-                # FIX: Was using 'source_sheet_name' in the log update
                 logs[-1] = f"--- Translating Sheet: {sheet_name} ({processed_count}/{total_cells} cells) -> {new_sheet_name} ---"
                 yield "\n".join(logs), None
                 
                 idx += BATCH_SIZE
                 time.sleep(SLEEP_BETWEEN_BATCHES)
             
-            # FIX: Was using 'source_sheet_name' here
             logs.append(f"✅ Finished sheet: {sheet_name}")
             yield "\n".join(logs), None
 
@@ -207,7 +212,6 @@ def process_translation(api_key: str, input_file, progress=gr.Progress(track_tqd
         yield "\n".join(logs), output_file_path
 
     except Exception as e:
-        # Use gr.Error to show a popup to the user
         raise gr.Error(f"An error occurred: {e}")
 
 # --- GRADIO INTERFACE ---
@@ -216,16 +220,17 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown(
         """
         # 批量Excel翻译工具 (ZH-VI) | Excel Translator (ZH-VI)
-        使用Google Gemini Pro进行中越翻译。
+        使用Google Gemini进行中越翻译。
         
         **使用说明 (Instructions):**
         1.  在下方输入您的 Google Gemini API Key。(API Key is kept private and not stored).
-        2.  上传您的 `.xlsx` 格式的Excel文件。
-        3.  点击 "开始翻译 (Translate)" 按钮。
-        4.  等待处理完成，处理日志会实时显示。
-        5.  完成后，在右侧下载翻译好的文件。
+        2.  选择您想使用的 Gemini 模型。(`gemini-1.5-flash` is recommended for speed and cost).
+        3.  上传您的 `.xlsx` 格式的Excel文件。
+        4.  点击 "开始翻译 (Translate)" 按钮。
+        5.  等待处理完成，处理日志会实时显示。
+        6.  完成后，在右侧下载翻译好的文件。
         """
-    )
+    ) # <-- MODIFIED: Added instruction for model selector
     with gr.Row():
         with gr.Column(scale=1):
             api_key_input = gr.Textbox(
@@ -233,6 +238,12 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
                 type="password",
                 placeholder="Enter your API key here..."
             )
+            # <-- ADDED: Dropdown for model selection
+            model_selector = gr.Dropdown(
+                label="选择Gemini模型 (Select Gemini Model)",
+                choices=GEMINI_MODELS,
+                value=GEMINI_MODELS[0] # Set default to flash
+            )
             file_input = gr.File(
                 label="Upload Excel File (.xlsx)",
                 file_types=[".xlsx"]
@@ -248,9 +259,10 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             )
             file_output = gr.File(label="下载翻译文件 (Download Translated File)")
     
+    # <-- MODIFIED: Add 'model_selector' to the inputs list
     translate_button.click(
         fn=process_translation,
-        inputs=[api_key_input, file_input],
+        inputs=[api_key_input, model_selector, file_input],
         outputs=[log_output, file_output]
     )