Update app.py

app.py

@@ -1,71 +1,63 @@
 import gradio as gr
 import re
 
-# Функция для обработки входного текста
-def process_text(input_text, comment_count, pattern):
-    # Печать входных данных для отладки
-    print(f"Входные данные: {input_text}")
+# Функция для извлечения комментариев
+def extract_comments(input_text):
+    # Шаблон для разделения комментариев по ключевым словам
+    comment_blocks = re.split(r'Ответить|Действия с комментарием|Скрыть ответы|Показать перевод', input_text)
 
-    # Регулярное выражение для извлечения данных
-    if pattern == "Нравится":
-        pattern = r"([a-zA-Z0-9_@]+)\s*(.*?)(?:\sНравится:\s*(\d+))?\s*$"
-    elif pattern == "Лайков":
-        pattern = r"([a-zA-Z0-9_@]+)\s*(.*?)(?:\sЛайков:\s*(\d+))?\s*$"
-    elif pattern == "Like":
-        pattern = r"([a-zA-Z0-9_@]+)\s*(.*?)(?:\sLike:\s*(\d+))?\s*$"
-    elif pattern == "likes":
-        pattern = r"([a-zA-Z0-9_@]+)\s*(.*?)(?:\s(\d+)\s*likes)?\s*$"
-    elif pattern == "like":
-        pattern = r"([a-zA-Z0-9_@]+)\s*(.*?)(?:\s(\d+)\s*like)?\s*$"
-    elif pattern == "like +":
-        pattern = r"([a-zA-Z0-9_@]+)\s*(.*?)(?:\s(\d+)\s*like)?\s*\+\s*$"
-    elif pattern == "Плюс":
-        pattern = r"([a-zA-Z0-9_@]+)\s*(.*?)(?:\s(\d+)\s*)\s*\+\s*$"
-    elif pattern == "Минус":
-        pattern = r"([a-zA-Z0-9_@]+)\s*(.*?)(?:\s(\d+)\s*)\s*-\s*$"
-    elif pattern == "Равно":
-        pattern = r"([a-zA-Z0-9_@]+)\s*(.*?)(?:\s(\d+)\s*)\s*=\s*$"
-    elif pattern == "Точка":
-        pattern = r"([a-zA-Z0-9_@]+)\s*(.*?)(?:\s(\d+)\s*)\s*\.\s*$"
-    elif pattern == "Запятая":
-        pattern = r"([a-zA-Z0-9_@]+)\s*(.*?)(?:\s(\d+)\s*)\s*\,\s*$"
-    elif pattern == "Точка с запятой":
-        pattern = r"([a-zA-Z0-9_@]+)\s*(.*?)(?:\s(\d+)\s*)\s*\;\s*$"
-
-    # Проверка пустого ввода
-    if not input_text.strip():
-        return "Пожалуйста, введите текст."
-
-    # Поиск всех совпадений в тексте
-    matches = re.findall(pattern, input_text)
+    # Паттерны для извлечения данных
+    patterns = [
+        r"([a-zA-Z0-9_@]+)\s*(.*?)(?:\sНравится:\s*(\d+))?\s*$",  # Нравится
+        r"([a-zA-Z0-9_@]+)\s*(.*?)(?:\sЛайков:\s*(\d+))?\s*$",      # Лайков
+        r"([a-zA-Z0-9_@]+)\s*(.*?)(?:\sLike:\s*(\d+))?\s*$",        # Like
+        r"([a-zA-Z0-9_@]+)\s*(.*?)(?:\s(\d+)\s*likes)?\s*$",        # likes
+        r"([a-zA-Z0-9_@]+)\s*(.*?)(?:\s(\d+)\s*like)?\s*$",         # like
+    ]
     
-    # Печать найденных совпадений для отладки
-    print(f"Найденные совпадения: {matches}")
-
-    # Проверка наличия совпадений
-    if not matches:
-        return "Совпадения не найдены. Пожалуйста, проверьте ввод."
-
-    # Проверка количества комментариев
-    actual_comment_count = len(matches)
-    if actual_comment_count != int(comment_count):
-        return f"Предупреждение: найдено {actual_comment_count} комментариев, но указано {comment_count}."
+    results = []
+    
+    # Для каждого блока текста
+    for block in comment_blocks:
+        block = block.strip()  # Убираем лишние пробелы и пустые строки
+        if block:  # Если блок не пустой
+            for pattern in patterns:
+                match = re.search(pattern, block)
+                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})
+                    break  # Переходим к следующему блоку, как только нашли совпадение
 
-    # Формируем вывод
-    output = []
-    for i, match in enumerate(matches, 1):
-        username, text, likes = match
-        output.append(f'{i}. Пользователь: "{username}", Текст: "{text.strip()}", Лайков: {likes}')
+    return results
 
-    return "\n".join(output)
+# Функция для отображения результатов в удобном виде
+def display_results(input_text):
+    results = extract_comments(input_text)
+    
+    if not results:
+        return "Нет комментариев для отображения."
+    else:
+        display_text = ""
+        for result in results:
+            display_text += f"Пользователь: {result['Пользователь']}\n"
+            display_text += f"Текст: {result['Текст']}\n"
+            display_text += f"Лайков: {result['Лайков']}\n"
+            display_text += "-" * 50 + "\n"
+        return display_text
 
-# Настройка Gradio интерфейса
+# Создание интерфейса с Gradio
 iface = gr.Interface(
-    fn=lambda input_text, comment_count, pattern: process_text(input_text, comment_count, pattern),  
-    inputs=[gr.Textbox(lines=10, placeholder="Введите текст сюда..."), 
-            gr.Number(label="Количество комментариев", value=0), 
-            gr.Dropdown(label="Шаблон", choices=["Нравится", "Лайков", "Like", "likes", "like", "like +", "Плюс", "Минус", "Равно", "Точка", "Запятая", "Точка с запятой"])],  
-    outputs=gr.Textbox(lines=10, placeholder="Результат...")
+    fn=display_results,  # Функция, которая будет вызываться
+    inputs="text",        # Текстовый ввод для пользователей
+    outputs="text",       # Вывод результатов в текстовом формате
+    title="Анализ комментариев",  # Название интерфейса
+    description="Введите текст комментариев, и система автоматически разделит их, извлекая информацию о пользователях и лайках.",
+    theme="default"       # Тема оформления
 )
 
 # Запуск интерфейса