Update app.py

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 import re
 
 # Функция для извлечения комментариев
-def extract_comments(input_text):
+def extract_comments(input_text, expected_count):
     # Шаблон для разделения комментариев по ключевым словам
     comment_blocks = re.split(r'Ответить|Действия с комментарием|Скрыть ответы|Показать перевод', input_text)
 
@@ -32,16 +32,9 @@ def extract_comments(input_text):
                     # Добавляем в результаты
                     results.append({"Пользователь": username, "Текст": text, "Лайков": likes})
                     break  # Переходим к следующему блоку, как только нашли совпадение
-
-    return results
-
-# Функция для отображения результатов в удобном виде
-def display_results(input_text):
-    results = extract_comments(input_text)
     
-    if not results:
-        return "Нет комментариев для отображения."
-    else:
+    # Проверяем, совпадает ли количество комментариев с ожидаемым
+    if len(results) == expected_count:
         display_text = ""
         for result in results:
             display_text += f"Пользователь: {result['Пользователь']}\n"
@@ -49,15 +42,25 @@ def display_results(input_text):
             display_text += f"Лайков: {result['Лайков']}\n"
             display_text += "-" * 50 + "\n"
         return display_text
+    else:
+        return f"Ошибка! Количество найденных комментариев: {len(results)}, ожидалось: {expected_count}."
+
+
+# Функция для отображения результатов в удобном виде
+def display_results(input_text, expected_count):
+    return extract_comments(input_text, expected_count)
 
 # Создание интерфейса с Gradio
 iface = gr.Interface(
     fn=display_results,  # Функция, которая будет вызываться
-    inputs="text",        # Текстовый ввод для пользователей
-    outputs="text",       # Вывод результатов в текстовом формате
+    inputs=[
+        gr.Textbox(label="Введите текст комментариев", lines=10),  # Текстовый ввод для комментариев
+        gr.Number(label="Ожидаемое количество комментариев", value=94)  # Ввод для количества комментариев
+    ],  
+    outputs="text",  # Вывод результатов в текстовом формате
     title="Анализ комментариев",  # Название интерфейса
-    description="Введите текст комментариев, и система автоматически разделит их, извлекая информацию о пользователях и лайках.",
-    theme="default"       # Тема оформления
+    description="Введите текст комментариев и ожидаемое количество, чтобы проверить, совпадает ли результат с ожиданиями.",
+    theme="default"  # Тема оформления
 )
 
 # Запуск интерфейса