Update aAzel.py

aAzel.py

@@ -1,3 +1,6 @@
+To implement prompt chaining for generating longer chapters without the need for a "Continue Writing" button, we can modify the `generate_chapter_part` function and the chapter generation logic. Here's the updated code:
+
+```python
 import google.generativeai as genai
 import streamlit as st
 from dotenv import load_dotenv
@@ -29,12 +32,20 @@ Embrace your persona as a masterful storyteller, capable of crafting captivating
 6. **Write with Passion and Purpose:** Infuse your writing with genuine passion and a clear sense of purpose.  Let your words resonate with the reader, leaving a lasting impression and sparking further thought or inspiration.
 """
 
-def generate_chapter_part(prompt, previous_output=None):
+def generate_chapter_part(prompt, previous_output=None, max_tokens=2048):
     """Generates a part of a chapter using prompt chaining based on previous output."""
     if previous_output:
         prompt = prompt + "\n\n" + previous_output
-    response = model.generate_content([author_persona, author_guidelines, prompt])
-    return response.text
+
+    response = ""
+    while True:
+        partial_response = model.generate_content([author_persona, author_guidelines, prompt], max_output_tokens=max_tokens)
+        response += partial_response.text
+        if len(model.count_tokens(response)) >= max_tokens:
+            break
+        prompt = partial_response.text
+
+    return response
 
 def display_text_with_wrapping(text, wrap_length=80):
     """Displays text with word wrapping for better readability."""
@@ -149,35 +160,11 @@ def main():
                            * Interspersed with real-world examples and historical anecdotes to illustrate key points.
                            """
                 
-                chapter_content = generate_chapter_part(prompt, st.session_state.chapter_content)
+                chapter_content = generate_chapter_part(prompt, st.session_state.chapter_content, max_tokens=4096)
                 st.session_state.chapter_content = chapter_content
                 st.markdown(f"## {chapter_name}")
                 display_text_with_wrapping(chapter_content)
 
-                # Continue writing option
-                if st.button("Continue Writing"):
-                    if "chapter_content" in st.session_state:
-                        prompt = f"""Continue writing the chapter titled '{chapter_name}' for a {genre} book titled '{book_title}'. 
-                                   The book is about: {book_description}. 
-                                   Here are additional details about the chapter: 
-                                   
-                                   {chapter_details}
-
-                                   Ensure the chapter has a clear format, engaging style, and fits within the overall narrative.
-                                   
-                                   **Writing Style:** 
-                                   * Direct and assertive, conveying a sense of urgency and importance.
-                                   * Engaging and accessible, avoiding overly technical jargon.
-                                   * Motivational and empowering, inspiring readers to embrace the challenge of cognitive growth.
-                                   * Interspersed with real-world examples and historical anecdotes to illustrate key points.
-                                   """
-                        chapter_content = generate_chapter_part(prompt, st.session_state.chapter_content)
-                        st.session_state.chapter_content = chapter_content
-                        st.markdown(f"## {chapter_name}")
-                        display_text_with_wrapping(chapter_content)
-                    else:
-                        st.warning("Please generate the chapter first.")
-
                 # Download chapter button
                 st.download_button(
                     label="Download Chapter",