Update selfapi_writer.py

selfapi_writer.py

@@ -64,6 +64,45 @@ class SelfApiWriter:
             ]
         }
 
+    def process_blueprint(self, blueprint: str) -> Dict[str, Any]:
+        """Process updated blueprint to generate new book 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.
+            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:
+
+            {blueprint}
+
+            Return only the JSON structure with no additional text."""
+
+            response = self.client.messages.create(
+                model="claude-3-sonnet-20240229",
+                max_tokens=2000,
+                temperature=0,
+                system=system_prompt,
+                messages=[{"role": "user", "content": prompt}]
+            )
+            
+            new_structure = json.loads(response.content[0].text)
+            self.book_structure = new_structure
+            return new_structure
+            
+        except Exception as e:
+            st.error(f"Error processing blueprint: {str(e)}")
+            return self.book_structure  # Return existing structure on error
+
     def write_introduction(self) -> str:
         """Generate the book's introduction"""
         try: