Update app.py

app.py

@@ -4,37 +4,36 @@ import gradio as gr
 import re
 import json
 
-# Инициализация модели и токенизатора
+# Инициализация модели
 def load_model():
     tokenizer = AutoTokenizer.from_pretrained(
         "cointegrated/rut5-base-multitask",
-        legacy=False  # Отключаем предупреждения
+        legacy=False
     )
     model = T5ForConditionalGeneration.from_pretrained(
         "cointegrated/rut5-base-multitask",
-        torch_dtype=torch.float16  # Экономия памяти
+        torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32
     )
     return tokenizer, model
 
 tokenizer, model = load_model()
 
-# Основная функция генерации
-def generate_seo_tags(description):
+# Функция генерации мета-тегов
+def generate_meta_tags(description):
     try:
-        # Подготовка промта
+        # Формируем четкий промт
         prompt = f"""
-Сгенерируй SEO-метатеги title и description для товара.
+Сгенерируй SEO-метатеги для товара на основе описания.
 Требования:
-- Title: до 60 символов, содержит ключевые слова
-- Description: до 160 символов, описание преимуществ
+- Title: до 60 символов, ключевые слова
+- Description: до 160 символов, преимущества
 
 Описание товара:
 {description.strip()}
 
-Выведи только JSON:
+Выведи только JSON-объект:
 {{"title": "...", "description": "..."}}
 """
-        # Токенизация и генерация
         inputs = tokenizer(prompt, return_tensors="pt", truncation=True, max_length=512)
         
         with torch.no_grad():
@@ -47,21 +46,22 @@ def generate_seo_tags(description):
                 early_stopping=True
             )
         
-        # Декодирование и очистка результата
+        # Обработка результата
         result = tokenizer.decode(outputs[0], skip_special_tokens=True)
         json_match = re.search(r'\{.*\}', result, re.DOTALL)
         
         if json_match:
-            seo_tags = json.loads(json_match.group())
-            # Проверка и обрезка по длине
-            seo_tags["title"] = seo_tags.get("title", "")[:60]
-            seo_tags["description"] = seo_tags.get("description", "")[:160]
-            return seo_tags
-        
+            tags = json.loads(json_match.group())
+            # Проверяем и обрезаем длину
+            return {
+                "title": tags.get("title", "")[:60],
+                "description": tags.get("description", "")[:160]
+            }
+    
     except Exception as e:
-        print(f"Ошибка: {e}")
+        print(f"Error: {e}")
     
-    # Фоллбэк если что-то пошло не так
+    # Fallback вариант
     clean_text = re.sub(r'[^\w\s]', '', description)
     return {
         "title": clean_text[:60],
@@ -70,32 +70,25 @@ def generate_seo_tags(description):
 
 # Интерфейс Gradio
 with gr.Blocks(title="SEO Генератор") as app:
-    gr.Markdown("## Генератор SEO-метатегов для товаров")
+    gr.Markdown("## Генератор мета-тегов для товаров")
     
     with gr.Row():
         with gr.Column():
-            input_text = gr.Textbox(
-                label="Описание товара",
-                placeholder="Введите описание товара...",
+            input_desc = gr.Textbox(
+                label="Введите описание товара",
+                placeholder="Например: Смартфон Samsung Galaxy S23...",
                 lines=5
             )
-            submit_btn = gr.Button("Сгенерировать")
+            btn = gr.Button("Сгенерировать", variant="primary")
         
         with gr.Column():
-            output_json = gr.JSON(
-                label="SEO-метатеги",
-                interactive=False
-            )
+            output_tags = gr.JSON(label="Результат")
     
-    submit_btn.click(
-        fn=generate_seo_tags,
-        inputs=input_text,
-        outputs=output_json
+    btn.click(
+        fn=generate_meta_tags,
+        inputs=input_desc,
+        outputs=output_tags
     )
 
 if __name__ == "__main__":
-    app.launch(
-        server_name="0.0.0.0",
-        server_port=7860,
-        show_error=True
-    )
+    app.launch(server_name="0.0.0.0", server_port=7860)