Update app.py

app.py

@@ -1,47 +1,51 @@
-import re
 import gradio as gr
+import re
 
-def parse_comments(raw_text):
-    # Разбиваем текст на отдельные комментарии по "Ответить", что указывает на конец каждого блока комментария
-    comment_blocks = raw_text.split("Ответить")
-    
-    results = []
-    for block in comment_blocks:
-        # Находим никнейм пользователя, начиная с начала строки до первого не-буквенного символа или цифры
-        user_match = re.search(r'^([a-zA-Z0-9_]+)', block.strip())
-        # Находим текст комментария, который начинается после никнейма и перед временем ("1 нед." или "1 дн.")
-        text_match = re.search(r'@\w+\s+(.+?)(?:\d+\s(?:нед|дн)\.)', block)
-        # Находим количество лайков, если оно указано
-        likes_match = re.search(r'Нравится:\s*(\d+)', block)
-        
-        if user_match and text_match:
-            user = user_match.group(1)
-            text = text_match.group(1).strip()
-            likes = int(likes_match.group(1)) if likes_match else 0
-            
-            # Добавляем результат в список только если текст не пустой
-            if text:
-                results.append({"Пользователь": user, "Текст": text, "Лайков": likes})
+# Функция для обработки входного текста
+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)
     
-    return results
+    # Печать найденных совпадений для отладки
+    print(f"Найденные совпадения: {matches}")
+
+    # Проверка наличия совпадений
+    if not matches:
+        return "Совпадения не найдены. Пожалуйста, проверьте ввод."
+
+    # Проверка количества комментариев
+    actual_comment_count = len(matches)
+    if actual_comment_count != int(comment_count):
+        return (f"Предупреждение: найдено {actual_comment_count} комментариев, "
+                f"хотя указано {comment_count}.\n"
+                "Пожалуйста, проверьте ввод.")
 
-def format_output(parsed_comments):
-    # Форматируем вывод в текстовом формате
-    formatted_output = ""
-    for comment in parsed_comments:
-        formatted_output += f"Пользователь: {comment['Пользователь']}\n"
-        formatted_output += f"Текст: {comment['Текст']}\n"
-        formatted_output += f"Лайков: {comment['Лайков']}\n\n"
-    return formatted_output
+    # Формируем вывод
+    output = []
+    for i, match in enumerate(matches, 1):
+        username, text, likes = match
+        output.append(f'{i}. Пользователь: "{username}", Текст: "{text.strip()}", Лайков: {likes or 0}')
 
-def process_text(raw_text):
-    parsed_comments = parse_comments(raw_text)
-    return format_output(parsed_comments)
+    return "\n".join(output)
 
 # Настройка Gradio интерфейса
 iface = gr.Interface(
     fn=process_text,
-    inputs=gr.Textbox(lines=10, placeholder="Введите текст сюда..."),
+    inputs=[
+        gr.Textbox(lines=10, placeholder="Введите текст сюда..."),       # Поле для текста комментариев
+        gr.Number(label="Количество комментариев по данным Instagram")  # Поле для количества комментариев
+    ],
     outputs=gr.Textbox(lines=10, placeholder="Результат...")
 )