fix: agent flow

services/model_handler.py

@@ -232,12 +232,14 @@ Output:"""
     
     def generate_answer(self, query: str) -> str:
         try:
+            logging.info(f"Generating answer for query: {query}")
             # Format translation prompt
             translation_prompt = self._format_prompt(
                 role="Translate the following text to English",
                 instructions="Provide a direct English translation of the input text.",
                 query=query
             )
+            logging.info(f"Translation prompt: {translation_prompt}")
             
             # Get English translation
             translation = self.translator.run(prompt=translation_prompt, stream=False)
@@ -245,12 +247,15 @@ Output:"""
                 logging.error("Translation failed")
                 return "Error: Unable to translate the query"
 
+            logging.info(f"Translation: {translation}")
+            
             # Format research prompt
             research_prompt = self._format_prompt(
                 role="Research Assistant",
                 instructions="Provide a clear and concise answer based on scientific sources.",
                 query=translation
             )
+            logging.info(f"Research prompt: {research_prompt}")
             
             # Get research results
             research_results = self.researcher.run(prompt=research_prompt, stream=False)
@@ -258,7 +263,41 @@ Output:"""
                 logging.error("Research failed")
                 return "Error: Unable to perform research"
 
-            return research_results
+            logging.info(f"Research results: {research_results}")
+            
+            # Format summary prompt
+            summary_prompt = self._format_prompt(
+                role="Summary Assistant",
+                instructions="Provide a clear and concise summary of the research results.",
+                query=research_results
+            )
+            logging.info(f"Summary prompt: {summary_prompt}")
+            
+            # Get summary
+            summary = self.summarizer.run(prompt=summary_prompt, stream=False)
+            if not summary:
+                logging.error("Summary failed")
+                return "Error: Unable to generate summary"
+
+            logging.info(f"Summary: {summary}")
+            
+            # Format presentation prompt
+            presentation_prompt = self._format_prompt(
+                role="Presentation Assistant",
+                instructions="Provide a clear and concise presentation of the research results.",
+                query=summary
+            )
+            logging.info(f"Presentation prompt: {presentation_prompt}")
+            
+            # Get presentation
+            presentation = self.presenter.run(prompt=presentation_prompt, stream=False)
+            if not presentation:
+                logging.error("Presentation failed")
+                return "Error: Unable to generate presentation"
+
+            logging.info(f"Presentation: {presentation}")
+
+            return presentation
         except Exception as e:
             logging.error(f"Error generating answer: {str(e)}")
             return f"Error: {str(e)}"