Update app.py

app.py

@@ -1,3 +1,4 @@
+from transformers import pipeline
 from dataclasses import dataclass, field
 from typing import List, Optional, Dict, Any
 import re
@@ -17,7 +18,6 @@ logger = logging.getLogger(__name__)
 class Comment:
     """
     Представляет комментарий Instagram со всеми метаданными и вложенной структурой.
-
     Attributes:
         id: Уникальный идентификатор комментария
         username: Имя пользователя
@@ -69,7 +69,6 @@ class InstagramCommentAnalyzer:
     def __init__(self, max_depth: int = 10, max_comment_length: int = 2200):
         """
         Инициализация анализатора с настраиваемыми параметрами.
-
         Args:
             max_depth: Максимальная глубина вложенности комментариев
             max_comment_length: Максимальная длина комментария
@@ -88,13 +87,25 @@ class InstagramCommentAnalyzer:
             'processed_hashtags': 0
         }
 
+        # Инициализация модели Hugging Face для анализа настроений
+        self.sentiment_analyzer = pipeline("sentiment-analysis")
+
+    def analyze_sentiment(self, text: str) -> str:
+        """
+        Анализ настроений в комментарии с использованием модели Hugging Face.
+        Args:
+            text: Текст комментария
+        Returns:
+            Строка с меткой настроения ('POSITIVE' или 'NEGATIVE')
+        """
+        result = self.sentiment_analyzer(text)
+        return result[0]['label']
+
     def normalize_text(self, text: str) -> str:
         """
         Нормализация входного текста.
-
         Args:
             text: Исходный текст
-
         Returns:
             Нормализованный текст
         """
@@ -109,7 +120,6 @@ class InstagramCommentAnalyzer:
     def extract_metadata(self, comment: Comment) -> None:
         """
         Извлечение метаданных из комментария.
-
         Args:
             comment: Объект комментария
         """
@@ -127,12 +137,10 @@ class InstagramCommentAnalyzer:
     def process_comment(self, text: str, parent_id: Optional[str] = None, level: int = 0) -> Optional[Comment]:
         """
         Обработка отдельного комментария.
-
         Args:
             text: Текст комментария
             parent_id: ID родительского комментария
             level: Уровень вложенности
-
         Returns:
             Обработанный объект Comment или None
         """
@@ -164,6 +172,9 @@ class InstagramCommentAnalyzer:
                 self.stats['truncated_comments'] += 1
                 comment.content = comment.content[:self.max_comment_length] + "..."
 
+            # Добавление анализа настроений
+            comment.sentiment = self.analyze_sentiment(comment.content)
+
             self.extract_metadata(comment)
             self.stats['total_comments'] += 1
             return comment
@@ -182,11 +193,9 @@ class InstagramCommentAnalyzer:
     def format_comment(self, comment: Comment, index: int) -> str:
         """
         Форматирование комментария для вывода.
-
         Args:
             comment: Объект комментария
             index: Номер комментария
-
         Returns:
             Отформатированная строка комментария
         """
@@ -195,16 +204,14 @@ class InstagramCommentAnalyzer:
 
         return (
             f'{index}. "{comment.username}" "{comment.time}" '
-            f'"{comment.content}" "Нравится {comment.likes}"'
+            f'"{comment.content}" "Нравится {comment.likes}" "Настроение {comment.sentiment}"'
         )
 
     def process_comments(self, text: str) -> List[str]:
         """
         Обработка всех комментариев в тексте.
-
         Args:
             text: Исходный текст с комментариями
-
         Returns:
             Список отформатированных комментариев
         """
@@ -234,28 +241,15 @@ def main():
     Пример использования анализатора.
     """
     # Пример входного текста
-    input_text = """
-    0001.minakov 53 нед.А что такое для,Вас Василина,волшебство?Отметки "Нравится": 3ОтветитьНравится
-    aliyllmn 54 нед.seni seviyorum:)Отметки "Нравится": 2ОтветитьПоказать переводНравится
+    example_text = """
+    user1 2 нед. This is a positive comment! Отметки "Нравится": 25
+    user2 3 нед. This is a negative comment! Отметки "Нравится": 5
     """
 
-    # Создание экземпляра анализатора
-    analyzer = InstagramCommentAnalyzer(max_depth=10)
-
-    # Обработка комментариев
-    results = analyzer.process_comments(input_text)
-
-    # Вывод результатов
-    print("\nОбработанные комментарии:")
-    print("-" * 50)
-    for result in results:
-        print(result)
-
-    # Вывод статистики
-    print("\nСтатистика обработки:")
-    print("-" * 50)
-    for key, value in analyzer.stats.items():
-        print(f"{key}: {value}")
+    analyzer = InstagramCommentAnalyzer()
+    formatted_comments = analyzer.process_comments(example_text)
+    for formatted_comment in formatted_comments:
+        print(formatted_comment)
 
 if __name__ == "__main__":
     main()