Update app.py

app.py

@@ -26,17 +26,44 @@ try:
 except ImportError:
     print("Устанавливаем необходимые пакеты...")
     install_packages()
-    from transformers import MarianMTModel, MarianTokenizer
-    import gradio as gr
+
+
+import subprocess
+import sys
+import warnings
+from transformers import MarianMTModel, MarianTokenizer
+import gradio as gr
+
+# Подавление предупреждений
+warnings.filterwarnings("ignore", message="Recommended: pip install sacremoses.")
+
+# Установка недостающих пакетов
+def install_packages():
+    required = ['sentencepiece', 'sacremoses', 'torch', 'gradio']
+    for package in required:
+        try:
+            __import__(package.split('==')[0])
+        except ImportError:
+            subprocess.check_call([sys.executable, "-m", "pip", "install", package])
+
+install_packages()
 
 # Настройки модели
 MODEL_NAME = "7-Sky/skyopus-pol-rus"
 DEFAULT_TRANSLATIONS = 3
 MAX_TRANSLATIONS = 5
 
-# Загрузка модели
-tokenizer = MarianTokenizer.from_pretrained(MODEL_NAME)
-model = MarianMTModel.from_pretrained(MODEL_NAME)
+# Загрузка модели с обработкой ошибок
+try:
+    tokenizer = MarianTokenizer.from_pretrained(MODEL_NAME)
+    model = MarianMTModel.from_pretrained(MODEL_NAME)
+except Exception as e:
+    print(f"Ошибка загрузки модели: {e}")
+    sys.exit(1)
+
+def clean_translation(text):
+    """Очистка перевода от языковых тегов и лишних пробелов"""
+    return text.replace(">>rus<<", "").strip()
 
 def translate_text(source_text, num_translations):
     if not source_text.strip():
@@ -44,53 +71,59 @@ def translate_text(source_text, num_translations):
     
     try:
         num_translations = min(int(num_translations), MAX_TRANSLATIONS)
-        text_with_token = ">>rus<< " + source_text
+        inputs = tokenizer(">>rus<< " + source_text, return_tensors="pt", padding=True)
         
-        inputs = tokenizer(text_with_token, return_tensors="pt", padding=True)
-        
-        translated_tokens = model.generate(
+        outputs = model.generate(
             **inputs,
             num_return_sequences=num_translations,
             num_beams=num_translations,
             max_length=512
         )
         
-        # Декодируем и удаляем >>rus<< из каждого варианта перевода
-        translations = []
-        for tokens in translated_tokens:
-            decoded = tokenizer.decode(tokens, skip_special_tokens=True)
-            # Удаляем >>rus<< если он есть в начале строки
-            if decoded.startswith(">>rus<<"):
-                decoded = decoded[7:].strip()
-            translations.append(decoded)
-        
-        return translations
+        return [clean_translation(tokenizer.decode(tokens, skip_special_tokens=True)) 
+               for tokens in outputs]
     except Exception as e:
-        return [f"Произошла ошибка: {str(e)}"]
+        return [f"Ошибка перевода: {str(e)}"]
 
-# Создание интерфейса
-with gr.Blocks(title="Польско-русский переводчик") as app:
-    gr.Markdown("## Польско-русский переводчик с несколькими вариантами перевода")
-    gr.Markdown(f"Модель: {MODEL_NAME}")
+# Интерфейс Gradio
+with gr.Blocks(title="Польско-русский переводчик", theme="soft") as app:
+    gr.Markdown("## 🎯 Точный перевод с польского на русский")
+    gr.Markdown(f"**Модель:** `{MODEL_NAME}`")
     
     with gr.Row():
         with gr.Column():
-            input_text = gr.Textbox(label="Польский текст", placeholder="Введите текст на польском...")
-            num_translations = gr.Slider(1, MAX_TRANSLATIONS, value=DEFAULT_TRANSLATIONS, step=1, 
-                                       label="Количество вариантов перевода")
-            translate_btn = gr.Button("Перевести")
+            input_text = gr.Textbox(
+                label="Польский текст",
+                placeholder="Введите текст для перевода...",
+                lines=3
+            )
+            num_translations = gr.Slider(
+                minimum=1,
+                maximum=MAX_TRANSLATIONS,
+                value=DEFAULT_TRANSLATIONS,
+                step=1,
+                label="Количество вариантов"
+            )
+            translate_btn = gr.Button("Перевести", variant="primary")
         
         with gr.Column():
-            output_text = gr.Textbox(label="Основной вариант перевода", interactive=False)
-            variants = gr.Textbox(label="Все варианты перевода", lines=5, interactive=False)
+            output_text = gr.Textbox(
+                label="Лучш��й вариант",
+                interactive=False,
+                lines=3
+            )
+            variants = gr.Textbox(
+                label=f"Все варианты (макс. {MAX_TRANSLATIONS})",
+                interactive=False,
+                lines=5
+            )
     
     def update_outputs(text, num):
         translations = translate_text(text, num)
-        if not translations:
-            return "", ""
-        main_translation = translations[0]
-        all_variants = "\n\n".join(f"{i+1}. {t}" for i, t in enumerate(translations))
-        return main_translation, all_variants
+        return (
+            translations[0] if translations else "",
+            "\n\n".join(f"{i+1}. {t}" for i, t in enumerate(translations)) if translations else ""
+        )
     
     translate_btn.click(
         fn=update_outputs,
@@ -98,14 +131,21 @@ with gr.Blocks(title="Польско-русский переводчик") as ap
         outputs=[output_text, variants]
     )
     
-    examples = gr.Examples(
+    gr.Examples(
         examples=[
             ["Powiedzieć a zrobić to nie to samo.", 3],
             ["O jego propozycji nawet nie warto mówić.", 3],
-            ["Nie od razu Rzym zbudowano.", 2]
+            ["Nie od razu Rzym zbudowano.", 2],
+            ["To jest przykładowe zdanie do tłumaczenia.", 1]
         ],
-        inputs=[input_text, num_translations]
+        inputs=[input_text, num_translations],
+        label="Примеры для теста"
     )
 
 if __name__ == "__main__":
-    app.launch()
+    app.launch(
+        server_name="0.0.0.0",
+        server_port=7860,
+        share=False,
+        show_error=True
+    )