Update app.py

app.py

@@ -37,6 +37,42 @@ load_dotenv()
 PIXABAY_API_KEY = os.getenv('PIXABAY_API_KEY')
 
 class EnhancedBookGenerator:
+    def _enhance_content(self, content: str, facts: Optional[List[str]] = None) -> str:
+        """Enhance content with facts and structure"""
+        if facts is None:
+            facts = []
+            
+        # Clean text
+        content = re.sub(r'\s+', ' ', content).strip()
+        
+        # Process with NLTK
+        sentences = sent_tokenize(content)
+        
+        # Add structure and facts
+        structured_content = []
+        current_section = []
+        fact_index = 0
+        
+        for i, sentence in enumerate(sentences):
+            current_section.append(sentence)
+            
+            # Add a fact every few sentences if available
+            if facts and i % 5 == 0 and fact_index < len(facts):
+                current_section.append(f"\nInteresting fact: {facts[fact_index]}\n")
+                fact_index += 1
+            
+            if (i + 1) % 5 == 0:
+                section_text = ' '.join(current_section)
+                if self.summarizer:
+                    heading = self.summarizer(section_text, max_length=10, min_length=5)[0]['summary_text']
+                    structured_content.append(f"\n## {heading.title()}\n")
+                structured_content.append(section_text)
+                current_section = []
+        
+        if current_section:
+            structured_content.append(' '.join(current_section))
+            
+        return '\n'.join(structured_content)
     def __init__(self):
         self.initialize_models()
         self.initialize_quality_metrics()
@@ -134,12 +170,27 @@ class EnhancedBookGenerator:
         }
 
     def generate_chapter_content(self, chapter_info: Dict) -> str:
-        """Generate chapter content with improved quality"""
+        """Generate chapter content incorporating user prompt"""
         try:
-            prompt = self._create_chapter_prompt(chapter_info)
-            content = self._generate_raw_content(prompt, chapter_info['word_count'])
-            enhanced_content = self._enhance_content(content)
+            # Include user prompt in chapter generation
+            base_prompt = self._create_chapter_prompt(chapter_info)
+            user_prompt = st.session_state.book_settings.get('user_prompt', '')
+            
+            if user_prompt:
+                enhanced_prompt = f"""
+                User's book vision: {user_prompt}
+                
+                Based on this vision, write a chapter that fits the following:
+                {base_prompt}
+                """
+            else:
+                enhanced_prompt = base_prompt
+                
+            content = self._generate_raw_content(enhanced_prompt, chapter_info['word_count'])
+            # Pass empty list for facts if none available
+            enhanced_content = self._enhance_content(content, facts=[])
             return enhanced_content
+            
         except Exception as e:
             st.error(f"Error generating chapter: {str(e)}")
             return f"Error generating chapter {chapter_info['title']}"
@@ -176,8 +227,11 @@ class EnhancedBookGenerator:
             
         return ' '.join(content_parts)
 
-    def _enhance_content(self, content: str, facts: List[str]) -> str:
+    def _enhance_content(self, content: str, facts: Optional[List[str]] = None) -> str:
         """Enhance content with facts and structure"""
+        if facts is None:
+            facts = []
+            
         # Clean text
         content = re.sub(r'\s+', ' ', content).strip()
         
@@ -192,8 +246,8 @@ class EnhancedBookGenerator:
         for i, sentence in enumerate(sentences):
             current_section.append(sentence)
             
-            # Add a fact every few sentences
-            if i % 5 == 0 and fact_index < len(facts):
+            # Add a fact every few sentences if available
+            if facts and i % 5 == 0 and fact_index < len(facts):
                 current_section.append(f"\nInteresting fact: {facts[fact_index]}\n")
                 fact_index += 1
             
@@ -371,9 +425,20 @@ def main():
         display_preview_download()
 
 def display_book_settings(templates):
-    """Display the book settings page"""
+    """Display the book settings page with enhanced user prompts"""
     st.header("Book Settings")
     
+    # User's creative prompt
+    st.subheader("Tell us about your book idea")
+    user_prompt = st.text_area(
+        "Describe your book idea, themes, feelings, and key elements you want to include",
+        value=st.session_state.get('user_prompt', ''),
+        height=150,
+        help="Be as detailed as possible. Include themes, mood, character ideas, plot elements, or any other aspects you want in your book.",
+        placeholder="Example: I want to write a book about a journey of self-discovery in a dystopian world. The main themes should include hope, resilience, and the power of human connection..."
+    )
+    st.session_state.user_prompt = user_prompt
+    
     col1, col2 = st.columns(2)
     
     with col1:
@@ -408,19 +473,32 @@ def display_book_settings(templates):
             key="words_per_chapter_input"
         )
 
-    # Chapter titles
+    # Generate intelligent chapter titles based on user prompt
+    if user_prompt and st.button("Generate Chapter Titles"):
+        with st.spinner("Generating chapter titles based on your prompt..."):
+            chapter_titles = generate_intelligent_chapter_titles(
+                user_prompt, 
+                num_chapters, 
+                template, 
+                genre
+            )
+            st.session_state.generated_chapter_titles = chapter_titles
+    else:
+        chapter_titles = st.session_state.get('generated_chapter_titles', 
+                        [f"Chapter {i+1}" for i in range(num_chapters)])
+
+    # Display and allow editing of generated chapter titles
     st.subheader("Chapter Titles")
-    chapter_titles = []
+    edited_chapter_titles = []
     cols = st.columns(3)
     for i in range(num_chapters):
         with cols[i % 3]:
-            default_title = templates[template]["example_titles"][i % len(templates[template]["example_titles"])]
             title = st.text_input(
                 f"Chapter {i+1}",
-                value=default_title,
+                value=chapter_titles[i] if i < len(chapter_titles) else f"Chapter {i+1}",
                 key=f"chapter_title_{i}"
             )
-            chapter_titles.append(title)
+            edited_chapter_titles.append(title)
 
     if st.button("Save Settings", key="save_settings_button"):
         st.session_state.book_settings = {
@@ -429,11 +507,31 @@ def display_book_settings(templates):
             "genre": genre,
             "num_chapters": num_chapters,
             "words_per_chapter": words_per_chapter,
-            "chapter_titles": chapter_titles,
-            "total_words": num_chapters * words_per_chapter
+            "chapter_titles": edited_chapter_titles,
+            "total_words": num_chapters * words_per_chapter,
+            "user_prompt": user_prompt
         }
         st.success("✅ Settings saved successfully! You can now proceed to Generate Content.")
 
+def generate_intelligent_chapter_titles(prompt, num_chapters, template, genre):
+    """Generate intelligent chapter titles based on user input"""
+    # This is a placeholder for the actual AI generation
+    # You would typically use an LLM API here
+    system_prompt = f"""
+    Based on the following user's book idea, generate {num_chapters} chapter titles.
+    The book is in the {genre} genre and follows the {template} style.
+    
+    User's book idea:
+    {prompt}
+    
+    Generate unique, engaging chapter titles that follow a logical progression
+    and reflect the themes and elements mentioned in the user's prompt.
+    """
+    
+    # For now, returning placeholder titles
+    # Replace this with actual AI generation
+    return [f"Generated Chapter {i+1}" for i in range(num_chapters)]
+
 def display_generate_content():
     """Display the content generation page"""
     if not st.session_state.book_settings: