Update aAzel.py

aAzel.py

@@ -39,66 +39,77 @@ Embrace your persona as a masterful storyteller, capable of crafting captivating
 def generate_book_component(prompt, previous_output=None):
     """Generates a book component using prompt chaining based on previous output."""
     if previous_output:
-        prompt = prompt + "\n\n" + previous_output 
+        prompt = prompt + "\n\n" + previous_output
     response = model.generate_content([author_persona, author_guidelines, prompt])
     return response.text
 
+
 def main():
     st.title("AI Author")
     st.header("Write your next bestseller!")
 
-    # Genre Selection
-    genres = ["Science Fiction", "Fantasy", "Mystery", "Thriller", "Romance", "Historical Fiction", "Horror", "Literary Fiction", "Young Adult", "Children's Literature", "Non-Fiction"]
+    # Book Information Input
+    book_title = st.text_input("Enter your book title:")
+    genres = ["Science Fiction", "Fantasy", "Mystery", "Thriller", "Romance", 
+              "Historical Fiction", "Horror", "Literary Fiction", "Young Adult", 
+              "Children's Literature", "Non-Fiction"]
     genre = st.selectbox("Choose a Genre:", genres)
-
-    # Book Description
-    book_description = st.text_area("Describe your book:", "Enter a detailed description of your book here...")
-
-    if st.button("Start Writing"):
-        if not book_description:
-            st.warning("Please provide a description for your book.")
-            return
-
-        # Prompt Chaining for Book Creation
-        title = generate_book_component(f"Generate a compelling title for a {genre} book about: {book_description}")
-        st.subheader("Title:")
-        st.write(title)
-
-        toc = generate_book_component(f"Create a detailed Table of Contents for a {genre} book titled '{title}' about: {book_description}", title)
-        st.subheader("Table of Contents:")
-        st.write(toc)
-
-        book_content = ""
-        chapters = toc.split("\n") 
-
+    book_description = st.text_area("Describe your book:", 
+                                   "Enter a detailed description of your book here...")
+
+    # Store book components in session state
+    if 'toc' not in st.session_state:
+        st.session_state.toc = None
+    if 'chapter_formats' not in st.session_state:
+        st.session_state.chapter_formats = {}
+    if 'book_content' not in st.session_state:
+        st.session_state.book_content = ""
+
+    # Generate Table of Contents
+    if st.button("Generate Table of Contents"):
+        if not book_title or not book_description:
+            st.warning("Please enter both the book title and description.")
+        else:
+            prompt = f"Create a detailed Table of Contents for a {genre} book titled '{book_title}' about: {book_description}"
+            st.session_state.toc = generate_book_component(prompt, book_title)
+            st.subheader("Table of Contents:")
+            st.write(st.session_state.toc)
+
+    # Generate Chapter Format and Chapter Content
+    if st.session_state.toc:
+        chapters = st.session_state.toc.split("\n")
         for i, chapter in enumerate(chapters):
-            if chapter: 
-                st.write(f"## Chapter {i+1}: {chapter}")
+            if chapter:
+                st.markdown(f"## Chapter {i+1}: {chapter}")
 
                 # Generate Chapter Format
-                format_prompt = f"Suggest a detailed format for writing chapter '{chapter}' of the {genre} book '{title}'. Consider the narrative flow and genre conventions."
-                chapter_format = generate_book_component(format_prompt, toc)
-                st.write("**Chapter Format:**")
-                st.write(chapter_format)
-
-                # Generate Chapter Content Following the Format
-                chapter_content = ""
-                format_sections = chapter_format.split("\n")
-                for section in format_sections:
-                    if section:
-                        section_prompt = f"Write the '{section.strip()}' section of chapter '{chapter}' for the {genre} book '{title}', ensuring it fits into the overall narrative and follows the previous content."
-                        section_content = generate_book_component(section_prompt, chapter_content)  # Pass previous sections as context
-                        chapter_content += f"{section.strip()}:\n{section_content}\n\n" 
-
-                st.write("**Chapter Content:**")
-                st.write(chapter_content)
-
-                book_content += f"## Chapter {i+1}: {chapter}\n\n{chapter_format}\n\n{chapter_content}\n\n"
-
+                if chapter not in st.session_state.chapter_formats:
+                    if st.button(f"Generate Format for Chapter {i+1}"):
+                        format_prompt = f"Suggest a detailed format for writing chapter '{chapter}' of the {genre} book '{book_title}'. Consider the narrative flow and genre conventions."
+                        st.session_state.chapter_formats[chapter] = generate_book_component(format_prompt, st.session_state.toc)
+                if chapter in st.session_state.chapter_formats:
+                    st.write("**Chapter Format:**")
+                    st.write(st.session_state.chapter_formats[chapter])
+
+                    # Generate Chapter Content 
+                    if st.button(f"Generate Chapter {i+1}"):
+                        chapter_content = ""
+                        format_sections = st.session_state.chapter_formats[chapter].split("\n")
+                        for section in format_sections:
+                            if section:
+                                section_prompt = f"Write the '{section.strip()}' section of chapter '{chapter}' for the {genre} book '{book_title}', ensuring it fits into the overall narrative and follows the previous content."
+                                section_content = generate_book_component(section_prompt, chapter_content) 
+                                chapter_content += f"{section.strip()}:\n{section_content}\n\n"
+                        st.session_state.book_content += f"## Chapter {i+1}: {chapter}\n\n{st.session_state.chapter_formats[chapter]}\n\n{chapter_content}\n\n"
+                        st.write("**Chapter Content:**")
+                        st.write(chapter_content)
+
+    # Download Button 
+    if st.session_state.book_content:
         st.download_button(
             label="Download Book",
-            data=book_content,
-            file_name=f"{title.replace(' ', '_')}.txt",
+            data=st.session_state.book_content,
+            file_name=f"{book_title.replace(' ', '_')}.txt",
             mime="text/plain"
         )