Update selfapi_writer.py

selfapi_writer.py

@@ -1,24 +1,16 @@
+# selfapi_writer.py
+
 from anthropic import Anthropic
 import streamlit as st
 import json
 import os
 import tiktoken
 from typing import Dict, Any, Optional, List, Tuple, Generator
-
-class ContentState:
-    """Tracks the state and progression of content generation"""
-    def __init__(self):
-        self.current_summary = ""
-        self.section_outlines = []
-        self.generated_sections = []
-        self.narrative_threads = []
-        self.key_points_covered = set()
-        self.transition_points = []
+from content_state import ContentState
 
 class SelfApiWriter:
     def __init__(self):
         """Initialize the Self.api writer with enhanced content tracking"""
-        # Try to get API key from environment variables first, then from secrets
         ANTHROPIC_API_KEY = os.getenv('api_key')
                     
         if not ANTHROPIC_API_KEY:
@@ -34,7 +26,7 @@ class SelfApiWriter:
         # Configuration for generation
         self.pages_per_chapter = 25
         self.words_per_page = 250
-        self.max_iterations = 10  # Reduced from 20 to 10
+        self.max_iterations = 10
         self.max_tokens = 15000
 
         # Token encoding
@@ -113,7 +105,7 @@ class SelfApiWriter:
         return outline
 
     def process_blueprint(self, blueprint: str) -> Dict[str, Any]:
-        """Process blueprint to extract complete writing guidelines and structure"""
+        """Process blueprint to extract structure and guidelines"""
         try:
             with st.spinner("Processing blueprint..."):
                 truncated_blueprint, overview_summary = self._truncate_blueprint(blueprint)
@@ -161,7 +153,7 @@ class SelfApiWriter:
                     temperature=0,
                     system=system_prompt,
                     messages=[{
-                        "role": "user",
+                        "role": "user", 
                         "content": f"""Analyze this book blueprint and extract ALL information:
 
                         {truncated_blueprint}
@@ -187,6 +179,12 @@ class SelfApiWriter:
             st.error(f"Error processing blueprint: {str(e)}")
             return None
 
+    def set_manual_content(self, content_id: str, content: str) -> None:
+        """Set pre-written content for a section"""
+        if content_id not in self.content_states:
+            self._initialize_content_state(content_id)
+        self.content_states[content_id].set_manual_content(content)
+
     def _generate_transition(self, content_id: str, prev_content: str, next_content: str) -> str:
         """Generate smooth transition between content sections"""
         state = self.content_states[content_id]
@@ -240,13 +238,22 @@ class SelfApiWriter:
                                 generate_func: callable,
                                 content_id: str,
                                 title: str,
-                                total_steps: int = 10) -> str:  # Default steps reduced to 10
+                                total_steps: int = 10) -> str:
         """Enhanced generation with content continuity tracking"""
         progress_bar = st.progress(0, text=f"Generating {title}...")
         full_content = ""
         state = self.content_states[content_id]
         
         try:
+            # If manual content exists, use it as a starting point
+            if state.manual_content:
+                full_content = state.manual_content + "\n\n"
+                # Generate initial summary from manual content
+                state.current_summary = self._generate_progressive_summary(
+                    content_id,
+                    full_content
+                )
+            
             # Generate initial outline
             outline = self._generate_section_outline(content_id, "section", title)
             points_per_iteration = max(1, len(outline) // total_steps)