Update app.py

app.py

@@ -7,35 +7,33 @@ def process_text(input_text, comment_count):
     print(f"Входные данные: {input_text}")
     print(f"Ожидаемое количество комментариев: {comment_count}")
 
-    # Регулярное выражение для извлечения данных
-    pattern = r"([a-zA-Z0-9_@]+(?:[a-zA-Z0-9_]+)?)\s*(.*?)\s*(?:Нравится:\s*(\d+))"
-
-    # Проверка пустого ввода
-    if not input_text.strip():
-        return "Пожалуйста, введите текст."
-
-    # Поиск всех совпадений в тексте
-    matches = re.findall(pattern, input_text)
+    # Шаблон для разделения комментариев по ключевым словам
+    comment_blocks = re.split(r'Ответить|Действия с комментарием|Скрыть ответы|Показать перевод', input_text)
+    
+    # Паттерн для извлечения данных из каждого блока
+    pattern = r"([a-zA-Z0-9_@]+)\s*(.*?)(?:\sНравится:\s*(\d+))?\s*$"
+    
+    results = []
+    for block in comment_blocks:
+        match = re.search(pattern, block.strip())
+        if match:
+            username = match.group(1)
+            text = match.group(2).strip()
+            likes = int(match.group(3)) if match.group(3) else 0
+            results.append({"Пользователь": username, "Текст": text, "Лайков": likes})
     
-    # Печать найденных совпадений для отладки
-    print(f"Найденные совпадения: {matches}")
-
-    # Проверка наличия совпадений
-    if not matches:
-        return "Совпадения не найдены. Пожалуйста, проверьте ввод."
-
     # Проверка количества комментариев
-    actual_comment_count = len(matches)
+    actual_comment_count = len(results)
     if actual_comment_count != int(comment_count):
         return (f"Предупреждение: найдено {actual_comment_count} комментариев, "
                 f"хотя указано {comment_count}.\n"
                 "Пожалуйста, проверьте ввод.")
-
+    
     # Формируем вывод
     output = []
-    for i, match in enumerate(matches, 1):
-        username, text, likes = match
-        output.append(f'{i}. Пользователь: "{username}", Текст: "{text.strip()}", Лайков: {likes or 0}')
+    for i, result in enumerate(results, 1):
+        username, text, likes = result['Пользователь'], result['Текст'], result['Лайков']
+        output.append(f'{i}. Пользователь: "{username}", Текст: "{text}", Лайков: {likes}')
 
     return "\n".join(output)