Update selfapi_writer.py

selfapi_writer.py

@@ -360,71 +360,83 @@ class SelfApiWriter:
         self.context['introduction'] = full_intro_content
         return full_intro_content
 
-    def write_chapter(self, part_idx: int, chapter_idx: int) -> str:
-        """Generate a chapter using enhanced content continuity"""
-        if not self.initialized:
-            raise ValueError("Writer not initialized. Process blueprint first.")
-            
-        content_id = f"part_{part_idx}_chapter_{chapter_idx}"
-        self._initialize_content_state(content_id)
+   def write_chapter(self, part_idx: int, chapter_idx: int, additional_prompt: str = "") -> str:
+    """Generate a chapter using enhanced content continuity and additional prompts"""
+    if not self.initialized:
+        raise ValueError("Writer not initialized. Process blueprint first.")
         
-        def generate_chapter_iteration(iteration: int,
-                                    previous_summary: str,
-                                    points_to_cover: List[str],
-                                    narrative_threads: List[str]) -> str:
-            """Generate a single chapter iteration with continuity context"""
-            part = self.book_structure["parts"][part_idx]
-            chapter_title = part["chapters"][chapter_idx]
-            part_title = part["title"]
-
-            system_prompt = f"""You are writing '{self.book_info.get('title', 'Untitled Book')}'
-            Chapter: {chapter_title}
-            Part: {part_title}
-            
-            Previous Content Summary: {previous_summary}
-            Points to Cover in This Section: {', '.join(points_to_cover)}
-            Active Narrative Threads: {', '.join(narrative_threads)}
-            
-            Writing Guidelines: {json.dumps(self.writing_guidelines, indent=2)}
-            
-            Create content that:
-            1. Builds naturally on previous sections
-            2. Develops the specified points
-            3. Maintains consistent narrative threads
-            4. Creates smooth transitions
-            5. Follows all style and structure guidelines"""
-
-            response = self.client.messages.create(
-                model=self.model,
-                max_tokens=2000,
-                temperature=0.7,
-                system=system_prompt,
-                messages=[{"role": "user", "content": f"Write the next section of Chapter: {chapter_title}"}]
-            )
-            
-            return response.content[0].text
-
-        full_chapter_content = self._generate_with_continuity(
-            generate_chapter_iteration,
-            content_id,
-            f"Chapter: {self.book_structure['parts'][part_idx]['chapters'][chapter_idx]}"
-        )
+    content_id = f"part_{part_idx}_chapter_{ch_idx}"
+    self._initialize_content_state(content_id)
+    
+    # Add any additional prompts to the content state
+    if additional_prompt:
+        self.content_states[content_id].add_custom_prompt(content_id, additional_prompt)
+    
+    def generate_chapter_iteration(iteration: int,
+                                previous_summary: str,
+                                points_to_cover: List[str],
+                                narrative_threads: List[str]) -> str:
+        """Generate a single chapter iteration with enhanced context"""
+        part = self.book_structure["parts"][part_idx]
+        chapter_title = part["chapters"][chapter_idx]
+        part_title = part["title"]
+        
+        # Get complete context including custom prompts
+        section_context = self.content_states[content_id].get_section_context(content_id)
         
-        if 'parts' not in self.context:
-            self.context['parts'] = []
+        # Enhanced system prompt with additional context
+        system_prompt = f"""You are writing '{self.book_info.get('title', 'Untitled Book')}'
+        Chapter: {chapter_title}
+        Part: {part_title}
         
-        while len(self.context['parts']) <= part_idx:
-            self.context['parts'].append({'chapters': []})
+        Blueprint Context: {self.context.get('full_original_blueprint', '')}
         
-        while len(self.context['parts'][part_idx]['chapters']) <= chapter_idx:
-            self.context['parts'][part_idx]['chapters'].append({})
+        Additional Instructions: {section_context['additional_prompt']}
+        Custom Guidelines: {section_context['custom_instructions']}
         
-        self.context['parts'][part_idx]['chapters'][chapter_idx] = {
-            'title': self.book_structure['parts'][part_idx]['chapters'][chapter_idx],
-            'content': full_chapter_content
-        }
+        Previous Content Summary: {previous_summary}
+        Points to Cover in This Section: {', '.join(points_to_cover)}
+        Active Narrative Threads: {', '.join(narrative_threads)}
         
-        return full_chapter_content
+        Writing Guidelines: {json.dumps(self.writing_guidelines, indent=2)}
+        
+        Create content that:
+        1. Builds naturally on previous sections
+        2. Incorporates the additional instructions and custom guidelines
+        3. Maintains consistent narrative threads
+        4. Creates smooth transitions
+        5. Follows all style and structure guidelines
+        
+        If additional instructions are provided, ensure they are seamlessly integrated
+        into the content while maintaining the overall style and structure."""
+
+        response = self.client.messages.create(
+            model=self.model,
+            max_tokens=2000,
+            temperature=0.7,
+            system=system_prompt,
+            messages=[{
+                "role": "user", 
+                "content": f"Write the next section of Chapter: {chapter_title}, incorporating any additional instructions provided."
+            }]
+        )
+        
+        return response.content[0].text
+
+    # Update generation process to include additional context
+    full_chapter_content = self._generate_with_continuity(
+        generate_chapter_iteration,
+        content_id,
+        f"Chapter: {self.book_structure['parts'][part_idx]['chapters'][chapter_idx]}"
+    )
+    
+    # Store context history
+    self.content_states[content_id].update_context_history(
+        content_id,
+        self.content_states[content_id].get_section_context(content_id)
+    )
+    
+    return full_chapter_content
 
     def _summarize_text(self, text: str) -> str:
         """Generate a concise summary of text"""