Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -289,9 +289,17 @@ class EnhancedPDFFormatter(FPDF):
|
|
289 |
return None
|
290 |
|
291 |
def initialize_session_state():
|
292 |
-
"""Initialize enhanced session state"""
|
293 |
if 'book_settings' not in st.session_state:
|
294 |
-
st.session_state.book_settings = {
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
295 |
if 'generated_book' not in st.session_state:
|
296 |
st.session_state.generated_book = None
|
297 |
if 'current_step' not in st.session_state:
|
@@ -300,9 +308,14 @@ def initialize_session_state():
|
|
300 |
st.session_state.user_prompt = ""
|
301 |
if 'author_name' not in st.session_state:
|
302 |
st.session_state.author_name = ""
|
|
|
|
|
303 |
|
304 |
def main():
|
305 |
st.set_page_config(page_title="AI Book Generator", layout="wide")
|
|
|
|
|
|
|
306 |
|
307 |
# Book templates
|
308 |
book_templates = {
|
@@ -346,6 +359,7 @@ def main():
|
|
346 |
["Book Settings", "Generate Content", "Preview & Download"]
|
347 |
)
|
348 |
st.session_state.current_step = selected_step
|
|
|
349 |
st.title("AI Book Generator")
|
350 |
|
351 |
# Main content area
|
|
|
289 |
return None
|
290 |
|
291 |
def initialize_session_state():
|
292 |
+
"""Initialize enhanced session state with default values"""
|
293 |
if 'book_settings' not in st.session_state:
|
294 |
+
st.session_state.book_settings = {
|
295 |
+
'title': '',
|
296 |
+
'template': '',
|
297 |
+
'genre': '',
|
298 |
+
'num_chapters': 0,
|
299 |
+
'words_per_chapter': 0,
|
300 |
+
'chapter_titles': [],
|
301 |
+
'total_words': 0
|
302 |
+
}
|
303 |
if 'generated_book' not in st.session_state:
|
304 |
st.session_state.generated_book = None
|
305 |
if 'current_step' not in st.session_state:
|
|
|
308 |
st.session_state.user_prompt = ""
|
309 |
if 'author_name' not in st.session_state:
|
310 |
st.session_state.author_name = ""
|
311 |
+
if 'book_generator' not in st.session_state:
|
312 |
+
st.session_state.book_generator = EnhancedBookGenerator()
|
313 |
|
314 |
def main():
|
315 |
st.set_page_config(page_title="AI Book Generator", layout="wide")
|
316 |
+
|
317 |
+
# Initialize session state at the start
|
318 |
+
initialize_session_state()
|
319 |
|
320 |
# Book templates
|
321 |
book_templates = {
|
|
|
359 |
["Book Settings", "Generate Content", "Preview & Download"]
|
360 |
)
|
361 |
st.session_state.current_step = selected_step
|
362 |
+
|
363 |
st.title("AI Book Generator")
|
364 |
|
365 |
# Main content area
|