Update selfapi_writer.py

selfapi_writer.py

@@ -55,11 +55,8 @@ class SelfApiWriter:
             overview_response = self.client.messages.create(
                 model=self.model,
                 max_tokens=1000,
+                system="You are an expert at creating concise summaries of book blueprints.",
                 messages=[
-                    {
-                        "role": "system",
-                        "content": "You are an expert at creating concise summaries of book blueprints."
-                    },
                     {
                         "role": "user",
                         "content": f"""The following blueprint was truncated due to length constraints. 
@@ -89,6 +86,7 @@ class SelfApiWriter:
             # Truncate blueprint if too long
             truncated_blueprint, overview_summary = self._truncate_blueprint(blueprint)
 
+            # Prepare system prompt for blueprint processing
             system_prompt = """You are an expert book planner analyzing a blueprint.
             Extract ALL relevant information and return it in a structured format.
             Include:
@@ -154,6 +152,9 @@ class SelfApiWriter:
             self.writing_guidelines = extracted_info["guidelines"]
             self.initialized = True
             
+            # Store full blueprint in context
+            self.context['full_original_blueprint'] = blueprint
+            
             return extracted_info
             
         except Exception as e:
@@ -172,24 +173,24 @@ class SelfApiWriter:
             # Iterative generation for introduction
             full_intro_content = ""
             for iteration in range(1, self.max_iterations + 1):
+                # Prepare system prompt with full context
+                system_prompt = f"""You are writing the introduction for '{self.book_info.get('title', 'Untitled Book')}'
+                Full Blueprint Context:
+                {full_blueprint}
+                
+                Core Vision: {self.book_info.get('vision', '')}
+                Target Audience: {self.book_info.get('target_audience', '')}
+                
+                Writing Style: {self.writing_guidelines.get('style', 'Academic and clear')}
+                Tone: {self.writing_guidelines.get('tone', 'Professional')}
+                """
+
                 response = self.client.messages.create(
                     model=self.model,
                     max_tokens=2000,
                     temperature=0.7,
+                    system=system_prompt,
                     messages=[
-                        {
-                            "role": "system",
-                            "content": f"""You are writing the introduction for '{self.book_info.get('title', 'Untitled Book')}'
-                            Full Blueprint Context:
-                            {full_blueprint}
-                            
-                            Core Vision: {self.book_info.get('vision', '')}
-                            Target Audience: {self.book_info.get('target_audience', '')}
-                            
-                            Writing Style: {self.writing_guidelines.get('style', 'Academic and clear')}
-                            Tone: {self.writing_guidelines.get('tone', 'Professional')}
-                            """
-                        },
                         {
                             "role": "user",
                             "content": f"""Write the introduction: "{self.book_structure.get('introduction', 'Book Introduction')}"
@@ -239,30 +240,30 @@ class SelfApiWriter:
             # Iterative generation for chapter
             full_chapter_content = ""
             for iteration in range(1, self.max_iterations + 1):
+                # Prepare system prompt with full context
+                system_prompt = f"""You are writing '{self.book_info.get('title', 'Untitled Book')}'
+                Full Blueprint Context:
+                {full_blueprint}
+                
+                Chapter: {chapter_title}
+                Part: {part_title}
+                
+                Target Audience: {self.book_info.get('target_audience', '')}
+                Writing Style: {self.writing_guidelines.get('style', 'Academic and clear')}
+                Tone: {self.writing_guidelines.get('tone', 'Professional')}
+                
+                Chapter Structure Requirements:
+                {', '.join(self.writing_guidelines.get('chapter_structure', []))}
+                
+                Content Requirements:
+                {', '.join(self.writing_guidelines.get('content_requirements', []))}"""
+
                 response = self.client.messages.create(
                     model=self.model,
                     max_tokens=2000,
                     temperature=0.7,
+                    system=system_prompt,
                     messages=[
-                        {
-                            "role": "system",
-                            "content": f"""You are writing '{self.book_info.get('title', 'Untitled Book')}'
-                            Full Blueprint Context:
-                            {full_blueprint}
-                            
-                            Chapter: {chapter_title}
-                            Part: {part_title}
-                            
-                            Target Audience: {self.book_info.get('target_audience', '')}
-                            Writing Style: {self.writing_guidelines.get('style', 'Academic and clear')}
-                            Tone: {self.writing_guidelines.get('tone', 'Professional')}
-                            
-                            Chapter Structure Requirements:
-                            {', '.join(self.writing_guidelines.get('chapter_structure', []))}
-                            
-                            Content Requirements:
-                            {', '.join(self.writing_guidelines.get('content_requirements', []))}"""
-                        },
                         {
                             "role": "user",
                             "content": f"""Write Chapter: "{chapter_title}" in Part {part_idx + 1}: "{part_title}"