Update multi_agent_book_workflow.py

multi_agent_book_workflow.py

@@ -84,21 +84,22 @@ class BookWritingOrchestrator:
             raise
 
     def _setup_ai_models(self):
-        """
-        Configure AI models with robust error handling
-        """
+    """
+    Configure AI models with robust error handling
+    """
         try:
             # LLM Configurations
             self.openai_llm = ChatOpenAI(
-                model="gpt-4-turbo",
+                model="gpt-4-turbo-preview",  # Updated model name
                 temperature=0.7,
                 api_key=self.openai_api_key
             )
             
+            # Updated Anthropic model configuration
             self.anthropic_llm = ChatAnthropic(
-                model='claude-3-opus-20240229',
+                model_name="claude-3-sonnet-20240229",  # Updated model name
                 temperature=0.7,
-                api_key=self.anthropic_api_key
+                anthropic_api_key=self.anthropic_api_key
             )
             
             # Global memory for cross-chapter context
@@ -110,6 +111,46 @@ class BookWritingOrchestrator:
             st.error(f"AI model setup failed: {e}")
             raise
 
+    def _fallback_book_concept(self, initial_prompt: str) -> Dict[str, Any]:
+        """
+        Provide a fallback response when book concept generation fails
+        
+        Args:
+            initial_prompt (str): Original user prompt
+            
+        Returns:
+            Dict: Basic book concept structure
+        """
+        return {
+            "title": "Untitled Project",
+            "genre": "General Fiction",
+            "target_audience": "General Adult",
+            "core_premise": initial_prompt,
+            "chapter_outline": ["Chapter 1: Introduction"],
+            "narrative_approach": "Standard Third-Person Narrative",
+            "status": "fallback_generated"
+        }
+
+    def _fallback_chapter_content(self, book_concept: Dict[str, Any], chapter_number: int) -> str:
+        """
+        Provide fallback chapter content when generation fails
+        
+        Args:
+            book_concept (Dict): Book concept data
+            chapter_number (int): Chapter number
+            
+        Returns:
+            str: Basic chapter content
+        """
+        return f"""
+        Chapter {chapter_number}
+        
+        [Placeholder content for {book_concept.get('title', 'Untitled')}]
+        
+        This is auto-generated fallback content due to an error in chapter generation.
+        Please try regenerating this chapter or contact support if the issue persists.
+        """
+
     def setup_agents(self):
         """
         Create specialized agents for book writing workflow