Update app.py

app.py

@@ -7,16 +7,12 @@ tokenizer = AutoTokenizer.from_pretrained("cointegrated/rut5-base-multitask")
 model = T5ForConditionalGeneration.from_pretrained("cointegrated/rut5-base-multitask")
 
 # Функция генерации мета-тегов
-def generate_meta_tags(description):
-    # Очищаем описание
-    description = description.strip()
-
-    # Формируем промт
+    # Формируем чёткий и понятный промт
     prompt = """
 Описание товара:
 {description}
 
-Задача: Создай SEO-оптимизированные мета-теги из описания товара.
+Задача: создать SEO мета-теги для этого товара.
 Формат вывода:
 {"title": "SEO заголовок до 60 символов", "description": "SEO описание до 160 символов"}
 """.format(description=description)
@@ -31,18 +27,22 @@ def generate_meta_tags(description):
             do_sample=True,
             temperature=0.7
         )
-    generated_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
-
-    # Принудительно исправляем JSON, если он сломан
-    import re
-    match = re.search(r'\{.*\}', generated_text, re.DOTALL)
-    if match:
-        generated_text = match.group(0)
-
-    generated_text = generated_text.replace('\n', ' ').replace('“','"').replace('”','"')
-    print("Prompt:", prompt)
-    print("Generated text:", generated_text)
-    return generated_text
+
+    generated_text = tokenizer.decode(outputs[0], skip_special_tokens=True).strip()
+
+    # Проверяем, является ли вывод JSON
+    if generated_text.startswith("{") and generated_text.endswith("}"):
+        # Если JSON валидный — возвращаем как есть
+        return generated_text.replace('\n', ' ').replace('“', '"').replace('”', '"')
+    else:
+        # Если не получилось — формируем безопасный вариант
+        title = description[:60]  # Берём начало описания как title
+        description_short = description[:160]
+
+        return '{"title": "%s", "description": "%s"}' % (
+            title.replace('"', "").replace("\\", ""),
+            description_short.replace('"', "").replace("\\", "")
+        )
 
 
 # Интерфейс Gradio