Update app.py

app.py

@@ -4,97 +4,68 @@ import gradio as gr
 import re
 import json
 
-# Инициализация модели
-def load_model():
-    tokenizer = AutoTokenizer.from_pretrained(
-        "cointegrated/rut5-base-multitask",
-        legacy=False
-    )
-    model = T5ForConditionalGeneration.from_pretrained(
-        "cointegrated/rut5-base-multitask",
-        torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32
-    )
-    return tokenizer, model
+tokenizer = AutoTokenizer.from_pretrained("cointegrated/rut5-base-multitask", legacy=False)
+model = T5ForConditionalGeneration.from_pretrained("cointegrated/rut5-base-multitask")
 
-tokenizer, model = load_model()
+def smart_truncate(text, max_len):
+    if len(text) <= max_len:
+        return text
+    return text[:text[:max_len+1].rfind(' ')].strip()
 
-# Функция генерации мета-тегов
-def generate_meta_tags(description):
-    try:
-        # Формируем четкий промт
-        prompt = f"""
-Напиши SEO-метатеги для товара на основе описания ниже.
-Требования STRICT:
-1. Title: ровно 55-60 символов, не обрезать слова
-2. Description: ровно 150-160 символов, не обрезать слова
-3. Обязательно включи: 
-   - Для title: бренд, модель, тип товара
-   - Для description: главные преимущества
-4. Используй продающий язык с упором на выгоды
-5. Никаких технических подробностей в title
-6. Не обрезай слова
+def generate_meta(description):
+    # Упрощенный промт с примерами
+    prompt = f"""
+Описание: {description.strip()}
+
+→ Сгенерируй SEO title (до 60 символов) и description (до 160 символов) в JSON. 
 
-Описание товара:
-{description.strip()}
+Примеры:
+title: "Аккумулятор Fenix ARB-L18-4000U 18650 - 4000 мАч"
+description: "Литий-ионный аккумулятор для фонарей Fenix. 3 вида зарядки, защита от перегрева."
 
-Выведи только JSON-объект:
+Вывод ТОЛЬКО как JSON:
 {{"title": "...", "description": "..."}}
 """
-        inputs = tokenizer(prompt, return_tensors="pt", truncation=True, max_length=512)
-        
-        with torch.no_grad():
-            outputs = model.generate(
-                **inputs,
-                max_new_tokens=200,
-                num_beams=7,
-                do_sample=False,
-                temperature=0.5,
-                early_stopping=True
-            )
-        
-        # Обработка результата
-        result = tokenizer.decode(outputs[0], skip_special_tokens=True)
-        json_match = re.search(r'\{.*\}', result, re.DOTALL)
-        
-        if json_match:
-            tags = json.loads(json_match.group())
-            # Проверяем и обрезаем длину
-            return {
-                "title": tags.get("title", "")[:60],
-                "description": tags.get("description", "")[:160]
-            }
+    inputs = tokenizer(prompt, return_tensors="pt", max_length=512, truncation=True)
     
-    except Exception as e:
-        print(f"Error: {e}")
+    with torch.no_grad():
+        outputs = model.generate(
+            **inputs,
+            max_new_tokens=200,
+            num_beams=5,
+            early_stopping=True,
+            no_repeat_ngram_size=2
+        )
     
-    # Fallback вариант
-    clean_text = re.sub(r'[^\w\s]', '', description)
-    return {
-        "title": clean_text[:60],
-        "description": clean_text[:160]
-    }
+    try:
+        result = tokenizer.decode(outputs[0], skip_special_tokens=True)
+        json_data = json.loads(re.search(r'\{.*\}', result, re.DOTALL).group())
+        
+        # Принудительная постобработка
+        if "Fenix" not in json_data["title"]:
+            json_data["title"] = f"Аккумулятор Fenix {json_data['title']}"
+            
+        return {
+            "title": smart_truncate(json_data["title"], 60),
+            "description": smart_truncate(json_data["description"], 160)
+        }
+    except:
+        # Фоллбэк
+        clean_text = re.sub(r'\s+', ' ', description)
+        return {
+            "title": smart_truncate(f"Аккумулятор Fenix {clean_text}", 60),
+            "description": smart_truncate(clean_text, 160)
+        }
 
-# Интерфейс Gradio
-with gr.Blocks(title="SEO Генератор") as app:
-    gr.Markdown("## Генератор мета-тегов для товаров")
-    
+# Интерфейс
+with gr.Blocks() as app:
+    gr.Markdown("## Генератор метатегов (контроль длины)")
+    inp = gr.Textbox(label="Описание товара", lines=7)
+    btn = gr.Button("Сгенерировать")
     with gr.Row():
-        with gr.Column():
-            input_desc = gr.Textbox(
-                label="Введите описание товара",
-                placeholder="Например: Смартфон Samsung Galaxy S23...",
-                lines=5
-            )
-            btn = gr.Button("Сгенерировать", variant="primary")
-        
-        with gr.Column():
-            output_tags = gr.JSON(label="Результат")
+        out_title = gr.Textbox(label="Title (до 60 символов)", interactive=False)
+        out_desc = gr.Textbox(label="Description (до 160 символов)", lines=3, interactive=False)
     
-    btn.click(
-        fn=generate_meta_tags,
-        inputs=input_desc,
-        outputs=output_tags
-    )
+    btn.click(generate_meta, inputs=inp, outputs=[out_title, out_desc])
 
-if __name__ == "__main__":
-    app.launch(server_name="0.0.0.0", server_port=7860)
+app.launch()