Update app.py

app.py

@@ -9,12 +9,12 @@ import os
 from dotenv import load_dotenv
 import threading
 
-#load_dotenv()
 
 class OutputParser:
     def __init__(self):
         self.buffer = ""
         self.current_agent = None
+        self.final_article_sent = False  # Add this flag
         
     def parse_output(self, text: str) -> List[ChatMessage]:
         messages = []
@@ -31,7 +31,7 @@ class OutputParser:
                 metadata={"title": f"🤖 {self.current_agent}"}
             ))
         
-        # Look for task information with full task list
+        # Look for task information
         task_match = re.search(r'\[INFO\]: == Starting Task: (.*?)(?=\n\n|\n> Entering|$)', cleaned_text, re.DOTALL)
         if task_match and self.current_agent:
             task_content = task_match.group(1).strip()
@@ -41,43 +41,31 @@ class OutputParser:
                 metadata={"title": f"📋 Task for {self.current_agent}"}
             ))
         
-        # Look for thought processes
-        thought_match = re.search(r'Thought: (.*?)(?=\nAction:|$)', cleaned_text, re.DOTALL)
-        if thought_match and self.current_agent:
-            thought_content = thought_match.group(1).strip()
-            messages.append(ChatMessage(
-                role="assistant",
-                content=thought_content,
-                metadata={"title": f"💭 {self.current_agent}'s Thoughts"}
-            ))
-        
-        # Look for final answers from non-Editor agents
-        if "Final Answer:" in cleaned_text and self.current_agent != "Editor":
+        # Look for final answers from all agents
+        if "Final Answer:" in cleaned_text:
             answer_match = re.search(r'Final Answer:\s*(.*?)(?=\n> Finished chain|$)', cleaned_text, re.DOTALL)
             if answer_match:
                 answer_content = answer_match.group(1).strip()
-                messages.append(ChatMessage(
-                    role="assistant",
-                    content=answer_content,
-                    metadata={"title": f"💡 Output from {self.current_agent}"}
-                ))
-        
-        # Special handling for Editor's final answer (the final article)
-        elif "Final Answer:" in cleaned_text and self.current_agent == "Editor":
-            answer_match = re.search(r'Final Answer:\s*(.*?)(?=\n> Finished chain|$)', cleaned_text, re.DOTALL)
-            if answer_match:
-                answer_content = answer_match.group(1).strip()
-                # First send the metadata marker
-                messages.append(ChatMessage(
-                    role="assistant",
-                    content="Final article is ready!",
-                    metadata={"title": "📝 Final Article"}
-                ))
-                # Then send the actual content without metadata
-                messages.append(ChatMessage(
-                    role="assistant",
-                    content=answer_content
-                ))
+                
+                if self.current_agent == "Editor" and not self.final_article_sent:
+                    # This is the final article - send only once
+                    messages.append(ChatMessage(
+                        role="assistant",
+                        content="Final Article",
+                        metadata={"title": "📝 Final Article"}
+                    ))
+                    messages.append(ChatMessage(
+                        role="assistant",
+                        content=answer_content
+                    ))
+                    self.final_article_sent = True
+                elif self.current_agent != "Editor":
+                    # This is an intermediate output (Planner or Writer)
+                    messages.append(ChatMessage(
+                        role="assistant",
+                        content=answer_content,
+                        metadata={"title": f"💡 Output from {self.current_agent}"}
+                    ))
         
         return messages