Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -422,16 +422,24 @@ def process_message(message, product_name):
|
|
422 |
continue
|
423 |
|
424 |
# Сохраняем первое слово как есть (инфлектированное)
|
425 |
-
first_word = words[0]
|
426 |
-
|
427 |
-
for i in range(len(product_words_original[1:])):
|
428 |
-
|
429 |
-
|
|
|
|
|
430 |
else:
|
431 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
432 |
|
433 |
# Остальные слова берем из оригинального названия продукта
|
434 |
-
replaced_words = [first_word] +
|
435 |
|
436 |
# Собираем обратно измененную часть
|
437 |
processed = ' '.join(replaced_words)
|
|
|
422 |
continue
|
423 |
|
424 |
# Сохраняем первое слово как есть (инфлектированное)
|
425 |
+
first_word = words[0] # «зарплатным»
|
426 |
+
others = []
|
427 |
+
for i in range(len(product_words_original[1:])): # product_name без первого слова
|
428 |
+
# Если второе слово в продукте = «Карта», то вставляем ровно его
|
429 |
+
if product_words_original[i+1] == "Карта":
|
430 |
+
# Берём слово из оригинальной фразы
|
431 |
+
others.append(words[i+1])
|
432 |
else:
|
433 |
+
# Смотрим, с какой буквы начинается?
|
434 |
+
# Если первая буква строчная, берём исходное (words[i+1])
|
435 |
+
# Иначе берём из product_words_original
|
436 |
+
if words[i+1] and words[i+1][0].islower():
|
437 |
+
others.append(words[i+1]) # Использовать форму, которую придумала модель
|
438 |
+
else:
|
439 |
+
others.append(product_words_original[i+1])
|
440 |
|
441 |
# Остальные слова берем из оригинального названия продукта
|
442 |
+
replaced_words = [first_word] + others
|
443 |
|
444 |
# Собираем обратно измененную часть
|
445 |
processed = ' '.join(replaced_words)
|