Update app.py

app.py

@@ -172,7 +172,191 @@ class BookWritingApp:
         if st.sidebar.button("Save Current Project"):
             self._save_current_project()
     
-    # ... (rest of the methods remain the same as in the previous implementation)
+    def render_book_generation_interface(self):
+        """
+        Render the main book generation interface
+        """
+        # Tabs for different stages of book writing
+        tab1, tab2, tab3 = st.tabs([
+            "Book Concept", 
+            "Chapter Generation", 
+            "Project Progress"
+        ])
+        
+        with tab1:
+            self._render_concept_development()
+        
+        with tab2:
+            self._render_chapter_generation()
+        
+        with tab3:
+            self._render_project_progress()
+    
+    def _render_concept_development(self):
+        """
+        Render the book concept development interface
+        """
+        st.header("📘 Book Concept Development")
+        
+        # Initial concept input
+        initial_concept = st.text_area(
+            "Describe Your Book Idea", 
+            height=200, 
+            placeholder="Enter a comprehensive description of your book concept..."
+        )
+        
+        if st.button("Generate Book Concept"):
+            # Ensure project orchestrator is available
+            if not st.session_state.project_orchestrator:
+                self._setup_project_orchestrator()
+            
+            with st.spinner("Developing Book Concept..."):
+                try:
+                    # Generate book concept using multi-agent approach
+                    book_concept = st.session_state.project_orchestrator.generate_book_concept(
+                        initial_concept
+                    )
+                    
+                    # Update session state
+                    st.session_state.book_concept = book_concept
+                    st.session_state.project_metadata.update({
+                        'title': book_concept.get('title', 'Untitled Project'),
+                        'genre': book_concept.get('genre', 'Unspecified')
+                    })
+                    
+                    # Display generated concept
+                    st.subheader("Generated Book Concept")
+                    st.json(book_concept)
+                
+                except Exception as e:
+                    st.error(f"Error generating book concept: {e}")
+    
+    def _render_chapter_generation(self):
+        """
+        Render the chapter generation interface
+        """
+        st.header("✍️ Chapter Generation")
+        
+        # Check if book concept exists
+        if not st.session_state.book_concept:
+            st.warning("Please generate a book concept first.")
+            return
+        
+        # Chapter generation controls
+        col1, col2 = st.columns(2)
+        
+        with col1:
+            # Chapter number selection
+            max_chapters = 20  # Can be adjusted
+            chapter_number = st.number_input(
+                "Select Chapter to Generate", 
+                min_value=1, 
+                max_value=max_chapters, 
+                value=len(st.session_state.generated_chapters) + 1
+            )
+        
+        with col2:
+            # Total chapters planning
+            total_chapters = st.number_input(
+                "Total Planned Chapters", 
+                min_value=1, 
+                max_value=max_chapters, 
+                value=st.session_state.project_metadata.get('total_chapters', 10)
+            )
+            st.session_state.project_metadata['total_chapters'] = total_chapters
+        
+        # Generate Chapter Button
+        if st.button("Generate Chapter"):
+            with st.spinner(f"Generating Chapter {chapter_number}..."):
+                try:
+                    # Generate chapter content
+                    chapter_content = st.session_state.project_orchestrator.generate_chapter_content(
+                        st.session_state.book_concept, 
+                        chapter_number
+                    )
+                    
+                    # Store generated chapter
+                    st.session_state.generated_chapters[chapter_number] = {
+                        'content': chapter_content,
+                        'status': 'Generated'
+                    }
+                    
+                    # Display Chapter Content
+                    st.subheader(f"Chapter {chapter_number}")
+                    st.write(chapter_content)
+                    
+                    # Optional: Edit Chapter
+                    edited_content = st.text_area(
+                        "Edit Chapter Content", 
+                        value=chapter_content, 
+                        height=400
+                    )
+                    
+                    # Save edited content
+                    if st.button(f"Save Chapter {chapter_number}"):
+                        st.session_state.generated_chapters[chapter_number]['content'] = edited_content
+                        st.success(f"Chapter {chapter_number} saved!")
+                
+                except Exception as e:
+                    st.error(f"Error generating chapter: {e}")
+    
+    def _render_project_progress(self):
+        """
+        Render project progress and tracking interface
+        """
+        st.header("📊 Project Progress")
+        
+        # Project Metadata Overview
+        st.subheader("Project Overview")
+        col1, col2 = st.columns(2)
+        
+        with col1:
+            st.metric("Book Title", st.session_state.project_metadata.get('title', 'Untitled'))
+            st.metric("Genre", st.session_state.project_metadata.get('genre', 'Unspecified'))
+        
+        with col2:
+            total_chapters = st.session_state.project_metadata.get('total_chapters', 0)
+            generated_chapters = len(st.session_state.generated_chapters)
+            
+            st.metric("Total Planned Chapters", total_chapters)
+            st.metric("Chapters Generated", generated_chapters)
+            
+            # Progress Bar
+            progress = generated_chapters / total_chapters if total_chapters > 0 else 0
+            st.progress(progress)
+        
+        # Chapter Navigation and Details
+        st.subheader("Generated Chapters")
+        
+        # Create tabs for each generated chapter
+        if st.session_state.generated_chapters:
+            chapter_tabs = st.tabs([
+                f"Chapter {ch_num}" for ch_num in sorted(st.session_state.generated_chapters.keys())
+            ])
+            
+            for i, ch_num in enumerate(sorted(st.session_state.generated_chapters.keys())):
+                with chapter_tabs[i]:
+                    chapter_data = st.session_state.generated_chapters[ch_num]
+                    st.write(chapter_data['content'])
+                    
+                    # Chapter status and actions
+                    col1, col2 = st.columns(2)
+                    with col1:
+                        status = st.selectbox(
+                            "Chapter Status", 
+                            ["Generated", "In Review", "Completed"],
+                            key=f"status_{ch_num}"
+                        )
+                    
+                    with col2:
+                        if st.button(f"Export Chapter {ch_num}"):
+                            # Export chapter functionality
+                            export_path = os.path.join('/data', f"chapter_{ch_num}.txt")
+                            with open(export_path, "w") as f:
+                                f.write(chapter_data['content'])
+                            st.success(f"Chapter {ch_num} exported to {export_path}!")
+        else:
+            st.info("No chapters generated yet. Start writing your book!")
     
     def run(self):
         """