Update multi_agent_book_workflow.py

multi_agent_book_workflow.py

@@ -233,68 +233,70 @@ class BookWritingOrchestrator:
         """
 
     def generate_book_concept(self, initial_prompt: str) -> Dict[str, Any]:
-        """
-        Generate a comprehensive book concept using multi-agent collaboration
-        """
-        try:
-            # Concept Development Task
-            concept_task = Task(
-                description=f"""
-                Develop a comprehensive book concept based on this initial idea:
-                {initial_prompt}
-                
-                Provide detailed outputs including:
-                1. Title: A unique and compelling book title
-                2. Genre: Primary genre and any relevant subgenres
-                3. Target Audience: Specific demographic and reader profile
-                4. Core Premise: The central concept or hook
-                5. Chapter Outline: Brief outline of proposed chapters
-                6. Narrative Approach: Point of view and stylistic elements
-                7. Key Themes: Major themes to be explored
-                
-                Format the output as a structured JSON object.
-                """,
-                agent=self.concept_agent
-            )
-            
-            # Research Validation Task
-            research_task = Task(
-                description="""
-                Review and enhance the generated book concept with:
-                1. Market analysis and genre conventions
-                2. Comparable successful titles
-                3. Unique selling points
-                4. Potential areas for deeper exploration
-                
-                Add these insights to the concept structure.
-                """,
-                agent=self.research_agent
-            )
-            
-            # Create Crew for Collaborative Processing
-            book_concept_crew = Crew(
-                agents=[self.concept_agent, self.research_agent],
-                tasks=[concept_task, research_task],
-                verbose=True
-            )
-            
-            # Execute Collaborative Workflow
-            result = book_concept_crew.kickoff()
-            
-            # Parse and structure the result
-            parsed_concept = self._parse_concept(result)
-            
-            # Store Context
-            self._store_context('book_concept', str(parsed_concept))
-            
-            return parsed_concept
-            
-        except Exception as e:
-            st.error(f"Book concept generation failed: {e}")
-            return self._fallback_book_concept(initial_prompt)
+    """
+    Generate a comprehensive book concept using multi-agent collaboration
+    """
+    try:
+        # Concept Development Task
+        concept_task = Task(
+            description=f"""
+            Develop a comprehensive book concept based on this initial idea:
+            {initial_prompt}
+            
+            Provide detailed outputs including:
+            1. Title: A unique and compelling book title
+            2. Genre: Primary genre and any relevant subgenres
+            3. Target Audience: Specific demographic and reader profile
+            4. Core Premise: The central concept or hook
+            5. Chapter Outline: Brief outline of proposed chapters
+            6. Narrative Approach: Point of view and stylistic elements
+            7. Key Themes: Major themes to be explored
+            
+            Format the output as a structured JSON object.
+            """,
+            expected_output="A detailed JSON-formatted book concept with title, genre, target audience, and other key elements",
+            agent=self.concept_agent
+        )
+        
+        # Research Validation Task
+        research_task = Task(
+            description="""
+            Review and enhance the generated book concept with:
+            1. Market analysis and genre conventions
+            2. Comparable successful titles
+            3. Unique selling points
+            4. Potential areas for deeper exploration
+            
+            Add these insights to the concept structure.
+            """,
+            expected_output="A detailed analysis and enhancement of the book concept with market insights and comparative analysis",
+            agent=self.research_agent
+        )
+        
+        # Create Crew for Collaborative Processing
+        book_concept_crew = Crew(
+            agents=[self.concept_agent, self.research_agent],
+            tasks=[concept_task, research_task],
+            verbose=True
+        )
+        
+        # Execute Collaborative Workflow
+        result = book_concept_crew.kickoff()
+        
+        # Parse and structure the result
+        parsed_concept = self._parse_concept(result)
+        
+        # Store Context
+        self._store_context('book_concept', str(parsed_concept))
+        
+        return parsed_concept
+        
+    except Exception as e:
+        st.error(f"Book concept generation failed: {e}")
+        return self._fallback_book_concept(initial_prompt)
 
     def generate_chapter_content(self, book_concept: Dict[str, Any], chapter_number: int) -> str:
-        """Generate content for a specific chapter"""
+    """Generate content for a specific chapter"""
         try:
             # Get previous context
             previous_context = self._retrieve_context(chapter_number - 1) if chapter_number > 1 else ""
@@ -316,6 +318,7 @@ class BookWritingOrchestrator:
                 
                 Generate a complete, polished chapter.
                 """,
+                expected_output="A complete, polished chapter with narrative continuity and character development",
                 agent=self.writing_agent
             )
             
@@ -330,6 +333,7 @@ class BookWritingOrchestrator:
                 
                 Provide a final, edited version.
                 """,
+                expected_output="A refined and polished version of the chapter with improved narrative quality",
                 agent=self.editing_agent
             )
             
@@ -353,9 +357,9 @@ class BookWritingOrchestrator:
             
             return chapter_content
             
-        except Exception as e:
-            st.error(f"Chapter generation failed: {e}")
-            return self._fallback_chapter_content(book_concept, chapter_number)
+    except Exception as e:
+        st.error(f"Chapter generation failed: {e}")
+        return self._fallback_chapter_content(book_concept, chapter_number)
 
     def _parse_concept(self, raw_concept: str) -> Dict[str, Any]:
         """Parse the raw concept output into a structured format"""