Оновлено модуль логування MAI-DX: повернуто вивід в консоль, вдосконалено обробку захопленого тексту та покращено документацію. Виправлено помилки в парсері для надійнішого захоплення розмов агентів.

enhanced_mai_dx_logger.py

@@ -1,7 +1,7 @@
 #!/usr/bin/env python3
 """
 Покращений модуль для повного логування діагностичних сесій MAI-DX 
-з детальним захопленням розмов між агентами. (ВЕРСІЯ З ВИПРАВЛЕНИМ ПАРСЕРОМ)
+з детальним захопленням розмов між агентами. (ВЕРСІЯ З ВИПРАВЛЕНИМ ПАРСЕРОМ І ПОВЕРНЕНИМ ВИВОДОМ)
 """
 import os
 import io
@@ -63,9 +63,13 @@ class EnhancedOutputCapture:
         sys.stdout = self.original_stdout
         
     def write(self, text):
+        """
+        FIX: Повертаємо вивід в консоль.
+        Цей метод тепер і записує вивід у буфер для парсингу,
+        і дублює його в оригінальну консоль.
+        """
         self.captured_output.write(text)
-        # Уникаємо подвійного виводу в консоль, якщо Gradio робить це сам
-        # self.original_stdout.write(text) 
+        self.original_stdout.write(text)
         
     def flush(self):
         self.original_stdout.flush()
@@ -96,7 +100,6 @@ class MAIDxConversationLogger:
         
     def finalize_and_save_session(self, session: DiagnosisSession, result: Any, raw_output: str, duration: float) -> DiagnosisSession:
         session.raw_output = raw_output
-        # Головний виклик, що парсить лог
         self._parse_captured_output(session, raw_output)
         
         session.diagnosis = getattr(result, 'final_diagnosis', 'N/A')
@@ -112,10 +115,6 @@ class MAIDxConversationLogger:
         return session
 
     def _parse_captured_output(self, session: DiagnosisSession, captured_text: str):
-        """
-        FIXED: Повністю перероблений, більш надійний парсер, який не залежить 
-        від рядка "Starting Diagnostic Loop".
-        """
         lines = captured_text.split('\n')
         current_round: Optional[AgentConversation] = None
         current_agent: Optional[str] = None
@@ -126,10 +125,8 @@ class MAIDxConversationLogger:
         for line in lines:
             stripped_line = line.strip()
 
-            # Визначаємо початок блоку агента
             if "Agent Name" in line and ("╭" in line or "┌" in line):
-                # Якщо це перший агент у раунді, створюємо новий раунд
-                if not current_round or in_agent_output: # in_agent_output check for nested blocks
+                if not current_round or in_agent_output:
                     round_number += 1
                     if current_round:
                         current_round.end_time = datetime.now().isoformat()
@@ -143,16 +140,13 @@ class MAIDxConversationLogger:
                     buffer = []
                 continue
 
-            # Визначаємо кінець блоку агента
             if in_agent_output and ("╰" in line or "└" in line):
                 if buffer and current_agent and current_round:
                     self._add_agent_message(current_round, current_agent, '\n'.join(buffer))
                 in_agent_output, current_agent, buffer = False, None, []
                 continue
 
-            # Збираємо вміст блоку
             if in_agent_output:
-                # Очищуємо рядки від символів рамок
                 clean_line = re.sub(r'^[│|]\s*|\s*[│|]\s*$', '', line)
                 buffer.append(clean_line)
                 continue
@@ -162,10 +156,8 @@ class MAIDxConversationLogger:
 
     def _add_agent_message(self, conversation: AgentConversation, agent_name: str, content: str):
         message_type = "output"
-        # Видаляємо зайві порожні рядки на початку та в кінці
         formatted_content = content.strip()
 
-        # Покращуємо визначення типу повідомлення
         if "Structured Output - Attempting Function Call Execution" in formatted_content:
             message_type = "function_call"
         elif "Score:" in formatted_content and "Justification:" in formatted_content: