Update selfapi_writer.py

selfapi_writer.py

@@ -4,7 +4,7 @@ import anthropic
 import streamlit as st
 import json
 import os
-from typing import Dict, Any
+from typing import Dict, Any, Optional
 
 class SelfApiWriter:
     def __init__(self):
@@ -17,75 +17,58 @@ class SelfApiWriter:
             
         if not api_key:
             raise ValueError("ANTHROPIC_API_KEY not found in environment variables or Streamlit secrets.")
-            
-        # Print debug info (remove in production)
-        print("Environment variables available:", [k for k in os.environ.keys()])
-        print("API key found:", bool(api_key))
         
-        self.client = anthropic.Anthropic(
-            api_key=api_key
-        )
+        self.client = anthropic.Anthropic(api_key=api_key)
         self.context = {}
-        self.book_structure = {
-            "introduction": "System Requirements: A Human's Guide to Being Human",
-            "parts": [
-                {
-                    "title": "The Human Input/Output System",
-                    "chapters": [
-                        "Rate Limiting Your Reality",
-                        "Debugging Your Attention Span",
-                        "The Mindfulness Microservice"
-                    ]
-                },
-                {
-                    "title": "Your Internal Neural Network",
-                    "chapters": [
-                        "Training Your Gut Algorithm",
-                        "The Wisdom Cache",
-                        "Pattern Recognition Beyond Logic"
-                    ]
-                },
-                {
-                    "title": "The Source Code of Being",
-                    "chapters": [
-                        "Quantum Mechanics of Consciousness",
-                        "The Purpose Protocol",
-                        "Refactoring Your Reality"
-                    ]
-                },
-                {
-                    "title": "The Universal Runtime Environment",
-                    "chapters": [
-                        "Distributed Consciousness Systems",
-                        "The Empathy Protocol",
-                        "Scaling Your Impact"
-                    ]
-                }
-            ]
-        }
+        self.book_structure = None
+        self.writing_guidelines = None
+        self.initialized = False
 
     def process_blueprint(self, blueprint: str) -> Dict[str, Any]:
-        """Process updated blueprint to generate new book structure"""
+        """Process blueprint to extract complete writing guidelines and structure"""
         try:
-            system_prompt = """You are an expert book planner analyzing a blueprint for 'Self.api: A Modern Spiritual Operating System'.
-            Extract and organize the book structure from the provided blueprint.
+            system_prompt = """You are an expert book planner analyzing a blueprint.
+            Extract ALL relevant information and return it in a structured format.
+            Include:
+            1. Book title and high-level information
+            2. Complete structure (introduction, parts, chapters)
+            3. All writing style guidelines
+            4. Content requirements and constraints
+            5. Target audience details
+            6. Chapter structure requirements
+            7. Tone and voice requirements
+            8. Any other relevant guidelines or requirements
+            
             Return a JSON structure with the following format:
             {
-                "introduction": "Introduction title",
-                "parts": [
-                    {
-                        "title": "Part title",
-                        "chapters": ["Chapter 1 title", "Chapter 2 title", ...]
-                    }
-                ]
-            }
-            Ensure all sections from the blueprint are included and properly organized."""
-
-            prompt = f"""Analyze this book blueprint and extract the structure:
+                "book_info": {
+                    "title": "Book title",
+                    "vision": "Core vision/purpose",
+                    "target_audience": "Detailed audience description"
+                },
+                "structure": {
+                    "introduction": "Introduction title",
+                    "parts": [
+                        {
+                            "title": "Part title",
+                            "chapters": ["Chapter 1 title", "Chapter 2 title", ...]
+                        }
+                    ]
+                },
+                "guidelines": {
+                    "style": "Writing style description",
+                    "tone": "Tone requirements",
+                    "chapter_structure": ["Required chapter components"],
+                    "content_requirements": ["Specific content requirements"],
+                    "practical_elements": ["Required practical elements"]
+                }
+            }"""
+
+            prompt = f"""Analyze this book blueprint and extract ALL information:
 
             {blueprint}
 
-            Return only the JSON structure with no additional text."""
+            Return only the JSON structure without any additional text."""
 
             response = self.client.messages.create(
                 model="claude-3-sonnet-20240229",
@@ -95,31 +78,45 @@ class SelfApiWriter:
                 messages=[{"role": "user", "content": prompt}]
             )
             
-            new_structure = json.loads(response.content[0].text)
-            self.book_structure = new_structure
-            return new_structure
+            extracted_info = json.loads(response.content[0].text)
+            
+            # Store extracted information
+            self.book_info = extracted_info["book_info"]
+            self.book_structure = extracted_info["structure"]
+            self.writing_guidelines = extracted_info["guidelines"]
+            self.initialized = True
+            
+            return extracted_info
             
         except Exception as e:
             st.error(f"Error processing blueprint: {str(e)}")
-            return self.book_structure  # Return existing structure on error
+            return None
 
     def write_introduction(self) -> str:
-        """Generate the book's introduction"""
+        """Generate the book's introduction based on extracted guidelines"""
+        if not self.initialized:
+            raise ValueError("Writer not initialized. Process blueprint first.")
+            
         try:
-            system_prompt = """You are writing the introduction to 'Self.api: A Modern Spiritual Operating System'.
-            The introduction should hook tech professionals by speaking their language while introducing the book's core premise.
-            Include the author's journey from debugging code to debugging consciousness."""
+            system_prompt = f"""You are writing the introduction for '{self.book_info["title"]}'
+            Core Vision: {self.book_info["vision"]}
+            Target Audience: {self.book_info["target_audience"]}
+            
+            Writing Style: {self.writing_guidelines["style"]}
+            Tone: {self.writing_guidelines["tone"]}
+            
+            Content Requirements:
+            {', '.join(self.writing_guidelines["content_requirements"])}"""
 
-            intro_prompt = """Write the introduction "System Requirements: A Human's Guide to Being Human"
+            intro_prompt = f"""Write the introduction: "{self.book_structure['introduction']}"
             
-            Key elements to include:
-            1. Opening hook: "In a world where we can Google anything except our own purpose..."
-            2. Author's personal journey
-            3. The core premise of viewing spirituality through an API/systems lens
-            4. Overview of the book's structure and approach
-            5. What readers will gain from this "operating manual for the human spirit"
+            Create an engaging opening that:
+            1. Introduces the book's core concept
+            2. Speaks directly to the target audience
+            3. Outlines the book's approach and structure
+            4. Sets the tone for the entire book
             
-            Make it compelling, technical yet accessible, and set the tone for the entire book."""
+            Follow ALL provided guidelines for style, tone, and content."""
 
             response = self.client.messages.create(
                 model="claude-3-sonnet-20240229",
@@ -138,35 +135,40 @@ class SelfApiWriter:
             return f"Error generating introduction: {str(e)}"
 
     def write_chapter(self, part_idx: int, chapter_idx: int) -> str:
-        """Generate a specific chapter"""
+        """Generate a chapter using extracted guidelines"""
+        if not self.initialized:
+            raise ValueError("Writer not initialized. Process blueprint first.")
+            
         try:
             part = self.book_structure["parts"][part_idx]
             chapter_title = part["chapters"][chapter_idx]
             part_title = part["title"]
             previous_chapter = self.context.get(f'part_{part_idx}_chapter_{chapter_idx-1}', '')
 
-            system_prompt = """You are writing 'Self.api: A Modern Spiritual Operating System', a book that bridges technology and spirituality for tech professionals.
-            Follow these guidelines:
-            - Use API and technical metaphors naturally and accurately
-            - Maintain a voice that's witty but wise (Douglas Adams meets Deepak Chopra)
-            - Structure each chapter with: System Log (personal story), Documentation (teaching), Implementation Guide (practical steps)
-            - Balance is 60% practical, 40% philosophical
-            - Include 3 "aha" moments and 2-3 quotable passages
-            - Ensure every concept links to both ancient wisdom and modern science
-            - Write for analytical minds seeking spiritual depth without woo-woo
-            - Include practical exercises every few pages"""
+            system_prompt = f"""You are writing '{self.book_info["title"]}'
+            Target Audience: {self.book_info["target_audience"]}
+            Writing Style: {self.writing_guidelines["style"]}
+            Tone: {self.writing_guidelines["tone"]}
+            
+            Chapter Structure Requirements:
+            {', '.join(self.writing_guidelines["chapter_structure"])}
+            
+            Content Requirements:
+            {', '.join(self.writing_guidelines["content_requirements"])}
+            
+            Practical Elements to Include:
+            {', '.join(self.writing_guidelines["practical_elements"])}"""
 
             chapter_prompt = f"""
             Write Chapter: "{chapter_title}" in Part {part_idx + 1}: "{part_title}"
             
             Previous Chapter Context: {previous_chapter[:1000] if previous_chapter else 'Starting new part'}
 
-            Requirements:
-            1. Open with a compelling "System Log" that relates to {chapter_title}
-            2. Use technically accurate API/programming metaphors
-            3. Include at least two practical exercises
-            4. End with clear implementation steps
-            5. Maintain the tech-spiritual bridge throughout
+            Follow ALL provided guidelines for:
+            1. Structure and organization
+            2. Style and tone
+            3. Practical elements and exercises
+            4. Content depth and requirements
             
             Begin writing the complete chapter now."""
 
@@ -184,4 +186,15 @@ class SelfApiWriter:
             
         except Exception as e:
             st.error(f"Error generating chapter: {str(e)}")
-            return f"Error generating chapter {chapter_title}: {str(e)}"
+            return f"Error generating chapter {chapter_title}: {str(e)}"
+
+    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
+        }