add chunk size function

app.py

@@ -182,7 +182,7 @@ def summary_translate(src_text, temp_tgt_text, tgt_language):
         translation = generate_translation(prompt_style, prompt)
         translations.append(translation)
 
-    best, score = evaluate_candidates(src_text, translations, target_language, session_id)
+    best, score = evaluate_candidates(src_text, translations, tgt_language, session_id)
     if cand_list:
         return best, score
     return "", 0
@@ -318,13 +318,13 @@ def process_text(text, src_language, target_language, max_iterations_value, thre
     bfn_combined_translation = ' '.join(bfn_translated_chunks)
     mpc_combined_translation = ' '.join(mpc_translated_chunks)
 
-    orig, best_score = summary_translate(org_combined_translation, target_language)
+    orig, best_score = summary_translate(text, org_combined_translation, target_language)
     orig_output = f"{orig}\n\nScore: {best_score:.2f}"
-    plan2align_trans, best_score = summary_translate(p2a_combined_translation, target_language)
+    plan2align_trans, best_score = summary_translate(text, p2a_combined_translation, target_language)
     plan2align_output = f"{plan2align_trans}\n\nScore: {best_score:.2f}"
-    best_candidate, best_score = summary_translate(bfn_combined_translation, target_language)
+    best_candidate, best_score = summary_translate(text, bfn_combined_translation, target_language)
     best_of_n_output = f"{best_candidate}\n\nScore: {best_score:.2f}"
-    mpc_candidate, best_score = summary_translate(mpc_combined_translation, target_language)
+    mpc_candidate, best_score = summary_translate(text, mpc_combined_translation, target_language)
     mpc_output = f"{mpc_candidate}\n\nScore: {mpc_score:.2f}"
 
     # if "Original" in translation_methods:
@@ -371,7 +371,7 @@ with gr.Blocks(title="Test-Time Machine Translation with Plan2Align") as demo:
             task_language_input = gr.Dropdown(
                 choices=target_languages,
                 value="English",
-                label="Task (Target) Language"
+                label="Target Language"
             )
             max_iterations_input = gr.Number(label="Max Iterations", value=6)
             threshold_input = gr.Number(label="Threshold", value=0.7)
@@ -382,7 +382,7 @@ with gr.Blocks(title="Test-Time Machine Translation with Plan2Align") as demo:
                 label="Translation Methods"
             )
             chunk_size_input = gr.Number(  # ✅ add chunk function
-                label="Chunk Size (Number of sentences per translation, -1 for all)",
+                label="Chunk Size (-1 for all)",
                 value=-1
             )
             translate_button = gr.Button("Translate")