fruitpicker01 commited on
Commit
2ff2fc0
·
verified ·
1 Parent(s): a486861

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +15 -7
app.py CHANGED
@@ -422,16 +422,24 @@ def process_message(message, product_name):
422
  continue
423
 
424
  # Сохраняем первое слово как есть (инфлектированное)
425
- first_word = words[0]
426
- clean_words = []
427
- for i in range(len(product_words_original[1:])):
428
- if product_words_original[1:][i] == "Карта":
429
- clean_words.append(words[i+1])
 
 
430
  else:
431
- clean_words.append(product_words_original[1:][i])
 
 
 
 
 
 
432
 
433
  # Остальные слова берем из оригинального названия продукта
434
- replaced_words = [first_word] + clean_words
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)