Merge pull request Bklieger#39 from bklieger-groq/main

fix: Improve prompt engineering for higher quality books, remove re-enable without clearing book contents

infinite_bookshelf/agents/structure_writer.py

@@ -20,7 +20,7 @@ def generate_book_structure(
             },
             {
                 "role": "user",
-                "content": f"Write a comprehensive structure, omiting introduction and conclusion sections (forward, author's note, summary), for a long (>300 page) book. It is very important that use the following subject and additional instructions to write the book. \n\n<subject>{prompt}</subject>\n\n<additional_instructions>{additional_instructions}</additional_instructions>",
+                "content": f"Write a comprehensive structure, omiting introduction and conclusion sections (forward, author's note, summary), for a book. Only provide up to one level of depth for nested sections. Make clear titles and descriptions that have no overlap with other sections. It is very important that use the following subject and additional instructions to write the book. \n\n<subject>{prompt}</subject>\n\n<additional_instructions>{additional_instructions}</additional_instructions>",
             },
         ],
         temperature=0.3,

main.py

@@ -150,8 +150,6 @@ def stream_section_content(sections):
         except json.JSONDecodeError:
             st.error("Failed to decode the book structure. Please try again.")
 
-        enable()
-
 except Exception as e:
     st.session_state.button_disabled = False
     st.error(e)