Update app.py

app.py

@@ -1,84 +1,89 @@
 import gradio as gr
-from models import load_translation_model, translate_text
-from utils import compute_bleu
-
+from transformers import MarianMTModel, MarianTokenizer
+from nltk.translate.bleu_score import sentence_bleu, SmoothingFunction
 
+# =============================
+# Словарь доступных моделей
+# =============================
 model_options = {
-"Русский → Английский": "Helsinki-NLP/opus-mt-ru-en",
-"Английский → Русский": "Helsinki-NLP/opus-mt-en-ru",
-"Казахский → Русский": "Helsinki-NLP/opus-mt-kk-ru",
-"Русский → Казахский": "Helsinki-NLP/opus-mt-ru-kk",
+    "Русский → Английский": "Helsinki-NLP/opus-mt-ru-en",
+    "Английский → Русский": "Helsinki-NLP/opus-mt-en-ru",
+    "Казахский → Русский": "Helsinki-NLP/opus-mt-kk-ru",
+    "Русский → Казахский": "Helsinki-NLP/opus-mt-ru-kk",
 }
 
-
-
-
+# =============================
+# Кэш моделей для ускорения
+# =============================
+model_cache = {}
+
+def load_translation_model(model_name):
+    if model_name in model_cache:
+        return model_cache[model_name]
+    tokenizer = MarianTokenizer.from_pretrained(model_name)
+    model = MarianMTModel.from_pretrained(model_name)
+    model_cache[model_name] = (model, tokenizer)
+    return model, tokenizer
+
+# =============================
+# Функция перевода
+# =============================
 def translate_interface(direction, text):
-if not text.strip():
-return "Введите текст для перевода!"
-
-
-model_name = model_options[direction]
-model, tokenizer = load_translation_model(model_name)
-result = translate_text(text, model, tokenizer)
-return result
-
-
-
-
+    if not text or len(text.strip()) == 0:
+        return "Введите текст для перевода!"
+    model_name = model_options[direction]
+    model, tokenizer = load_translation_model(model_name)
+    batch = tokenizer([text], return_tensors="pt", padding=True)
+    gen = model.generate(**batch)
+    result = tokenizer.batch_decode(gen, skip_special_tokens=True)[0]
+    return result
+
+# =============================
+# Функция BLEU
+# =============================
 def bleu_interface(direction, original, reference):
-if not original.strip() or not reference.strip():
-return "Введите текст и эталон!"
-
-
-model_name = model_options[direction]
-model, tokenizer = load_translation_model(model_name)
-
-
-translation = translate_text(original, model, tokenizer)
-bleu = compute_bleu(reference, translation)
-
-
-return f"BLEU: {bleu:.4f}"
-
-
-
-
+    if not original.strip() or not reference.strip():
+        return "Введите текст и эталон!"
+    translation = translate_interface(direction, original)
+    smoothie = SmoothingFunction().method4
+    ref_tokens = [reference.split()]
+    hyp_tokens = translation.split()
+    bleu = sentence_bleu(ref_tokens, hyp_tokens, smoothing_function=smoothie)
+    return f"BLEU: {bleu:.4f}"
+
+# =============================
+# Gradio UI
+# =============================
 with gr.Blocks(theme=gr.themes.Soft(primary_hue="purple", neutral_hue="slate", secondary_hue="pink")) as demo:
-gr.Markdown("""
-# 🌙 Мощный Машинный Перевод
-Тёмная тема, стильный UI, BLEU, множество моделей.
-""")
-
-
-with gr.Row():
-direction = gr.Dropdown(list(model_options.keys()), label="Направление перевода")
-
-
-with gr.Row():
-input_text = gr.Textbox(lines=4, label="Введите текст")
-output_text = gr.Textbox(lines=4, label="Перевод")
-
-
-translate_btn = gr.Button("🚀 Перевести")
-translate_btn.click(fn=translate_interface, inputs=[direction, input_text], outputs=output_text)
-
-
-gr.Markdown("---")
-gr.Markdown("### 📊 BLEU Оценка качества")
+    gr.Markdown("""
+    # 🌙 Мощный Машинный Перевод
+    Тёмная тема, стильный UI, BLEU, множество моделей.
+    """)
 
+    with gr.Row():
+        direction = gr.Dropdown(list(model_options.keys()), label="Направление перевода")
 
-with gr.Row():
-original = gr.Textbox(lines=3, label="Исходный текст")
-reference = gr.Textbox(lines=3, label="Эталонный перевод")
-bleu_out = gr.Textbox(label="BLEU Score")
+    with gr.Row():
+        input_text = gr.Textbox(lines=4, label="Введите текст")
+        output_text = gr.Textbox(lines=4, label="Перевод")
 
+    translate_btn = gr.Button("🚀 Перевести")
+    translate_btn.click(fn=translate_interface, inputs=[direction, input_text], outputs=output_text)
 
-bleu_btn = gr.Button("📈 Посчитать BLEU")
-bleu_btn.click(fn=bleu_interface, inputs=[direction, original, reference], outputs=bleu_out)
+    # Кнопка копирования перевода
+    copy_btn = gr.Button("📋 Коп��ровать перевод")
+    copy_btn.click(lambda text: text, inputs=output_text, outputs=output_text)
 
+    gr.Markdown("---")
+    gr.Markdown("### 📊 BLEU Оценка качества")
 
+    with gr.Row():
+        original = gr.Textbox(lines=3, label="Исходный текст")
+        reference = gr.Textbox(lines=3, label="Эталонный перевод")
+        bleu_out = gr.Textbox(label="BLEU Score")
 
+    bleu_btn = gr.Button("📈 Посчитать BLEU")
+    bleu_btn.click(fn=bleu_interface, inputs=[direction, original, reference], outputs=bleu_out)
 
 if __name__ == "__main__":
-demo.launch()
+    demo.launch()