fruitpicker01 commited on
Commit
99fbfbd
·
verified ·
1 Parent(s): 3fe19fc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -9
app.py CHANGED
@@ -296,7 +296,7 @@ def process_message(message, product_name):
296
 
297
  if not matches:
298
  # Если совпадений нет, вернуть исходное сообщение с капитализацией предложений
299
- return capitalize_sentences(message)
300
 
301
  # Получаем позиции всех слов в сообщении
302
  word_positions = get_word_positions(message)
@@ -320,13 +320,6 @@ def process_message(message, product_name):
320
  start_char = word_positions[start_word_idx][1]
321
  end_char = word_positions[end_word_idx][2]
322
 
323
- # Проверяем, есть ли знаки препинания перед совпадением
324
- if start_char > 0 and final_message[start_char -1] in ['«', '»', '-', '–', '.', ',', '!', '?', ';', ':']:
325
- start_char -=1 # Включаем знак препинания в изменяемую часть
326
-
327
- # Проверяем, есть ли знаки препинания после совпадения
328
- if end_char < len(final_message) and final_message[end_char] in ['«', '»', '-', '–', '.', ',', '!', '?', ';', ':']:
329
- end_char +=1 # Включаем знак препинания в изменяемую часть
330
 
331
  # Извлечение изменяемой части
332
  matched_substring = final_message[start_char:end_char]
@@ -344,9 +337,15 @@ def process_message(message, product_name):
344
 
345
  # Сохраняем первое слово как есть (инфлектированное)
346
  first_word = words[0]
 
 
 
 
 
 
347
 
348
  # Остальные слова берем из оригинального названия продукта
349
- replaced_words = [first_word] + product_words_original[1:]
350
 
351
  # Собираем обратно измененную часть
352
  processed = ' '.join(replaced_words)
 
296
 
297
  if not matches:
298
  # Если совпадений нет, вернуть исходное сообщение с капитализацией предложений
299
+ return message
300
 
301
  # Получаем позиции всех слов в сообщении
302
  word_positions = get_word_positions(message)
 
320
  start_char = word_positions[start_word_idx][1]
321
  end_char = word_positions[end_word_idx][2]
322
 
 
 
 
 
 
 
 
323
 
324
  # Извлечение изменяемой части
325
  matched_substring = final_message[start_char:end_char]
 
337
 
338
  # Сохраняем первое слово как есть (инфлектированное)
339
  first_word = words[0]
340
+ clean_words = []
341
+ for i in range(len(product_words_original[1:])):
342
+ if product_words_original[1:][i] == "Карта":
343
+ clean_words.append(words[i+1])
344
+ else:
345
+ clean_words.append(product_words_original[1:][i])
346
 
347
  # Остальные слова берем из оригинального названия продукта
348
+ replaced_words = [first_word] + clean_words
349
 
350
  # Собираем обратно измененную часть
351
  processed = ' '.join(replaced_words)