Update app.py

app.py

@@ -1,4 +1,4 @@
-# File: app.py
+# app.py
 
 import streamlit as st
 import json
@@ -15,11 +15,30 @@ class SelfApiApp:
             layout="wide"
         )
         
-        # Initialize data manager
-        self.data_manager = DataManager()
+        # Initialize data manager with explicit path
+        self.data_path = os.path.join(os.getcwd(), 'data')
+        self.data_manager = DataManager(self.data_path)
         
         # Initialize session state
         self._initialize_session_state()
+        
+        # Create necessary directories
+        self._ensure_directories()
+
+    def _ensure_directories(self):
+        """Create necessary directories"""
+        directories = [
+            self.data_path,
+            os.path.join(self.data_path, 'blueprints'),
+            os.path.join(self.data_path, 'book_content'),
+            os.path.join(self.data_path, 'exports'),
+            os.path.join(self.data_path, 'chapters'),
+            os.path.join(self.data_path, 'manual_content')
+        ]
+        
+        for dir_path in directories:
+            os.makedirs(dir_path, exist_ok=True)
+            st.write(f"Directory exists: {dir_path} - {os.path.exists(dir_path)}")
 
     def _initialize_session_state(self):
         """Initialize session state variables"""
@@ -38,6 +57,9 @@ class SelfApiApp:
             
         if 'structure_initialized' not in st.session_state:
             st.session_state.structure_initialized = False
+            
+        if 'manual_content' not in st.session_state:
+            st.session_state.manual_content = {}
 
     @property
     def default_blueprint(self):
@@ -101,45 +123,74 @@ Psychographic: Analytical minds seeking spiritual depth without the woo-woo
    - Links to both ancient wisdom and modern science"""
 
     def save_current_state(self):
-        """Save current state to data directory"""
+        """Save current state with verification"""
         try:
-            # Save blueprint if it exists
             if st.session_state.blueprint:
                 blueprint_file = self.data_manager.save_blueprint(st.session_state.blueprint)
-                st.success(f"Blueprint saved: {blueprint_file}")
+                # Verify file was saved
+                blueprint_path = os.path.join(self.data_path, 'blueprints', blueprint_file)
+                if os.path.exists(blueprint_path):
+                    st.success(f"Blueprint saved: {blueprint_file}")
+                    st.write(f"File size: {os.path.getsize(blueprint_path)} bytes")
+                else:
+                    st.error(f"Failed to save blueprint: File not found at {blueprint_path}")
             
-            # Save book content if it exists
             if st.session_state.book_content:
                 content_file = self.data_manager.save_book_content(st.session_state.book_content)
-                st.success(f"Book content saved: {content_file}")
-            
-            # Save individual chapters
-            for part_idx, part in enumerate(st.session_state.book_content.get('parts', [])):
-                for ch_idx, chapter in enumerate(part.get('chapters', [])):
-                    if chapter.get('content'):
-                        self.data_manager.save_chapter(part_idx, ch_idx, chapter)
+                # Verify file was saved
+                content_path = os.path.join(self.data_path, 'book_content', content_file)
+                if os.path.exists(content_path):
+                    st.success(f"Book content saved: {content_file}")
+                    st.write(f"File size: {os.path.getsize(content_path)} bytes")
+                else:
+                    st.error(f"Failed to save book content: File not found at {content_path}")
             
         except Exception as e:
             st.error(f"Error saving state: {e}")
+            st.write(f"Current working directory: {os.getcwd()}")
+            st.write(f"Data path: {self.data_path}")
 
     def load_latest_state(self):
-        """Load latest state from data directory"""
+        """Load latest state with verification"""
         try:
             # Load latest blueprint
             blueprint = self.data_manager.load_latest_blueprint()
             if blueprint:
                 st.session_state.blueprint = blueprint
+                st.success("Latest blueprint loaded")
             
             # Load latest book content
             content = self.data_manager.load_latest_book_content()
             if content:
                 st.session_state.book_content = content
-                
-            st.success("Latest state loaded successfully!")
+                st.success("Latest book content loaded")
             
+            # Load manual content
+            for part_idx in range(10):  # Assuming max 10 parts
+                for ch_idx in range(10):  # Assuming max 10 chapters per part
+                    manual_content = self.data_manager.load_manual_content(part_idx, ch_idx)
+                    if manual_content:
+                        content_key = f"manual_content_p{part_idx}_ch{ch_idx}"
+                        st.session_state.manual_content[content_key] = manual_content
+                        
         except Exception as e:
             st.error(f"Error loading state: {e}")
 
+    def copy_to_clipboard(self, text: str):
+        """Copy text to clipboard using JavaScript"""
+        js_code = f"""
+        <script>
+        function copyToClipboard() {{
+            const text = {json.dumps(text)};
+            navigator.clipboard.writeText(text)
+                .then(() => alert('Copied to clipboard!'))
+                .catch(err => alert('Failed to copy: ' + err));
+        }}
+        </script>
+        <button onclick="copyToClipboard()">Copy to Clipboard</button>
+        """
+        st.components.v1.html(js_code, height=50)
+
     def render_blueprint_editor(self):
         """Render the blueprint editor interface"""
         st.header("📝 Blueprint Editor")
@@ -250,11 +301,25 @@ Psychographic: Analytical minds seeking spiritual depth without the woo-woo
             intro_col1, intro_col2 = st.columns([3, 1])
             with intro_col1:
                 st.markdown("### Introduction")
+                # Manual content for introduction
+                manual_intro = st.text_area(
+                    "Pre-written content for introduction",
+                    value=st.session_state.manual_content.get('introduction', ''),
+                    height=200,
+                    key="manual_intro"
+                )
+                if manual_intro != st.session_state.manual_content.get('introduction', ''):
+                    st.session_state.manual_content['introduction'] = manual_intro
+                    self.data_manager.save_manual_content(-1, -1, manual_intro)
+                
                 if st.session_state.book_content.get('introduction'):
                     st.success("✓ Generated")
+                    
             with intro_col2:
                 if st.button("Generate Introduction"):
                     with st.spinner("Generating introduction..."):
+                        if manual_intro:
+                            st.session_state.writer.set_manual_content('introduction', manual_intro)
                         intro_content = st.session_state.writer.write_introduction()
                         st.session_state.book_content["introduction"] = intro_content
                         self.save_current_state()
@@ -269,7 +334,21 @@ Psychographic: Analytical minds seeking spiritual depth without the woo-woo
                     for ch_idx, ch_title in enumerate(part["chapters"]):
                         ch_col1, ch_col2 = st.columns([3, 1])
                         with ch_col1:
-                            st.markdown(f"- {ch_title}")
+                            st.markdown(f"#### {ch_title}")
+                            
+                            # Manual content input for chapter
+                            content_key = f"manual_content_p{part_idx}_ch{ch_idx}"
+                            manual_content = st.text_area(
+                                "Pre-written content",
+                                value=st.session_state.manual_content.get(content_key, ''),
+                                height=200,
+                                key=content_key
+                            )
+                            
+                            if manual_content != st.session_state.manual_content.get(content_key, ''):
+                                st.session_state.manual_content[content_key] = manual_content
+                                self.data_manager.save_manual_content(part_idx, ch_idx, manual_content)
+                            
                             chapter_exists = (
                                 len(st.session_state.book_content.get('parts', [])) > part_idx and
                                 len(st.session_state.book_content['parts'][part_idx].get('chapters', [])) > ch_idx and
@@ -277,9 +356,16 @@ Psychographic: Analytical minds seeking spiritual depth without the woo-woo
                             )
                             if chapter_exists:
                                 st.success("✓ Generated")
+                                
                         with ch_col2:
                             if st.button(f"Generate", key=f"gen_{part_idx}_{ch_idx}"):
                                 with st.spinner(f"Generating {ch_title}..."):
+                                    if manual_content:
+                                        st.session_state.writer.set_manual_content(
+                                            f"part_{part_idx}_chapter_{ch_idx}",
+                                            manual_content
+                                        )
+                                    
                                     chapter_content = st.session_state.writer.write_chapter(part_idx, ch_idx)
                                     
                                     # Ensure part exists
@@ -310,6 +396,9 @@ Psychographic: Analytical minds seeking spiritual depth without the woo-woo
             if st.button("Generate All"):
                 with st.spinner("Generating entire book..."):
                     # Generate introduction
+                    manual_intro = st.session_state.manual_content.get('introduction', '')
+                    if manual_intro:
+                        st.session_state.writer.set_manual_content('introduction', manual_intro)
                     intro_content = st.session_state.writer.write_introduction()
                     st.session_state.book_content["introduction"] = intro_content
                     
@@ -326,6 +415,15 @@ Psychographic: Analytical minds seeking spiritual depth without the woo-woo
                             
                             for ch_idx, ch_title in enumerate(part["chapters"]):
                                 with st.spinner(f"Generating {ch_title}..."):
+                                    content_key = f"manual_content_p{part_idx}_ch{ch_idx}"
+                                    manual_content = st.session_state.manual_content.get(content_key, '')
+                                    
+                                    if manual_content:
+                                        st.session_state.writer.set_manual_content(
+                                            f"part_{part_idx}_chapter_{ch_idx}",
+                                            manual_content
+                                        )
+                                    
                                     chapter_content = st.session_state.writer.write_chapter(part_idx, ch_idx)
                                     
                                     # Ensure chapter exists
@@ -362,10 +460,29 @@ Psychographic: Analytical minds seeking spiritual depth without the woo-woo
             st.warning("Please process the blueprint first in the Blueprint Editor tab.")
             return
         
-        if not st.session_state.book_content.get('introduction') and not st.session_state.book_content.get('parts'):
-            st.info("No content generated yet. Use the Generator tab to create content.")
-            return
-        
+        # Show current file status
+        st.subheader("File Status")
+        col1, col2 = st.columns(2)
+        with col1:
+            st.write("Blueprint Files:")
+            blueprint_dir = os.path.join(self.data_path, 'blueprints')
+            if os.path.exists(blueprint_dir):
+                files = os.listdir(blueprint_dir)
+                for f in files:
+                    st.write(f"- {f}")
+            else:
+                st.write("No blueprint directory found")
+                
+        with col2:
+            st.write("Content Files:")
+            content_dir = os.path.join(self.data_path, 'book_content')
+            if os.path.exists(content_dir):
+                files = os.listdir(content_dir)
+                for f in files:
+                    st.write(f"- {f}")
+            else:
+                st.write("No content directory found")
+
         # Add export button
         if st.button("Export Book"):
             try:
@@ -388,11 +505,15 @@ Psychographic: Analytical minds seeking spiritual depth without the woo-woo
             st.subheader("Book Information")
             st.json(st.session_state.book_content['book_info'])
         
-       # Show introduction
+        # Show introduction
         if st.session_state.book_content.get('introduction'):
             st.subheader("Introduction")
             with st.expander("View Introduction", expanded=True):
                 st.markdown(st.session_state.book_content['introduction'])
+                
+                # Add copy button for introduction
+                if st.button("Copy Introduction"):
+                    self.copy_to_clipboard(st.session_state.book_content['introduction'])
         
         # Show generated parts and chapters
         for part_idx, part in enumerate(st.session_state.book_content.get('parts', [])):
@@ -405,8 +526,7 @@ Psychographic: Analytical minds seeking spiritual depth without the woo-woo
                         
                         # Add copy button for chapter
                         if st.button(f"Copy Chapter", key=f"copy_{part_idx}_{ch_idx}"):
-                            st.write("Chapter content copied to clipboard!")
-                            st.toast("Chapter copied!")
+                            self.copy_to_clipboard(chapter['content'])
 
     def run(self):
         """Run the Streamlit application"""
@@ -432,4 +552,4 @@ def main():
     app.run()
 
 if __name__ == "__main__":
-    main()
+    main()