Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -296,7 +296,7 @@ def process_message(message, product_name):
|
|
296 |
|
297 |
if not matches:
|
298 |
# Если совпадений нет, вернуть исходное сообщение с капитализацией предложений
|
299 |
-
return
|
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] +
|
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)
|