Update app.py

app.py

@@ -28,22 +28,45 @@ def extract_comment_data(comment_text: str) -> Tuple[Optional[str], Optional[str
         if 'отметок "Нравится"' in comment_text:
             return None, None, 0, 0
 
-        # Извлекаем имя пользователя
-        username_match = re.search(r"Фото профиля ([^\n]+)", comment_text)
-        if not username_match:
-            return None, None, 0, 0
+        # Извлекаем имя пользователя (поддержка обоих форматов)
+        username = None
+        
+        # Формат 1: "Фото профиля username\nusername\n"
+        if "Фото профиля" in comment_text:
+            username_match = re.search(r"Фото профиля ([^\n]+)", comment_text)
+            if username_match:
+                username = username_match.group(1).strip()
         
-        username = username_match.group(1).strip()
+        # Формат 2: Прямое указание имени пользователя
+        else:
+            lines = comment_text.split('\n')
+            if lines and lines[0].strip():
+                username = lines[0].strip()
+
+        if not username:
+            return None, None, 0, 0
         
         # Извлекаем текст комментария
         lines = comment_text.split('\n')
         comment = ""
+        
+        # Ищем комментарий после времени
+        time_pattern = r'\d+\s*(?:ч\.|нед\.)'
         for i, line in enumerate(lines):
-            if username in line and i + 1 < len(lines):
-                comment = lines[i + 1].strip()
-                comment = re.sub(r'\d+\s*(?:ч\.|нед\.)\s*$', '', comment)
-                comment = re.sub(r'"Нравится":\s*\d+\s*Ответить\s*$', '', comment)
+            if re.search(time_pattern, line):
+                if i + 1 < len(lines):
+                    comment = lines[i + 1].strip()
                 break
+            elif username in line and i + 2 < len(lines):
+                # Проверяем следующую строку после юзернейма
+                next_line = lines[i + 1].strip()
+                if not re.search(time_pattern, next_line):
+                    comment = next_line
+                    break
+        
+        # Очищаем комментарий
+        comment = re.sub(r'\d+\s*(?:ч\.|нед\.)\s*$', '', comment)
+        comment = re.sub(r'"Нравится":\s*\d+\s*Ответить\s*$', '', comment)
         
         # Извлекаем лайки
         likes_match = re.search(r'"Нравится":\s*(\d+)', comment_text)