Create app.py

app.py

@@ -0,0 +1,243 @@
+# File: app.py
+# Location: /app.py (root directory for Hugging Face Spaces)
+# Description: Main Streamlit application for AI Book Writing Assistant
+# Deployment: Directly used in Hugging Face Spaces
+
+import streamlit as st
+import os
+import json
+from multi_agent_book_workflow import BookWritingOrchestrator
+
+class BookWritingApp:
+    def __init__(self):
+        """
+        Initialize the Streamlit application for book writing
+        
+        Manages session state for the entire book writing project
+        """
+        # Set page configuration
+        st.set_page_config(
+            page_title="AI Book Writing Assistant",
+            page_icon="📖",
+            layout="wide"
+        )
+        
+        # Initialize session state
+        if 'project_orchestrator' not in st.session_state:
+            st.session_state.project_orchestrator = None
+        
+        if 'book_concept' not in st.session_state:
+            st.session_state.book_concept = None
+        
+        # Track generated chapters
+        if 'generated_chapters' not in st.session_state:
+            st.session_state.generated_chapters = {}
+        
+        # Track project metadata
+        if 'project_metadata' not in st.session_state:
+            st.session_state.project_metadata = {
+                'title': 'Untitled Project',
+                'genre': 'Unspecified',
+                'total_chapters': 0
+            }
+
+    def render_book_generation_interface(self):
+        """
+        Comprehensive interface for book generation with progress tracking
+        """
+        # Tabs for different aspects of book writing
+        tab1, tab2, tab3 = st.tabs([
+            "Book Concept", 
+            "Chapter Generation", 
+            "Book Progress"
+        ])
+        
+        with tab1:
+            self._render_concept_development()
+        
+        with tab2:
+            self._render_chapter_generation()
+        
+        with tab3:
+            self._render_book_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"):
+            if not st.session_state.project_orchestrator:
+                st.session_state.project_orchestrator = BookWritingOrchestrator()
+            
+            with st.spinner("Developing Book Concept..."):
+                try:
+                    book_concept = st.session_state.project_orchestrator.generate_book_concept(
+                        initial_concept
+                    )
+                    st.session_state.book_concept = book_concept
+                    
+                    # Update project metadata
+                    st.session_state.project_metadata.update({
+                        'title': book_concept.get('title', 'Untitled Project'),
+                        'genre': book_concept.get('genre', 'Unspecified')
+                    })
+                    
+                    # Display Concept Details
+                    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 with progress tracking
+        """
+        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_book_progress(self):
+        """
+        Render a comprehensive view of book writing progress
+        """
+        st.header("📊 Book Writing Progress")
+        
+        # Project Metadata
+        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
+                            with open(f"chapter_{ch_num}.txt", "w") as f:
+                                f.write(chapter_data['content'])
+                            st.success(f"Chapter {ch_num} exported!")
+        else:
+            st.info("No chapters generated yet. Start writing your book!")
+
+    def run(self):
+        """
+        Run the Streamlit application
+        """
+        # Main title
+        st.title("🚀 AI-Powered Book Writing Assistant")
+        
+        # Render the comprehensive book generation interface
+        self.render_book_generation_interface()
+
+def main():
+    # Initialize and run the application
+    app = BookWritingApp()
+    app.run()
+
+if __name__ == "__main__":
+    main()