Update selfapi_writer.py

selfapi_writer.py

@@ -1,5 +1,3 @@
-# selfapi_writer.py
-
 from anthropic import Anthropic
 import streamlit as st
 import json
@@ -12,7 +10,7 @@ class SelfApiWriter:
     def __init__(self):
         """Initialize the Self.api writer with enhanced content tracking"""
         ANTHROPIC_API_KEY = os.getenv('api_key')
-                    
+            
         if not ANTHROPIC_API_KEY:
             raise ValueError("Anthropic API key not found. Please ensure ANTHROPIC_API_KEY is set.")
         
@@ -235,10 +233,10 @@ class SelfApiWriter:
         return response.content[0].text
 
     def _generate_with_continuity(self, 
-                                generate_func: callable,
-                                content_id: str,
-                                title: str,
-                                total_steps: int = 10) -> str:
+                                    generate_func: callable,
+                                    content_id: str,
+                                    title: str,
+                                    total_steps: int = 10) -> str:
         """Enhanced generation with content continuity tracking"""
         progress_bar = st.progress(0, text=f"Generating {title}...")
         full_content = ""
@@ -319,9 +317,9 @@ class SelfApiWriter:
         self._initialize_content_state(content_id)
             
         def generate_intro_iteration(iteration: int,
-                                   previous_summary: str,
-                                   points_to_cover: List[str],
-                                   narrative_threads: List[str]) -> str:
+                                        previous_summary: str,
+                                        points_to_cover: List[str],
+                                        narrative_threads: List[str]) -> str:
             """Generate a single iteration of the introduction"""
             full_blueprint = self.context.get('full_original_blueprint', '')
 
@@ -360,129 +358,82 @@ class SelfApiWriter:
         self.context['introduction'] = full_intro_content
         return full_intro_content
 
-   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.")
-        
-    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)
-        
-        # 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}
-        
-        Blueprint Context: {self.context.get('full_original_blueprint', '')}
-        
-        Additional Instructions: {section_context['additional_prompt']}
-        Custom Guidelines: {section_context['custom_instructions']}
-        
-        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. 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
+    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.")
+            
+        content_id = f"part_{part_idx}_chapter_{chapter_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)
+            
+            # 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}
+            
+            Blueprint Context: {self.context.get('full_original_blueprint', '')}
+            
+            Additional Instructions: {section_context['additional_prompt']}
+            Custom Guidelines: {section_context['custom_instructions']}
+            
+            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. 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."""
 
-    # 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
+            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
 
-    def _summarize_text(self, text: str) -> str:
-        """Generate a concise summary of text"""
-        response = self.client.messages.create(
-            model=self.model,
-            max_tokens=300,
-            temperature=0.3,
-            messages=[{
-                "role": "user",
-                "content": f"Summarize the key points from this text:\n\n{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]}"
         )
-        return response.content[0].text
-
-    def _generate_conclusion(self, content_id: str, full_content: str) -> str:
-        """Generate a conclusion that ties everything together"""
-        state = self.content_states[content_id]
-        
-        conclusion_prompt = f"""Create a conclusion that ties together:
-
-        Content Summary: {state.current_summary}
-        Key Points Covered: {', '.join(state.key_points_covered)}
-        Narrative Threads: {', '.join(state.narrative_threads)}
-        
-        The conclusion should:
-        1. Summarize main arguments
-        2. Connect key themes
-        3. Reinforce core messages
-        4. Provide closure while maintaining interest"""
-
-        response = self.client.messages.create(
-            model=self.model,
-            max_tokens=500,
-            temperature=0.7,
-            messages=[{"role": "user", "content": conclusion_prompt}]
+       
+        # Store context history
+        self.content_states[content_id].update_context_history(
+            content_id,
+            self.content_states[content_id].get_section_context(content_id)
         )
-        
-        return response.content[0].text
+       
+        return full_chapter_content
 
-    def get_current_structure(self) -> Optional[Dict[str, Any]]:
-        """Get current book structure and guidelines"""
-        if not self.initialized:
-            return None
-            
-        return {
-            "book_info": self.book_info,
-            "structure": self.book_structure,
-            "guidelines": self.writing_guidelines
-        }
+    def